atlas

Research map / Large language models / Instruction and fine-tuning

Selecting Large Language Model to Fine-tune via Rectified Scaling Law

ICML 2024 · Large language models › Instruction and fine-tuning · approach: Efficient LLM fine-tuning systems · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes a Rectified Scaling Law with a pre-learned data size term that fits the two-phase fine-tuning loss curve, and an Accept-then-Stop algorithm that uses it to select the best LLM to fine-tune at hundreds of times lower cost.

Problem

Selecting the best pre-trained LLM to fine-tune under resource constraints, without fully fine-tuning every candidate; existing scaling laws fail to capture the fine-tuning loss curve, which has a pre-power phase at small data sizes.

Approach

Discovers a 'pre-power phase' in low-data fine-tuning scaling curves and adds a pre-learned data size term D_l to the scaling law (B/(D_l+D^beta)+E), then uses a stop-when-phase-changes rule to extrapolate the power-phase line from small subsets to the full dataset for model ranking.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use AtS instead of the strongest listed selection baseline, SubTuningLLM selection · FLAN (average across budget ratios)PearCorr (%): 62.7 vs 33.5Helped
Use AtS instead of the strongest listed selection baseline, SubTuningLLM selection · WMT19 (average across budget ratios)PearCorr (%): 84.6 vs 63.4Helped
Use AtS instead of the strongest listed selection baseline, SubTuningLLM selection · Gigaword (average across budget ratios)PearCorr (%): 93.8 vs 79.9Helped
Use the Rectified Scaling Law instead of the vanilla scaling lawfine-tuning loss prediction · Fine-tuning test-loss fit across 30 models and three datasetsRMSD: 0.036 -> 0.007Helped
Use AtS rather than fitting the Rectified Scaling Law to all observed points (OurFit)LLM selection · FLAN (γ=1/512)PearCorr (%): 36.8 -> 45.6Helped
Use AtS rather than fitting the vanilla law to all observed points (VanillaFit)LLM selection · FLAN (γ=1/512)PearCorr (%): 20.7 -> 45.6Helped
Use AtS rather than fitting the Rectified Scaling Law to all observed points (OurFit)LLM selection · Gigaword (γ=1/512)PearCorr (%): 78.5 -> 91.0Helped
Vary AtS outlier tolerance δ and minimum accepted-pair count k (ablation)AtS · Gigaword (γ=1/512)PearCorr (%)No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: FLAN WMT19 En-Zh Gigaword

Benchmarks: FLAN WMT19 En-Zh Gigaword

Metrics: PearCorr RelAcc RMSD

Baselines: ModelSize ZeroShot SubTuning VanillaFit OurFit

Builds on: Scaling Laws for Neural Language Models Broken neural scaling laws Chinchilla

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy