用奇异值初始化低秩矩阵,让大模型微调更高效
SVFit: Parameter-Efficient Fine-Tuning of Large Pre-Trained Models Using Singular Values
- 用SVD提取关键奇异值作为可训练参数
- 仅需LoRA 1/16参数量,性能却更优
- 适合资源受限环境下的模型快速适配
大规模预训练模型在自然语言处理和计算机视觉任务中表现优异,但全量微调面临巨大显存压力。参数高效微调(PEFT)方法如LoRA通过只调整少量参数缓解此问题,但通常对低秩矩阵采用随机初始化,导致梯度下降效率低且泛化能力差。为此,我们提出SVFit,一种基于奇异值分解(SVD)的新型PEFT方法:对预训练权重矩阵进行SVD,保留能捕捉99%以上信息的前r个关键奇异值作为可训练参数,用于缩放矩阵的基本子空间,实现快速领域适应。在自然语言理解、文本生成图像及图像分类等多个任务上,实验表明SVFit显著优于LoRA,同时仅需16倍更少的可训练参数。
原文摘要 · Abstract (English)
Large pre-trained models (LPMs) have demonstrated exceptional performance in diverse natural language processing and computer vision tasks. However, fully fine-tuning these models poses substantial memory challenges, particularly in resource-constrained environments. Parameter-efficient fine-tuning (PEFT) methods, such as LoRA, mitigate this issue by adjusting only a small subset of parameters. Nevertheless, these methods typically employ random initialization for low-rank matrices, which can lead to inefficiencies in gradient descent and diminished generalizability due to suboptimal starting points. To address these limitations, we propose SVFit, a novel PEFT approach that leverages singular value decomposition (SVD) to initialize low-rank matrices using critical singular values as trainable parameters. Specifically, SVFit performs SVD on the pre-trained weight matrix to obtain the best rank-r approximation matrix, emphasizing the most critical singular values that capture over 99% of the matrix's information. These top-r singular values are then used as trainable parameters to scale the fundamental subspaces of the matrix, facilitating rapid domain adaptation. Extensive experiments across various pre-trained models in natural language understanding, text-to-image generation, and image classification tasks reveal that SVFit outperforms LoRA while requiring 16 times fewer trainable parameters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。