Weights & Biases
探索 Weights & Biases 如何为 Ultralytics YOLO26 简化 MLOps。学习跟踪实验、优化超参数并管理工件以获得更好的模型。
Weights & Biases(通常简称为 W&B 或 WandB)是一个全面的机器学习运维(MLOps)平台,旨在帮助数据科学家和机器学习工程师简化模型开发工作流。作为一个以开发者为先的工具,它充当了集中式的记录系统,用于追踪实验、对数据集和模型进行版本控制,并实时可视化性能指标。在人工智能复杂的领域中,保持可复现性以及对训练运行的可见性至关重要;Weights & Biases 通过自动记录超参数、系统指标和输出文件来解决这一问题,从而使团队能够高效地比较不同的实验并识别出表现最好的配置。
机器学习的核心能力#
Weights & Biases 的主要价值在于它能够组织深度学习模型训练中通常混乱的过程。它提供了一套可以直接与 PyTorch 和 Ultralytics 生态系统等流行框架集成的工具。
- 实验追踪:此功能记录所有配置参数,例如学习率、批大小和模型架构。它还记录随时间变化的动态指标,如损失函数和准确率,并以交互式图表的形式呈现它们。
- 超参数优化:W&B Sweeps 自动化了超参数调优的过程。通过探索不同的参数组合,用户无需人工干预即可最大化诸如平均精度均值 (mAP) 等模型性能指标。
- 资产管理:为确保完整的血缘追踪,W&B Artifacts 对数据集和模型检查点进行版本控制。这使用户能够准确追溯是哪个数据版本产生了特定的模型,这是健壮的模型监控的关键组成部分。
- 系统监控:该平台追踪硬件使用情况,包括 GPU 利用率、内存消耗和温度。这有助于在计算密集的训练期间识别瓶颈并确保高效的资源分配。
实际应用#
Weights & Biases 在各个行业中被广泛使用,以加速计算机视觉和 NLP 解决方案的部署。
-
协作研发:大型 AI 研究团队使用 W&B 即时共享实验结果。例如,开发自动驾驶感知系统的团队可以让多名工程师训练不同的目标检测架构。W&B 将这些运行聚合到一个单一仪表板中,使团队能够协作分析哪种架构能最好地处理边缘情况,从而促进更快的迭代周期。
-
生产模型维护:在工业环境中(如制造质量控制),必须定期使用新数据重新训练模型,以防止数据偏移。W&B 帮助工程师将候选生产模型的性能与当前基线进行比较,确保只有具有优异精确率和召回率的模型才被部署到边缘端。
与 Ultralytics YOLO 的集成#
Weights & Biases 与 Ultralytics 之间的集成非常顺畅,为目标检测、分割和姿态估计任务提供了丰富的可视化效果。当训练像 YOLO26 这样的现代模型时,集成会自动记录指标、边界框预测和混淆矩阵。
此代码片段演示了如何利用自动记录功能。只需安装客户端,训练过程就会将结果同步到云端。
from ultralytics import YOLO
# Ensure the wandb client is installed
# pip install wandb
# Load the YOLO26 model (latest generation)
model = YOLO("yolo26n.pt")
# Train the model on the COCO8 dataset
# The integration automatically detects wandb and logs metrics
model.train(data="coco8.yaml", epochs=5, project="YOLO26_Experiments", name="run_01")区别:平台与神经网络参数#
务必将“Weights & Biases”平台与神经网络中权重 (weights) 和 偏置 (biases) 的基础概念区分开来。
- **权重和偏置(参数):**在神经网络中,“权重”是决定神经元之间连接强度的可学习参数,而“偏置”是允许激活函数发生偏移的附加参数。这些是在反向传播过程中优化的数学值。
- **Weights & Biases(平台):**这是本页面讨论的外部软件工具。虽然该平台追踪神经网络权重和偏置的值与梯度以供分析,但它是位于训练数据和过程之上的管理层,而不是数学组件本身。
对于希望在实验追踪的同时管理从标注到部署的整个生命周期的用户,Ultralytics Platform 还提供了强大的工具,用以补充Weights & Biases 集成所提供的详细指标记录。






