术语表

Weights & Biases

使用Weights & Biases 简化机器学习工作流程。跟踪、可视化和协作实验,实现更快、可重复的人工智能开发。

使用Ultralytics HUB 对YOLO 模型进行简单培训

了解更多

在机器学习和人工智能开发领域,有效管理实验和理解模型行为是成功的关键。Weights & Biases (W&B)是一个功能强大的平台,旨在简化这些流程,提供跟踪、可视化和协作机器学习项目的工具。它可以帮助个人和团队优化工作流程、了解模型性能并复制实验,最终加快人工智能解决方案的开发和部署。

Weights & Biases 是什么?

Weights & Biases 是一个全面的 MLOps(机器学习操作)平台,专为增强机器学习从业人员和研究人员的工作流程而设计。它是一个集中式系统,用于跟踪和可视化机器学习实验的各个方面,从数据集和超参数到训练指标和模型版本。通过提供清晰、有序的实验流程概览,Weights & Biases 可促进更好的模型开发、更便捷的协作和更可重复的结果。它可与流行的机器学习框架无缝集成,如 PyTorch和TensorFlow 等流行的机器学习框架无缝集成,使其成为适用于各种人工智能项目(包括使用Ultralytics YOLO 模型的项目)的通用工具。

主要特点Weights & Biases

Weights & Biases 提供了一套旨在改进机器学习工作流程的功能:

  • 实验跟踪:监控并记录关键的实验细节,如超参数、模型配置、训练指标(如损失和准确率)以及系统资源使用情况。这样就可以轻松比较和分析不同的运行情况,以确定最佳设置并跟踪随时间推移取得的进展。例如,在训练用于物体检测的Ultralytics YOLOv8 模型时,W&B 可以实时跟踪平均精度 (mAP)和损失曲线。
  • 数据可视化:通过交互式可定制仪表盘深入了解实验结果。实时可视化指标、训练曲线甚至模型预测。通过这些可视化功能,可以更轻松地识别趋势、发现异常并了解不同参数对模型性能的影响。可视化对象检测结果(如叠加在图像上的边界框)对于调试和提高模型准确性特别有用。
  • 协作工具:通过轻松共享实验结果、仪表盘和报告,促进团队合作。通过集中管理实验数据和见解,团队可以更有效地开展协作,从而更容易复制结果并在彼此工作的基础上更进一步。这对使用Ultralytics HUB 开发的项目尤为有利,因为团队可以集体管理和跟踪其模型训练进度。
  • 集成能力:Weights & Biases 可与各种机器学习工具和平台顺利集成,包括流行的框架,如 PyTorch和TensorFlow 等流行框架,以及Ultralytics HUB 等平台。这使得用户可以轻松地将 W&B 整合到现有的工作流程中,而不会造成重大干扰。详细的集成指南可用于Ultralytics YOLO,简化了将Ultralytics 项目连接到 W&B 平台的过程。

真实世界的应用Weights & Biases

Weights & Biases 被广泛应用于各个领域,以促进机器学习的发展:

示例 1:加强医疗保健中的医学图像分析

在医疗保健领域,医学图像分析对于准确诊断和治疗规划至关重要。为使用核磁共振扫描进行肿瘤检测等任务开发人工智能模型的团队利用Weights & Biases 对不同模型和训练配置的性能进行细致的跟踪和比较。通过监控跨训练时程的验证损失、准确性曲线下面积(AUC)等指标,研究人员可以确保模型不断改进,并找出最有效的方法。他们还可以对样本预测进行可视化,以确保人工智能模型能正确识别医学图像中的肿瘤,从而提高人工智能驱动的诊断工具的可靠性。

示例 2:优化零售库存管理中的目标检测

零售企业将物体检测模型用于各种应用,包括库存管理和优化库存水平。通过将Weights & Biases 与 Ultralytics YOLOv8,零售公司可以实时跟踪其模型的性能。例如,他们可以监控检测货架上产品的模型的推理速度、精确度召回率。通过这种实时反馈,可以对模型进行微调,以达到最佳精度和速度,确保高效的库存跟踪,减少缺货,从而提高运营效率和客户满意度。

Weights & Biases 与同类工具的比较

虽然有 TensorBoard 和 MLflow 等其他实验跟踪工具,但Weights & Biases 凭借其全面、开发人员优先的方法脱颖而出。与主要侧重于可视化的 TensorBoard 和强调模型部署的 MLflow 不同,Weights & Biases 提供了一个在实验跟踪、可视化和团队协作方面都很出色的集成平台。其用户友好型仪表板和协作功能使其对从事复杂人工智能项目的团队特别有吸引力,为管理从实验到模型完善的整个机器学习生命周期提供了强大的解决方案。

阅读全部