DoRA: Weight-Decomposed Low-Rank Adaptation

在广泛使用的参数高效微调(PEFT)方法中,由于避免了额外的推断成本,LoRA及其变体已经获得了相当大的流行度。然而,这些方法与完全微调(FT)之间仍然存在精度差距。在本研究中,我们首先引入了一种新颖的权重分解分析来研究FT和LoRA之间的内在差异。为了模拟FT的学习能力,我们提出了一种称为Weight-Decomposed LowRank Adaptation(DoRA)的方法。DoRA将预训练权重分解为两个组成部分:大小和方向,用于微调,具体使用LoRA进行方向更新,以有效地最小化可训练参数的数量。通过采用DoRA,我们增强了LoRA的学习能力和训练稳定性,同时避免了任何额外的推断开销。DoRA在各种下游任务上(如常识推理、视觉指令调整和图像/视频文本理解)上对LLaMA、LLaVA和VL-BART进行微调,始终优于LoRA。
Among the widely used parameter-efficient finetuning (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 LowRank 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 DoRA, we enhance both the learning capacity and training stability of LoRA while avoiding any additional inference overhead. DoRA 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.
许愿