Yolo Vision Shenzhen
Shenzhen
Únete ahora
Glosario

Weights & Biases

Descubra cómo Weights & Biases MLOps para Ultralytics . Aprenda a track , optimizar los hiperparámetros y gestionar los artefactos para obtener mejores modelos.

Weights & Biases a menudo abreviado como W&B o WandB) es una plataforma integral de operaciones de aprendizaje automático (MLOps) diseñada para ayudar a los científicos de datos y a los ingenieros de aprendizaje automático a optimizar su flujo de trabajo de desarrollo de modelos. Como herramienta pensada principalmente para desarrolladores, actúa como un sistema central de registro para realizar un seguimiento de los experimentos, gestionar las versiones de los conjuntos de datos y modelos, y visualizar las métricas de rendimiento en tiempo real. En el complejo panorama de la inteligencia artificial, es fundamental mantener la reproducibilidad y la visibilidad de las ejecuciones de entrenamiento; Weights & Biases esta cuestión registrando automáticamente los hiperparámetros, las métricas del sistema y los archivos de salida, lo que permite a los equipos comparar experimentos dispares e identificar de manera eficiente las configuraciones con mejor rendimiento.

Capacidades básicas en aprendizaje automático

El principal valor de Weights & Biases en su capacidad para organizar el proceso, a menudo caótico, de entrenamiento de modelos de aprendizaje profundo. Proporciona un conjunto de herramientas que se integran directamente con marcos populares como PyTorch y Ultralytics .

  • Seguimiento de experimentos: esta función registra todos los parámetros de configuración, como la velocidad de aprendizaje, el tamaño del lote y la arquitectura del modelo. También registra métricas dinámicas como las funciones de pérdida y la precisión a lo largo del tiempo, y las presenta en gráficos interactivos.
  • Optimización de hiperparámetros: W&B Sweeps automatiza el proceso de ajuste de hiperparámetros. Al explorar diferentes combinaciones de parámetros, los usuarios pueden maximizar las métricas de rendimiento del modelo, como la precisión media (mAP), sin intervención manual.
  • Gestión de artefactos: Para garantizar un seguimiento completo del linaje, W&B Artifacts controla las versiones de los conjuntos de datos y los puntos de control de los modelos. Esto permite a los usuarios rastrear exactamente qué versión de los datos produjo un modelo específico, un componente clave para una supervisión sólida de los modelos.
  • Supervisión del sistema: la plataforma realiza un seguimiento del uso del hardware, incluyendo GPU , consumo de memoria y temperatura. Esto ayuda a identificar cuellos de botella y a garantizar una asignación eficiente de los recursos durante las sesiones de entrenamiento que requieren un uso intensivo de la computación .

Aplicaciones en el mundo real

Weights & Biases utiliza ampliamente en diversas industrias para acelerar la implementación de soluciones de visión por ordenador y PLN.

  1. Investigación y desarrollo colaborativos: los grandes equipos de investigación en IA utilizan W&B para compartir los resultados experimentales al instante. Por ejemplo, un equipo que desarrolla un sistema de percepción para vehículos autónomos puede contar con varios ingenieros que entrenan diferentes arquitecturas de detección de objetos. W&B agrega estas ejecuciones en un único panel de control, lo que permite al equipo analizar de forma colaborativa qué arquitectura gestiona mejor los casos extremos, lo que fomenta ciclos de iteración más rápidos.
  2. Mantenimiento de modelos de producción: En entornos industriales, como el control de calidad de la fabricación, los modelos deben volver a entrenarse periódicamente con nuevos datos para evitar la deriva de datos. W&B ayuda a los ingenieros a comparar el rendimiento de un modelo de producción candidato con la línea de base actual, lo que garantiza que solo se implementen en el borde los modelos con una precisión y un recuerdo superiores.

Integración con Ultralytics YOLO

La integración entre Weights & Biases Ultralytics perfecta y proporciona visualizaciones enriquecidas para tareas de detección de objetos, segmentación y estimación de poses . Al entrenar un modelo moderno como YOLO26, la integración registra automáticamente métricas, predicciones de cuadros delimitadores y matrices de confusión.

Este fragmento muestra cómo aprovechar las capacidades de registro automático. Con solo instalar el cliente, el proceso de entrenamiento sincronizará los resultados en la nube.

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")

Distinción: parámetros de plataforma frente a parámetros de red neuronal

Es importante distinguir la plataformaWeights & Biases» de los conceptos fundamentales de las redes neuronales de pesos y sesgos.

  • Weights and Biases parámetros): En una red neuronal, las «ponderaciones» son los parámetros aprendibles que determinan la fuerza de la conexión entre neuronas, y los «sesgos» son parámetros adicionales que permiten desplazar la función de activación. Estos son los valores matemáticos optimizados durante la retropropagación.
  • Weights & Biases plataforma): Esta es la herramienta de software externa que se analiza en esta página. Si bien la plataforma realiza un seguimiento de los valores y gradientes de weights and biases de la red neuronal weights and biases su análisis, se trata de una capa de gestión situada por encima de los datos y el proceso de entrenamiento, no de los componentes matemáticos en sí mismos.

Para los usuarios que deseen gestionar todo el ciclo de vida, incluyendo las anotaciones y la implementación, además del seguimiento de los experimentos, la Ultralytics también ofrece herramientas robustas que complementan el registro detallado de métricas proporcionado por la Weights & Biases .

Únase a la comunidad Ultralytics

Únete al futuro de la IA. Conecta, colabora y crece con innovadores de todo el mundo

Únete ahora