Zubnet AIAprenderWiki › Early Stopping
Training

Early Stopping

Patience, Validation-Based Stopping
Parar o treinamento quando a performance em um validation set separado para de melhorar, em vez de treinar por um número fixo de passos. Enquanto o treinamento continua, a loss de treinamento continua caindo mas a loss de validação eventualmente começa a subir — o modelo está fazendo overfitting nos dados de treinamento. Early stopping pega esse ponto de inflexão e salva o melhor modelo antes que a qualidade degrade.

Por que importa

Early stopping é a técnica de regularização mais simples e efetiva para fine-tuning. Sem ela, você arrisca treinar por tempo demais e destruir as capacidades que queria preservar. Com ela, o modelo automaticamente para em seu melhor ponto. O parâmetro “patience” (quantas avaliações sem melhoria antes de parar) é um dos hiperparâmetros mais importantes em fine-tuning.

Deep Dive

The process: (1) split your data into training and validation sets, (2) evaluate on the validation set periodically during training, (3) track the best validation metric (loss, accuracy, F1), (4) if the metric hasn't improved for N evaluations (patience), stop training and revert to the checkpoint with the best validation score. This prevents the model from memorizing training data beyond the point where it helps generalization.

In LLM Fine-Tuning

For LLM fine-tuning, early stopping is especially important because catastrophic forgetting can destroy base model capabilities. A model fine-tuned for too long on customer support data might become great at support but lose its ability to do math or write code. Monitoring validation loss across multiple task types (not just the fine-tuning task) helps catch this. Typical fine-tuning runs are 1–5 epochs with patience of 2–3 evaluations.

Not Used in Pre-Training

Interestingly, LLM pre-training rarely uses early stopping. The training runs are so expensive and the datasets so large that models typically train for a predetermined number of tokens (based on scaling laws). Overfitting is rare during pre-training because the model usually never sees the same data twice. Early stopping is primarily a fine-tuning and classical ML technique.

Conceitos relacionados

← Todos os termos
ESC