atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

LoRA+: Efficient Low Rank Adaptation of Large Models

ICML 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: LoRA variants and theory · 18 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.

Shows that standard LoRA uses the same learning rate for adapter matrices A and B, which is provably inefficient at large width, and proposes LoRA+ with a fixed larger learning rate ratio for B than A.

Problem

Standard LoRA finetuning sets the same learning rate for the two low-rank adapter matrices A and B, and there is no principled guidance for choosing the learning rate; at large width this prevents efficient feature learning.

Approach

Using infinite-width scaling arguments, the paper shows that efficient LoRA feature learning requires eta_B / eta_A = Theta(n) (eta_A ~ n^-1, eta_B ~ 1), and proposes LoRA+ which fixes a ratio lambda = eta_B/eta_A > 1 (empirically 2^1 to 2^4) and tunes only eta_A, keeping the cost of standard LoRA.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Set different learning rates for A and B, using a fixed ratioLoRA fine-tuning · Multiple language-model tasksperformance: 1%-2%Helped
Set different learning rates for A and B, using a fixed ratioLoRA fine-tuning · Multiple language-model tasksfinetuning speed: up to ∼2X SpeedUpHelped
Set η_B ≫ η_A instead of η_A = η_BLoRA fine-tuning · MMLUaccuracy: roughly 1.3%Helped
Use equal learning rates for A and BLoRA fine-tuning · MMLUaccuracy: roughly 1.3%Hurt
Use η_B ≫ η_A instead of η_A ≈ η_BRoBERTa-base and GPT-2 LoRA · MNLI and QQPtest accuracyHelped
Use η_B ≫ η_A instead of η_A ≈ η_BRoBERTa-base LoRA · SST2 and QNLItest accuracyMixed
Use η_B ≫ η_A instead of η_A = η_BLlama-7b LoRA · MNLItest accuracyNo clear effect
Use η_A = η_BAdam-trained LoRA · Infinite-width feature learningfeature-update efficiencyHurt
Set η_A = Θ(n^{-1}) and η_B = Θ(1)LoRA fine-tuning · Infinite-width feature learningfeature-update efficiencyHelped
Use Init 1 instead of Init 2 (ablation)Llama-7b LoRA fine-tuned on flan-v2 · MMLUaccuracy: 0.6%Helped
Use Init 1 instead of Init 2 (ablation)Llama-7b LoRA fine-tuned on flan-v2 · flan-v2optimal η_B / η_A ratio: twice as largeMixed
Set η_B > η_ALlama-7b LoRA fine-tuned on flan-v2 · flan-v2 training losstraining lossHurt

Surprising findings

Limitations

Experimental setup

Datasets: MNLI QQP SST2 QNLI flan-v2 synthetic sin dataset

Benchmarks: GLUE MMLU MNLI QQP SST2 QNLI

Metrics: test accuracy train loss test loss speedup

Baselines: LoRA with equal learning rates (eta_A = eta_B)

Builds on: LoRA Maximal Update Parametrization (muP) Adam Tensor Programs

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