通过稀疏频域变换,仅用0.08%参数实现高效微调。
S2FT: Parameter-Efficient Fine-Tuning in Sparse Spectrum Domain

- 将权重变化转换至稀疏频域,仅微调少量频谱系数。
- 在多个任务上仅用0.08%参数即达优异性能。
- 适合资源受限场景下的大模型快速适配。
参数高效微调(PEFT)通过仅更新少量参数,实现大模型在下游任务上的适配。近期基于傅里叶变换的方法通过仅微调少数频谱系数进一步降低参数量,其假设权重变化δW在空间域具有稀疏频谱。然而本文发现,权重变化的频谱实际呈幂均匀分布,表明仅微调少量频谱系数不足以准确建模。为此,我们提出S2FT:寻找一个可逆变换,将潜在的空间域稀疏频谱矩阵映射为真实权重变化,并在该稀疏频域进行微调。首先预估粗略权重变化作为先验;鉴于稀疏频谱常对应局部平滑结构,我们将变换视为对预估权重变化的行列重排操作,以保留神经元结构信息的同时实现平滑。最后采用最近邻搜索求解重排问题,获得可逆变换。大量实验表明,S2FT仅使用0.08%训练参数即取得优越性能。
原文摘要 · Abstract (English)
Parameter Efficient Fine-Tuning (PEFT) is a key technique for adapting a large pretrained model to downstream tasks by fine-tuning only a small number of parameters. Recent methods based on Fourier transforms have further reduced the fine-tuned parameters scale by only fine-tuning a few spectral coefficients. Its basic assumption is that the weight change δW is a spatial-domain matrix with a sparse spectrum. However, in this paper, we observe that the spectrum of weight change is not sparse, but instead distributed like power-uniform. This fact implies that fine-tuning only a few spectral coefficients is insufficient to accurately model the weight change with uniform spectrum. To address this issue, we propose to seek an invertible transformation that can transform a latent spatial-domain matrix with sparse spectrum to the weight change, and then perform PEFT on such sparse spectrum domain with few spectral coefficients, called S2FT. To seek such transformation, we first pre-estimate a coarse weight change as a prior. Then, inspired by that sparse spectrum often correspond to locally smooth spatial structures, we regard this transformation as a row and column rearrangement operation on the pre-estimated weight change that smooth spatial structures while keep the structure information of neurons. Finally, we propose to solve the rearrangement search problem in a simple nearest neighbor search manner, thereby obtaining the invertible transformation. Extensive results show our S2FT achieves superior performance by only using 0.08% training parameters.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。