Yolo Vision Shenzhen
Shenzhen
Únete ahora
Glosario

Ajuste fino

Ajuste modelos de aprendizaje automático como Ultralytics YOLO para tareas específicas. Descubra aquí métodos, aplicaciones y buenas prácticas.

El ajuste fino es una técnica técnica de aprendizaje automático especializar un modelo preentrenado para una tarea o conjunto de datos específicos. En lugar de que requiere grandes cantidades de datos etiquetados y potencia computacional, el ajuste fino aprovecha el conocimiento existente de un modelo que ya ha aprendido patrones a partir de una gran cantidad de datos generales. de un modelo que ya ha aprendido patrones a partir de un amplio conjunto de datos generales. Este proceso es una aplicación práctica del aprendizaje por transferencia que permite a los desarrolladores lograr un alto rendimiento en problemas específicos -como detectar defectos de fabricación poco comunes o clasificar imágenes médicas- con recursos significativamente reducidos. o la clasificación de imágenes médicas, con recursos considerablemente reducidos.

Cómo funciona el ajuste fino

El proceso comienza con un modelo de base, como como un modelo de visión entrenado en ImageNet o un modelo lingüístico entrenado en el corpus de Internet. Estos modelos poseen un sólido conocimiento de características fundamentales como los bordes, las texturas y las formas de las imágenes, o la gramática y la semántica de los textos. de las imágenes, o la gramática y la semántica de los textos. Durante el ajuste, el modelo se expone a un nuevo conjunto de datos a un nuevo conjunto de datos personalizado, más pequeño y pertinente para la aplicación de destino.

El proceso de entrenamiento consiste en ajustar los pesos del modelo para adaptarlos a los matices de los nuevos datos. los nuevos datos. Normalmente, esto se hace utilizando un para preservar las valiosas características aprendidas durante la fase inicial de preentrenamiento, permitiendo al mismo tiempo que el modelo se adapte. En muchos flujos de trabajo de visión por ordenador (VC), los ingenieros pueden congelar las primeras capas de la columna vertebral, que detect básicas y sólo afinar las capas más profundas y el cabezal de capas más profundas y el cabezal de detección predicciones finales.

Ajuste fino con Ultralytics YOLO

Adaptar un modelo de vanguardia como Ultralytics YOLO11 a a sus datos específicos. La biblioteca se encarga automáticamente de las complejidades de cargar los pesos y configurar el bucle de entrenamiento.

El siguiente ejemplo muestra cómo cargar un modelo YOLO11 preentrenado y afinarlo en un conjunto de datos de muestra.

from ultralytics import YOLO

# Load a pre-trained YOLO11 model (uses COCO weights by default)
model = YOLO("yolo11n.pt")

# Fine-tune the model on a specific dataset configuration
# 'epochs' sets the training duration, and 'imgsz' defines input resolution
results = model.train(data="coco8.yaml", epochs=20, imgsz=640)

Aplicaciones en el mundo real

El perfeccionamiento tiende un puente entre las capacidades generales de la IA y los requisitos específicos del sector.

  • La IA en la sanidad: General pueden identificar objetos cotidianos como coches o gatos. Sin embargo, al ajustar estos modelos a conjuntos de datos de análisis de imágenes médicas, como radiografías radiografías o resonancias magnéticas, los médicos pueden crear herramientas especializadas para detect con gran o tumores cerebrales con gran precisión.
  • Fabricación industrial: En las líneas de producción, un modelo estándar puede no reconocer componentes patentados o defectos sutiles. Al ajustar un modelo un modelo sobre imágenes de la línea de montaje específica, los fabricantes pueden automatizar control de calidad para detectar defectos como microfisuras o desalineaciones, lo que mejora notablemente la eficacia.

Ajuste Fino vs. Conceptos Relacionados

Distinguir el ajuste fino de otros métodos de adaptación es crucial para seleccionar el enfoque adecuado:

  • Extracción de características: En este enfoque, el modelo preentrenado se trata como un extractor de características fijo. Los pesos se congelan por completo, y sólo se utiliza un clasificador simple (como un ( como un modelo de regresión lineal ). los resultados. El ajuste fino va un paso más allá mediante la actualización de los parámetros internos del modelo.
  • Ajuste fino eficiente de parámetros (PEFT): Mientras que el ajuste fino estándar actualiza todos o la mayoría de los parámetros del modelo, los métodos PEFT como LoRA (Low-Rank Adaptation) sólo actualizan un pequeño subconjunto de parámetros. Esto hace que el proceso sea mucho más rápido y requiera menos memoria, sobre todo en el caso de modelos lingüísticos masivos. grandes modelos lingüísticos (LLM).
  • Prompt Engineering: En consiste en crear entradas específicas (prompts) para guiar la salida de un modelo congelado sin cambiar sus pesos internos o entrenarlo con nuevos datos. o entrenarlo con nuevos datos.

Herramientas y Frameworks

Para llevar a cabo el ajuste fino, los desarrolladores recurren a marcos sólidos como PyTorch y TensorFlowque proporcionan la infraestructura necesaria para el descenso de gradiente y la retropropagación. Las bibliotecas modernas de Por ejemplo, el ecosistema Ultralytics permite la formación y validación de modelos sin fisuras. formación y validación de modelos. Al preparar los datos para para el ajuste fino, es esencial garantizar de alta calidad es esencial para evitar evitar el sesgo de los conjuntos de datos, que puede sesgar el en situaciones reales.

Únase a la comunidad Ultralytics

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

Únete ahora