Large neural networks are typically fine-tuned on new tasks, but this can cause them to forget what they learned during pretraining. Also, having separate models for each task takes up a lot of storage. LoRA is a popular method that adapts a model by training only a small number of extra parameters, and it has been observed to both resist forgetting and allow merging of adapters from different tasks into one model. This paper provides a theoretical explanation for these properties using the concept of 'margin', a measure of how confidently a model classifies data. The authors analyze a simplified version of fine-tuning and show that, for a certain range of regularization, LoRA adapters naturally find the solution that maximizes the margin on the new task. This finding leads to a simple formula for how to combine adapters, called MaLoRA, which matches or beats existing merging methods. The main result is that there is a direct trade-off controlled by a single parameter: a lower value gives better adaptation but more forgetting, and the paper identifies the best trade-off point.
Theory explains why low-rank adapters resist forgetting and merge well
A margin-based analysis of LoRA reveals a precise trade-off between retaining pretrained knowledge and adapting to new tasks, and yields a simple training-free merging rule.
Top University
Ziqing Xu · Hancheng Min · Lachlan Ewen MacDonald · Salma Tarmoun · Enrique Mallada · Weijie Su · +1 more
University of Pennsylvania · Shanghai Jiao Tong University · Johns Hopkins University
Research Digest··3 min read

In one line
LoRA's robustness to forgetting and adapter merging stem from its optimization aligning with max-margin solutions.
Before you read this
An idea this paper builds onRead it as
The authors propose a unified theoretical explanation for two surprising properties of low-rank adaptation (LoRA): its robustness to catastrophic forgetting and the ability to merge independently trained adapters.
The authors studied LoRA fine-tuning in a simplified setting: multiclass linear classification with a pretrained model, where fine-tuning data for each new task is nearly orthogonal to both the pretrained data and other tasks' data.
How it works
stage by stageThe Daily Edition
The front page in your inbox each weekday morning: the lead, the picks, and why each one earned its place.