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?