arXiv:2410.13618cs.CV2024-10被引 4

LoLDU用分解法让微调参数减少2600倍,仍保持高精度。

LoLDU: Low-Rank Adaptation via Lower-Diag-Upper Decomposition for Parameter-Efficient Fine-Tuning

  • 用下-对角-上分解初始化低秩矩阵,加速收敛
  • 相比全量微调,性能差距小于1%(在多个数据集上)
  • 适合资源有限却要高性能微调的场景

模型规模激增导致微调需大量计算资源。现有方法如低秩适配(LoRA)通过随机初始化低秩矩阵逼近权重更新,但易导致收敛慢、精度下降。为此,我们提出LoLDU,一种参数高效微调方法,相较常规PEFT方法将可训练参数减少2600倍,同时保持相近性能。LoLDU采用下-对角-上分解(LDU)初始化低秩矩阵,提升收敛速度与正交性,并专注于优化对角矩阵以实现缩放变换。据我们所知,这是目前参数最少的PEFT方法。我们在4个指令遵循数据集、6个自然语言理解(NLU)数据集、8个图像分类数据集及图像生成任务中,针对LLaMA2、RoBERTa、ViT和Stable Diffusion等多种模型进行了全面实验,验证其有效性。代码已开源:https://github.com/SKDDJ/LoLDU。

原文摘要 · Abstract (English)

The rapid growth of model scale has necessitated substantial computational resources for fine-tuning. Existing approach such as Low-Rank Adaptation (LoRA) has sought to address the problem of handling the large updated parameters in full fine-tuning. However, LoRA utilize random initialization and optimization of low-rank matrices to approximate updated weights, which can result in suboptimal convergence and an accuracy gap compared to full fine-tuning. To address these issues, we propose LoLDU, a Parameter-Efficient Fine-Tuning (PEFT) approach that significantly reduces trainable parameters by 2600 times compared to regular PEFT methods while maintaining comparable performance. LoLDU leverages Lower-Diag-Upper Decomposition (LDU) to initialize low-rank matrices for faster convergence and orthogonality. We focus on optimizing the diagonal matrix for scaling transformations. To the best of our knowledge, LoLDU has the fewest parameters among all PEFT approaches. We conducted extensive experiments across 4 instruction-following datasets, 6 natural language understanding (NLU) datasets, 8 image classification datasets, and image generation datasets with multiple model types (LLaMA2, RoBERTa, ViT, and Stable Diffusion), providing a comprehensive and detailed analysis. Our open-source code can be accessed at \href{https://github.com/SKDDJ/LoLDU}{https://github.com/SKDDJ/LoLDU}.

参数效率微调低秩分解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。