O que é
Cada etapa de treinamento calcula, para cada peso, qual direção reduziria a perda. A taxa de aprendizado decide o quão longe ir nessa direção. É um número único, geralmente escrito em notação científica: 1e-5 significa 0,00001. Parece minúsculo porque um modelo tem bilhões de pesos e cada etapa move todos eles de uma vez; pequenas mudanças individuais se somam a uma grande mudança de comportamento.
new weight = old weight − learning rate × (gradient, as scaled by the optimizer)
1e-4 = 0.0001 2e-5 = 0.00002 1e-5 = 0.00001Com Adam e AdamW, o gradiente é primeiro normalizado por peso, de modo que a taxa de aprendizado fica próxima ao tamanho real da mudança de cada peso por etapa, independentemente do tamanho bruto do gradiente. É por isso que as taxas típicas se mantêm razoavelmente entre modelos de tamanho semelhante, e por que uma taxa dez vezes mais alta é um problema grave em vez de um pequeno.
Muito baixa, adequada, muito alta
| O que você vê | Causa provável | Tente |
|---|---|---|
| A perda cai muito lentamente; as respostas quase não diferem do modelo base | Muito baixa | Aumente-a, ex.: um multiplicador de 1,5–2 |
| Uma queda rápida, seguida por um nivelamento suave que a perda de validação acompanha | Adequada | Mantenha-a |
| A perda é irregular, apresenta picos ou volta a subir; as respostas tornam-se repetitivas ou estranhas | Muito alta | Reduza-a pela metade; adicione ou prolongue o warm-up |
| A perda de treinamento cai rápido, mas a perda de validação sobe cedo | Muito alta para este tamanho de dados, ou muitas épocas | Diminua a taxa ou o número de épocas |
| A perda torna-se NaN (não é um número) | Alta demais, ou um problema numérico | Diminua-a substancialmente |
Uma taxa muito alta também agrava o esquecimento: grandes atualizações sobrescrevem habilidades gerais que o modelo base possuía, mesmo quando a perda nos seus dados parece adequada.
Cronogramas: aquecimento e decaimento
- Warm-up. Durante os primeiros por cento das etapas, a taxa sobe linearmente de quase zero até o seu pico. No início, as médias móveis do Adam baseiam-se em muito poucos gradientes, de modo que passos de tamanho normal correm o risco de um primeiro movimento ruim. O warm-up gradual foi popularizado para o treinamento com lotes grandes e agora é uma prática padrão.
- Decaimento. Após o pico, a taxa diminui em direção a zero, para que as atualizações finais façam ajustes finos em vez de saltar em torno de uma boa solução. O decaimento linear a reduz em linha reta e é o padrão no Trainer da Hugging Face. O decaimento cosseno segue meia onda de cosseno: permanece alto por mais tempo e depois diminui.
- Constante. Algumas execuções curtas mantêm a taxa plana após o warm-up. É simples e adequado quando há apenas algumas etapas.
warm-up (step < W): lr = peak × step / W
linear decay: lr = peak × (T − step) / (T − W)
cosine decay: lr = peak × ½ × (1 + cos(π × (step − W) / (T − W)))
W = warm-up steps, T = total stepsValores típicos
| Tipo de treinamento | Taxa de aprendizado típica |
|---|---|
| Ajuste fino completo de um modelo de linguagem (AdamW) | Em torno de 1e-5 |
| Adaptador LoRA | Em torno de 1e-4 a 2e-4, aproximadamente 10× a taxa do ajuste fino completo |
| Ajuste de preferência (DPO) | Inferior à execução supervisionada correspondente |
| Ajuste fino por reforço | Passos muito menores; cerca de 1e-5 com LoRA |
| Modelos hospedados com um multiplicador | Padrão do provedor × 0,5–2 |
Estes são pontos de partida, não regras. A melhor taxa depende do modelo, dos dados e do tamanho do lote, e a única maneira confiável de encontrá-la é comparando algumas execuções na perda de validação e na avaliação.
Por que métodos de preferência e RL utilizam passos menores
O treinamento supervisionado aprende a partir de respostas fixas e corretas. O ajuste de preferência e o aprendizado por reforço aprendem a partir de comparações e pontuações das próprias saídas do modelo, um sinal com mais ruído, e seu propósito geralmente é refinar um modelo que já funciona. Passos grandes o afastariam muito desse ponto de partida, onde as respostas podem se desviar, ficar mais longas ou começar a manipular o que o sinal recompensa. Taxas menores mantêm cada atualização modesta.
- Suponha que a taxa padrão do provedor para este modelo e conjunto de dados seja 2e-5. Provedores hospedados geralmente não a exibem.
- Multiplicador 1: a execução utiliza 2e-5.
- Multiplicador 2: 4e-5. Vale a pena testar quando uma execução completa quase não alterou o modelo.
- Multiplicador 0.5: 1e-5. Vale a pena testar quando a perda de validação aumentou precocemente ou as respostas a perguntas gerais ficaram piores do que as do modelo base.
- Altere em fatores de aproximadamente 2. Ajustes menores geralmente se perdem no ruído entre as execuções.
Erros comuns
- Aumentar a taxa de aprendizado para fazer um conjunto de dados pequeno ensinar mais. Isso geralmente adiciona instabilidade; mais dados ou dados melhores ajudam mais.
- Usar uma taxa de fine-tuning completo para LoRA, ou uma taxa de LoRA para fine-tuning completo: um mal aprende, o outro pode danificar o modelo.
- Pular o warm-up com uma taxa alta, o que torna as primeiras atualizações as maiores e as menos informadas.
- Julgar a taxa apenas pela perda de treinamento. Uma taxa muito alta ainda pode apresentar queda na perda de treinamento, enquanto a perda de validação e as respostas pioram.
Referências e leitura complementar
Fontes externas abrem em uma nova aba.
- 01OptimizationHugging Face Transformers docs · huggingface.coAdamW e os cronogramas padrão (linear, cosseno, constante com warm-up) conforme implementados.
- 02SGDR: Stochastic Gradient Descent with Warm RestartsLoshchilov & Hutter, 2016 · arXiv · arxiv.orgIntroduziu o cronograma de taxa de aprendizado cosseno agora usado em todo o treinamento de modelos de linguagem.
- 03Accurate, Large Minibatch SGD: Training ImageNet in 1 HourGoyal et al., 2017 · arXiv · arxiv.orgPor que um warm-up gradual estabiliza o início do treinamento.
- 04LoRA Without RegretThinking Machines Lab, 2025 · thinkingmachines.aiMede a melhor taxa de aprendizado do LoRA em cerca de dez vezes a taxa de fine-tuning completo.
- 05Learning rateWikipedia · en.wikipedia.org