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.
-
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.
-
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 .