Glossário

Afinação

Descobre como o ajuste fino de modelos pré-treinados como YOLO optimiza a IA para tarefas específicas de um domínio com menos dados e recursos.

Treina os modelos YOLO simplesmente
com Ultralytics HUB

Aprende mais

O ajuste fino é o processo de pegar num modelo de aprendizagem automática pré-treinado e adaptá-lo a uma tarefa ou conjunto de dados específico, treinando-o com novos dados. Esta abordagem aproveita o conhecimento fundamental que o modelo já adquiriu, reduzindo os recursos computacionais e o tempo necessário em comparação com o treino de um modelo de raiz. O ajuste fino é especialmente eficaz em cenários em que os dados rotulados são escassos ou quando se pretende otimizar um modelo para uma aplicação específica de um domínio.

Importância do ajuste fino

Os modelos pré-treinados, como os utilizados na aprendizagem por transferência, são treinados em conjuntos de dados grandes e genéricos, como o ImageNet ou o COCO. Embora estes modelos sejam excelentes em tarefas gerais, o ajuste fino permite-lhes especializarem-se, concentrando-se em nuances específicas do domínio. Por exemplo, um modelo treinado para a deteção geral de objectos pode ser aperfeiçoado para detetar anomalias médicas em radiografia ou espécies animais específicas na monitorização da vida selvagem.

O ajuste fino é uma técnica económica e escalável que minimiza a necessidade de grandes conjuntos de dados e recursos computacionais extensivos. Isto torna-a particularmente valiosa para aplicações em sectores como os cuidados de saúde, a indústria transformadora e a agricultura.

Como funciona o ajuste fino

Normalmente, o ajuste fino envolve as seguintes etapas:

  1. Seleção de um modelo pré-treinado: Escolhe um modelo pré-treinado numa tarefa semelhante à sua tarefa alvo. Por exemplo, os modelos Ultralytics YOLO são pré-treinados em conjuntos de dados como COCO para tarefas de deteção de objectos.
  2. Congelamento de camadas: Inicialmente, algumas camadas do modelo pré-treinado são congeladas para reter as caraterísticas generalizadas. Apenas as camadas superiores são treinadas novamente no novo conjunto de dados.
  3. Personalizar o modelo: Modifica a arquitetura, se necessário. Isto pode incluir a adição de camadas específicas de tarefas, tais como cabeças de classificação para classificação de imagens ou camadas de regressão para estimativa de pose.
  4. Treina com novos dados: Treina o modelo no novo conjunto de dados usando uma taxa de aprendizagem menor, garantindo que os recursos originais não sejam sobrescritos excessivamente.
  5. Avalia o desempenho: Avalia o modelo ajustado utilizando métricas como o mAP e a pontuação F1 para garantir que o modelo tem um bom desempenho na tarefa específica.

Para obter um guia detalhado sobre a formação e o aperfeiçoamento dos modelos YOLO , consulta Ultralytics YOLO documentação de formação.

Aplicações reais do ajuste fino

1. Imagiologia médica

O ajuste fino é amplamente utilizado na imagiologia médica para tarefas como a deteção de tumores ou a segmentação de órgãos. Por exemplo, um modelo pré-treinado em tarefas gerais de deteção de objectos pode ser aperfeiçoado utilizando um conjunto de dados médicos para identificar tumores cerebrais em exames de ressonância magnética. Isto permite aos profissionais de saúde obter uma maior precisão e eficiência no diagnóstico.

Lê mais sobre a forma como a IA está a transformar o sector dos cuidados de saúde em IA nos cuidados de saúde.

2. Agricultura inteligente

Na agricultura, a afinação permite o desenvolvimento de modelos para desafios específicos, como a deteção de pragas ou a monitorização de culturas. Por exemplo, um modelo geral de deteção de objectos pode ser afinado com um conjunto de dados de pragas agrícolas locais, melhorando a precisão na identificação e mitigação de ameaças às culturas.

Sabe mais sobre o papel da IA na agricultura em IA na Agricultura.

3. Automatização do comércio retalhista

Os ambientes de retalho beneficiam de modelos aperfeiçoados para tarefas como a gestão de inventário e a análise do comportamento dos clientes. Um modelo pré-treinado pode ser ajustado para detetar com precisão tipos de produtos específicos ou analisar padrões de tráfego pedonal nas lojas.

Para obteres informações sobre o impacto da IA na eficiência do retalho, explora a IA no fabrico.

Afinação vs. Técnicas relacionadas

O ajustamento fino partilha semelhanças com a aprendizagem por transferência, mas diferem no âmbito e na aplicação. A aprendizagem por transferência envolve a utilização de um modelo pré-treinado como um extrator de caraterísticas sem alterar os seus pesos, enquanto o ajuste fino ajusta os pesos do modelo para a nova tarefa. O ajuste fino também difere do ajuste de hiperparâmetros, que optimiza as definições do modelo, mas não envolve a reciclagem das camadas do modelo.

Além disso, a afinação fina é diferente da aprendizagem zero-shot ou da aprendizagem few-shot. Estas técnicas visam generalizar para tarefas inéditas com o mínimo ou nenhum dado novo, ao passo que a afinação fina se baseia num conjunto de dados específico para melhorar o desempenho numa tarefa específica.

Vantagens da afinação com Ultralytics HUB

Ultralytics O HUB simplifica o processo de afinação, fornecendo uma plataforma intuitiva para a formação e implementação de modelos. Com funcionalidades como formação baseada na nuvem e modelos Ultralytics YOLO pré-treinados, os utilizadores podem facilmente afinar modelos para diversas aplicações sem necessitarem de conhecimentos técnicos extensivos. Sabe mais sobre o Ultralytics HUB e as suas capacidades.

O ajuste fino é uma pedra angular da aprendizagem automática moderna, permitindo que os programadores e investigadores personalizem modelos para tarefas específicas de forma eficiente. Ao tirar partido de modelos pré-treinados e de ferramentas como o Ultralytics HUB, o ajuste fino torna-se um método acessível e poderoso para otimizar o desempenho da IA em aplicações do mundo real.

Lê tudo