atlas

Research map / Large language models / Language, safety and interpretability

When Scaling Meets LLM Finetuning: The Effect of Data, Model and Finetuning Method

ICLR 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 28 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.

Systematically studies scaling of LLM finetuning (full-model tuning, prompt tuning, LoRA) over model size, pretraining data, PET parameter size and finetuning data, proposing a multiplicative joint scaling law on bilingual 1B-16B LLMs for translation and summarization.

Problem

Understanding whether and how scaling factors (LLM model size, pretraining data size, PET parameter size, finetuning data size) affect the performance of different LLM finetuning methods, especially in the data-limited regime.

Approach

Proposes a multiplicative joint scaling law between finetuning data size and each other scaling factor (model size, pretraining data, PET parameters), treating finetuning data as the pivot factor, and compares FMT, prompt tuning and LoRA scaling behavior.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Scale LLM model size rather than pretraining data sizeLLM finetuning with FMT, Prompt, and LoRA · WMT14 En-De, WMT19 En-Zh, and MLSumPPL scaling exponent: α_m > α_pHelped
Increase PET parameter size by enlarging prompt length or LoRA rank (ablation)Prompt and LoRA · WMT14 En-De, WMT19 En-Zh, and MLSumfinetuning performance: |α_t| ≪ 1e-2No clear effect
Increase LoRA rank (ablation)LoRA · WMT14 En-Definetuning performance: -0.0017Hurt
Increase finetuning data sizeFMT · WMT14 En-De, WMT19 En-Zh, and MLSumfinetuning performanceHelped
Increase finetuning data sizeFMT compared with PET · WMT14 En-De, WMT19 En-Zh, and MLSumfinetuning performance: β: FMT 0.15, 0.14, 0.087; Prompt 0.051, 0.015, 0.025; LoRA 0.081, 0.025, 0.03Helped
Use LoRA instead of Prompt with more finetuning dataPET · WMT14 En-De, WMT19 En-Zh, and MLSumfinetuning performanceHelped
Use Prompt instead of LoRA with only a few thousand finetuning examplesPET · Translation and summarizationfinetuning performanceMixed
Use Prompt or LoRA rather than FMTFinetuned LLMs evaluated on related zero-shot translation · Flores200translation qualityHelped
Increase finetuning data for FMTFMT evaluated on few-shot generation · WMT14 En-De, WMT19 En-Zh, and MLSumfew-shot BLEURT/RougeLHurt
Use PET rather than FMTFinetuned LLMs evaluated on few-shot generation · WMT14 En-De, WMT19 En-Zh, and MLSumfew-shot capabilityHelped
Use the multiplicative rather than additive joint scaling lawLLM finetuning scaling prediction · WMT14 En-De, WMT19 En-Zh, and MLSumheld-out fitting error: WMT En-De avg: 0.0048 vs 0.0079; WMT En-Zh avg: 0.018 vs 0.022; MLSum avg: 0.014 vs 0.013Helped

Surprising findings

Limitations

Experimental setup

Datasets: WMT14 English-German WMT19 English-Chinese MLSum CNN/Daily-Mail Flores200

Benchmarks: WMT14 En-De WMT19 En-Zh MLSum Flores200 zero-shot translation

Metrics: token-level perplexity BLEURT RougeL mean absolute deviation

Baselines: full-model tuning prompt tuning LoRA

Builds on: LoRA prompt tuning Adafactor T5X

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