atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

Sparse is Enough in Fine-tuning Pre-trained Large Language Models

ICML 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Multilingual language representation pretraining · 1 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 SIFT, a gradient-based sparse fine-tuning method that updates only the top-x% gradient components (selected from the first batch) of pre-trained weights, motivated by a PAC-Bayes prior-shift analysis of loss landscapes and gradient quasi-sparsity.

Problem

Why pre-trained language models have low intrinsic dimension for fine-tuning, and how to fine-tune them efficiently without adding extra modules.

Approach

Views pre-training as a shift of the PAC-Bayes prior that yields quasi-sparse gradients; fine-tunes only the top-x% gradient-magnitude components of the pre-trained weights (sparse increment), with a hook-based memory-efficient implementation and no added modules.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use gradient-based component selection instead of random selection for Wq, Wk, Wv, Wo updates (ablation)SIFT on RoBERTa-Large · RTEaccuracy: 83.8 -> 85.9Helped
Use gradient-based component selection instead of random selection for Wq, Wk, Wv, Wo updates (ablation)SIFT on RoBERTa-Large · MRPCMatthew’s correlation: 89.5 -> 90.4Helped
Select sparse components in Wv only rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · RTEaccuracy: 85.9 -> 85.9No clear effect
Select sparse components in Wv only rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · MRPCMatthew’s correlation: 90.4 -> 90.4No clear effect
Select sparse components in Wq and Wv rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · RTEaccuracy: 85.9 -> 85.2Hurt
Select sparse components in Wq and Wv rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · MRPCMatthew’s correlation: 90.4 -> 90.2Hurt
Select sparse components in Wq, Wk, Wv rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · RTEaccuracy: 85.9 -> 84.8Hurt
Select sparse components in Wq, Wk, Wv rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · MRPCMatthew’s correlation: 90.4 -> 88.7Hurt
Select sparse components across all attention parameters rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · RTEaccuracy: 85.9 -> 86.6Helped
Select sparse components across all attention parameters rather than Wq, Wk, Wv, Wo (ablation)SIFT on RoBERTa-Large · MRPCMatthew’s correlation: 90.4 -> 89.0Hurt
Apply sparse updates to a smaller share of parameters by varying the sparsity budget (ablation)SIFT on RoBERTa-Large · MRPCperformanceMixed

Surprising findings

Limitations

Experimental setup

Datasets: GLUE Alpaca

Benchmarks: GLUE MMLU HumanEval

Metrics: accuracy Matthews correlation Pearson correlation pass@1 pass@10

Baselines: Full fine-tuning Adapter (Adpt^H) Adapter (Adpt^P) LoRA Random selection Vanilla Llama

Builds on: LoRA Adapter Prefix-Tuning Bias-only fine-tuning PAC-Bayes bounds

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