SVFit: Parameter-Efficient Fine-Tuning of Large Pre-Trained Models Using Singular Values
Chengwei Sun, Jiwei Wei, Yujia Wu, Yiming Shi, Shiyuan He, Zeyu Ma, Ning Xie, Yang Yang
TL;DR
SVFit tackles the memory and efficiency barriers of fine-tuning large pre-trained models by leveraging singular value decomposition to initialize low-rank adapters. It decomposes each weight matrix as $W = W_r + W_e$, trains only the top-$r$ singular values in $W_r$ while freezing $W_e$ and the associated subspaces, enabling rapid domain adaptation with a drastically reduced parameter budget. Empirically, SVFit outperforms LoRA and PiSSA across natural language understanding, image classification, and DreamBooth tasks, achieving comparable or better performance with roughly 16× fewer trainable parameters. This approach offers practical gains for resource-constrained deployment and broad applicability to diverse downstream tasks, with potential for dynamic budget allocation and extension to more complex domains.
Abstract
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.
