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

DoRA: Weight-Decomposed Low-Rank Adaptation

About

Among the widely used parameter-efficient fine-tuning (PEFT) methods, LoRA and its variants have gained considerable popularity because of avoiding additional inference costs. However, there still often exists an accuracy gap between these methods and full fine-tuning (FT). In this work, we first introduce a novel weight decomposition analysis to investigate the inherent differences between FT and LoRA. Aiming to resemble the learning capacity of FT from the findings, we propose Weight-Decomposed Low-Rank Adaptation (DoRA). DoRA decomposes the pre-trained weight into two components, magnitude and direction, for fine-tuning, specifically employing LoRA for directional updates to efficiently minimize the number of trainable parameters. By employing \ours, we enhance both the learning capacity and training stability of LoRA while avoiding any additional inference overhead. \ours~consistently outperforms LoRA on fine-tuning LLaMA, LLaVA, and VL-BART on various downstream tasks, such as commonsense reasoning, visual instruction tuning, and image/video-text understanding. Code is available at https://github.com/NVlabs/DoRA.

Shih-Yang Liu, Chien-Yi Wang, Hongxu Yin, Pavlo Molchanov, Yu-Chiang Frank Wang, Kwang-Ting Cheng, Min-Hung Chen• 2024

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
Accuracy37.44
1891
Mathematical ReasoningGSM8K
Accuracy76.7
1362
Code GenerationHumanEval
Pass@119.75
1036
Question AnsweringARC Challenge
Accuracy50.8
906
Mathematical ReasoningGSM8K (test)
Accuracy81.2
900
Multi-task Language UnderstandingMMLU
Accuracy11.67
876
Image ClassificationCIFAR-100 (val)
Accuracy94.11
776
Mathematical ReasoningGSM8K (test)
Accuracy75.66
770
Commonsense ReasoningPIQA
Accuracy82.7
751
ReasoningBBH
Accuracy41.96
672
Showing 10 of 221 rows
...

Other info

Code

Follow for update