Skip to content
ML Academy

Neural networks, deep learning, NLP · insight-1435

Do not stop at the first rise in validation loss without checking a longer horizon

Evidence type
empirical finding
Difficulty
hard
Roles
ML engineer, Data scientist
Industries
cross-industry

Summary: situation, action, reason, boundary

  1. Situation

    An over-parameterised network's validation error rises in mid-training, then, with patience set to five epochs, training halts and a smaller model is chosen instead.

  2. Action

    Run one or two reference jobs with patience disabled to the full schedule, compare them with the early-stopped model on the same validation set, and set patience from what you see.

  3. Reason

    Nakkiran and colleagues report that test performance can worsen and then improve as model size or training time increases, so early stopping can end a run before a better region.

  4. Boundary

    Large, over-parameterised networks trained with noisy labels or to near-zero training loss.

Symptom or risk

In some regimes error first rises and then falls again with more training or larger models, so a short patience rule can lock in a worse optimum.

Exceptions

Small, well-regularised models where the curve is a clean U shape.

Trade-offs

Extra full-length runs; justify them by the stakes of the decision.

Pitfalls

Treating the effect as a licence to train indefinitely when the final epochs are simply overfitting.

Lessons and concepts

Claims and their sources