LARGO提升小模型微调鲁棒性,低计算开销下抗领域偏移。
LARGO: Low-Rank Regulated Gradient Projection for Robust Parameter Efficient Fine-Tuning
- 用动态梯度投影约束低秩更新,保持各层独立性。
- 在多个数据集上实现最优的域内与域外性能平衡。
- 适合需要高效且稳定微调的大模型应用者。
参数高效微调方法虽大幅降低了大规模预训练模型适配下游任务的计算负担,但现有方法在领域偏移下常难以保持稳健表现。为此,我们提出低秩调节梯度投影(LARGO)算法,将动态约束引入低秩适配方法。LARGO通过并行可训练的梯度投影,动态调控层间更新,在保留预训练模型的分布外鲁棒性的同时,维持层间独立性;并通过减少层间梯度依赖,确保计算效率。此外,利用预训练权重的奇异值分解进行结构化初始化,最小化对预训练知识的偏离。在多种基准上的大量实验表明,LARGO在域内与域外场景均达到当前最优性能,相比现有参数高效微调方法显著降低计算开销。代码即将开源。
原文摘要 · Abstract (English)
The advent of parameter-efficient fine-tuning methods has significantly reduced the computational burden of adapting large-scale pretrained models to diverse downstream tasks. However, existing approaches often struggle to achieve robust performance under domain shifts while maintaining computational efficiency. To address this challenge, we propose Low-rAnk Regulated Gradient Projection (LARGO) algorithm that integrates dynamic constraints into low-rank adaptation methods. Specifically, LARGO incorporates parallel trainable gradient projections to dynamically regulate layer-wise updates, retaining the Out-Of-Distribution robustness of pretrained model while preserving inter-layer independence. Additionally, it ensures computational efficiency by mitigating the influence of gradient dependencies across layers during weight updates. Besides, through leveraging singular value decomposition of pretrained weights for structured initialization, we incorporate an SVD-based initialization strategy that minimizing deviation from pretrained knowledge. Through extensive experiments on diverse benchmarks, LARGO achieves state-of-the-art performance across in-domain and out-of-distribution scenarios, demonstrating improved robustness under domain shifts with significantly lower computational overhead compared to existing PEFT methods. The source code will be released soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。