arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.13618cs.CV
收藏

LoLDU:基于下三角-对角-上三角分解的低秩适配方法用于参数高效微调

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

University of Electronic Science and Technology of China(电子科技大学);Center for Future Media(未来媒体中心);School of Computer Science and Engineering(计算机科学与工程学院)

Yiming Shi, Jiwei Wei, Yujia Wu, Ran Ran, Chengwei Sun, Shiyuan He, Yang Yang

AI总结:

本文提出LoLDU参数高效微调方法,利用下三角-对角-上三角分解初始化低秩矩阵并优化对角矩阵,在保持性能的同时将可训练参数量减少2600倍,成为参数量最少的PEFT方法。

摘要:

模型规模的快速增长使得微调需要大量计算资源。现有方法如低秩适配(LoRA)试图解决全量微调中需要处理大量更新参数的问题。然而,LoRA利用低秩矩阵的随机初始化和优化来近似更新权重,这可能导致收敛效果不理想,并且与全量微调相比存在精度差距。为了解决这些问题,我们提出了LoLDU,一种参数高效微调(PEFT)方法,与常规PEFT方法相比,可训练参数量显著减少了2600倍,同时保持了可比的性能。LoLDU利用下三角-对角-上三角分解(LDU)来初始化低秩矩阵,以实现更快的收敛和正交性。我们重点优化对角矩阵以进行缩放变换。据我们所知,LoLDU在所有PEFT方法中参数量最少。我们在4个指令遵循数据集、6个自然语言理解(NLU)数据集、8个图像分类数据集以及图像生成数据集上,使用多种模型类型(LLaMA2、RoBERTa、ViT和Stable Diffusion)进行了广泛实验,提供了全面而详细的分析。我们的开源代码可通过\href{https://github.com/SKDDJ/LoLDU}{https://github.com/SKDDJ/LoLDU}访问。

英文摘要:

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}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑