Fine-Tuning

Fine-tuning adapts a pretrained model to a target task by continuing training on task data, either updating all weights or a small subset (parameter-efficient fine-tuning).

The basic objective is supervised training initialized from pretrained weights:

Trainable Scope

Fine-tuning is not one protocol. The trainable parameter set can vary:

Common choices include:

  • Head-only tuning.
  • Last-layer or last-block tuning.
  • Full-model tuning.
  • Adapter tuning.
  • Low-rank adaptation.
  • Prompt or prefix tuning.

The optimization problem is:

Reporting fine-tuning without trainable scope is underspecified.

Learning Rate and Forgetting

Fine-tuning changes the pretrained representation. A large update can improve target performance while degrading broad capabilities:

Catastrophic forgetting risk grows when target data is small, narrow, or distribution-shifted from pretraining:

This is only a heuristic, but it captures the habit: small target sets usually need conservative learning rates, regularization, early stopping, or parameter-efficient updates.

Selection Rule

The final checkpoint should be selected by validation data:

Then the test set is used once:

If test performance guides checkpoint selection, the reported result is no longer a clean test estimate.

Why It Matters

  • The standard way to specialize a general pretrained model.
  • Parameter-efficient variants cut compute and storage per task.
  • Sensitive to learning rate, data size, and catastrophic forgetting.
  • Strong results can come from pretraining, adaptation budget, or validation leakage, so the protocol must be explicit.

Failure Modes

  • Comparing methods with different tuning budgets.
  • Reporting the best seed without a fixed seed policy.
  • Tuning preprocessing or threshold choices on test data.
  • Freezing the encoder for one model but fully tuning another without stating parameter counts.
  • Forgetting general capability while improving a narrow validation set.
  • Overfitting a small target dataset after many hyperparameter trials.

Checks

  • Full fine-tuning vs. adapter/low-rank methods for the data budget?
  • Is the model overfitting a small target set?
  • Did adaptation degrade general capabilities learned in pretraining?
  • Is the adaptation protocol recorded with trainable scope, validation rule, search budget, and final test boundary?
  • Are trainable parameter count, optimizer, schedule, seed policy, and early stopping rule reported?
  • Is the target split independent from pretraining data, source data, and model-selection choices?
  • Is performance compared to linear probing and a simple supervised baseline?