atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

LoRA Training in the NTK Regime has No Spurious Local Minima

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

Theoretically analyzes LoRA fine-tuning in the NTK regime, showing a low-rank full fine-tuning solution exists (rank r with r(r+1)/2 ≤ KN), that LoRA with r(r+1)/2 > KN has no spurious local minima under an ε-perturbation, and that the resulting low-rank solution generalizes via Rademacher bounds.

Problem

Theoretical understanding of why LoRA fine-tuning is trainable (finds good solutions despite nonconvex quadratic parameterization) and generalizes, which prior work has not addressed.

Approach

Treats LoRA's factorization as a Burer–Monteiro-style quadratic parameterization of a nuclear-norm-regularized linearized (NTK) problem; shows a low-rank minimizer exists via dimension counting, and uses a generic positive semidefinite perturbation plus Sard's theorem to show that rank r with r(r+1)/2 > KN yields no spurious SOSPs, so (S)GD converges to near-global low-rank solutions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use higher LoRA rank (ablation)LoRA fine-tuning of linearized RoBERTa-base · SST-2, MR, CR, QNLI, Subj, QQPtraining convergence rateHelped
Replace LoRA with full fine-tuningLinearized RoBERTa-base fine-tuning · SST-2, MR, CR, QNLI, Subj, QQPtraining convergence rateHelped
Vary LoRA rank or use full fine-tuning instead of LoRA (ablation)Linearized pretrained models · NLP tasks; Bean disease dataset; SUPERBfinal training lossNo clear effect
Use LoRA rank ≥11Linearized RoBERTa-base · SST-2, MR, CR, QNLI, Subj, QQPtraining lossNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: SST-2 MR CR QNLI Subj QQP bean disease dataset SUPERB

Benchmarks: SST-2 MR CR QNLI Subj QQP bean disease SUPERB speech

Metrics: training loss vs epochs test accuracy (appendix)

Baselines: full fine-tuning (without LoRA) on linearized model

Builds on: neural tangent kernel Burer-Monteiro factorization LoRA prompt-based fine-tuning (LM-BFF) Ge et al. perturbed matrix factorization landscape results

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