Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Asymmetry in Low-Rank Adapters of Foundation Models

About

Parameter-efficient fine-tuning optimizes large, pre-trained foundation models by updating a subset of parameters; in this class, Low-Rank Adaptation (LoRA) is particularly effective. Inspired by an effort to investigate the different roles of LoRA matrices during fine-tuning, this paper characterizes and leverages unexpected asymmetry in the importance of low-rank adapter matrices. Specifically, when updating the parameter matrices of a neural network by adding a product $BA$, we observe that the $B$ and $A$ matrices have distinct functions: $A$ extracts features from the input, while $B$ uses these features to create the desired output. Based on this observation, we demonstrate that fine-tuning $B$ is inherently more effective than fine-tuning $A$, and that a random untrained $A$ should perform nearly as well as a fine-tuned one. Using an information-theoretic lens, we also bound the generalization of low-rank adapters, showing that the parameter savings of exclusively training $B$ improves the bound. We support our conclusions with experiments on RoBERTa, BART-Large, LLaMA-2, and ViTs.

Jiacheng Zhu, Kristjan Greenewald, Kimia Nadjahi, Haitz S\'aez de Oc\'ariz Borde, Rickard Br\"uel Gabrielsson, Leshem Choshen, Marzyeh Ghassemi, Mikhail Yurochkin, Justin Solomon• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationOffice-Home (test)
Mean Accuracy89
402
Image ClassificationOfficeHome--
161
Paraphrase DetectionMRPC
Empirical Error G(W)0.27
90
Relational ReasoningCLUTRR
Empirical Generalization Error G(W)1.04
60
Linguistic AcceptabilityCoLA (test)
Avg Accuracy69.3
56
Image ClassificationOfficeHome
Generalization Error (G(W))0.08
40
Linguistic AcceptabilityCOLA
Generalization Error Approximation (G(W))0.29
40
Paraphrase IdentificationPAWS
G(W) Error Approximation0.98
40
Question ClassificationTREC50
Generalization Error G(W)1.53
30
Image ClassificationCIFAR-10 (test)
Test Accuracy99
24
Showing 10 of 25 rows

Other info

Follow for update