English

Asymmetry in Low-Rank Adapters of Foundation Models

Machine Learning 2024-02-28 v2

Abstract

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 BABA, we observe that the BB and AA matrices have distinct functions: AA extracts features from the input, while BB uses these features to create the desired output. Based on this observation, we demonstrate that fine-tuning BB is inherently more effective than fine-tuning AA, and that a random untrained AA 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 BB improves the bound. We support our conclusions with experiments on RoBERTa, BART-Large, LLaMA-2, and ViTs.

Keywords

Cite

@article{arxiv.2402.16842,
  title  = {Asymmetry in Low-Rank Adapters of Foundation Models},
  author = {Jiacheng Zhu and Kristjan Greenewald and Kimia Nadjahi and Haitz Sáez de Ocáriz Borde and Rickard Brüel Gabrielsson and Leshem Choshen and Marzyeh Ghassemi and Mikhail Yurochkin and Justin Solomon},
  journal= {arXiv preprint arXiv:2402.16842},
  year   = {2024}
}

Comments

17 pages, 2 figures, 9 tables