Yeda AI Tips · #045

A Model That Memorizes Isn't Smart

A fine-tuned model that repeats your training examples word for word looks impressive for about five seconds — right up until you realize it can't do anything else. That's not learning, it's memorizing, and it's one of the most common failure modes in fine-tuning: training for too many passes over the same data.

Why more epochs isn't automatically better

Each full pass over your training set is an epoch. Early epochs teach the model the general shape of the task — the style, the structure, the kind of answer you want. But keep running additional epochs on the same fixed dataset, and at some point the model stops learning the pattern and starts learning the specific examples. It's overfitting: your training loss keeps dropping, but that drop no longer reflects the model getting better at the task in general — it reflects the model getting better at recalling exactly what it already saw.

This is why chasing the lowest possible training loss is the wrong goal. Push training loss low enough and you're not measuring generalization anymore — you're measuring how well the model has memorized its training set, which tells you nothing about how it'll perform on a prompt it hasn't seen.

Reading the loss curve

The signal to watch for is shape, not just direction. A normal training run's loss drops quickly at first, then flattens out as the model converges on the general pattern. Overfitting shows up as a second act: the loss, having flattened or bottomed out, starts creeping back up (when measured on held-out data) or the training loss keeps falling while quality on anything outside the training set gets worse. That upward turn — or that growing gap between train and held-out performance — is your stop signal, not a number to push through.

How to actually do it

  1. Start with a small epoch budget. Most fine-tuning jobs only need roughly one to three epochs over the data — treat higher numbers as the exception, not the default.
  2. Checkpoint after every epoch. Don't wait until the run finishes to find out you overshot.
  3. Hold a slice of data out of training. Never train on 100% of what you have; keep enough aside to actually test generalization.
  4. Evaluate each checkpoint on the held-out data, not on examples the model has already memorized. A model that scores perfectly on its own training set tells you nothing.
  5. Stop at the checkpoint before performance on held-out data turns downward — that's usually one to three epochs, but the loss curve is the actual authority, not a fixed rule.
epochs: 1–3, checkpoint each

Pitfalls

Resources

Fine-tuning a model for production? Yeda AI designs, audits, and ships production LLM systems.

Talk to us · Read the blog