通过低秩分解识别关键权重,用5%参数实现更强推理能力。
LIFT the Veil for the Truth: Principal Weights Emerge after Rank Reduction for Reasoning-Focused Supervised Fine-Tuning
- 基于低秩近似后权重幅值筛选关键参数
- 仅更新5%主权重,推理性能超越全量微调
- 适合资源有限但需强推理的模型优化场景
近期研究表明,对大型语言模型进行少量高质量数据的监督微调可显著提升推理能力。然而,全量微调(Full FT)计算成本高,且在数据有限时易过拟合与灾难性遗忘。稀疏微调虽具效率优势,但在大模型时代因难以定位真正影响推理的关键参数而进展缓慢。本文提出:经低秩近似后幅值最大的权重为关键权重,称为主权重(Principal Weights)。令人意外的是,基于幅值的稀疏微调在大模型上表现不佳,但经低秩处理后效果大幅提升。据此提出低秩感知稀疏微调(LIFT),仅更新前5%主权重,在推理任务上持续优于全量微调,同时保持与主流参数高效微调方法相当的内存开销。在算术推理等目标领域表现优异,并比全量微调和LoRA多保留20%源域知识。代码已开源。
原文摘要 · Abstract (English)
Recent studies have shown that supervised fine-tuning of LLMs on a small number of high-quality datasets can yield strong reasoning capabilities. However, full fine-tuning (Full FT), while powerful, is computationally expensive and susceptible to overfitting and catastrophic forgetting, particularly when data is limited. Sparse fine-tuning, which previously achieved notable success by updating only a small subset of model parameters, offers a promising trade-off between efficiency and effectiveness. Yet, it has lagged behind in the LLM era due to the difficulty of identifying parameters truly critical for reasoning. In this work, we state that weights with the largest magnitude after low-rank approximation are critical weights for fine-tuning, which we call Principal Weights. Surprisingly, while magnitude-based sparse fine-tuning performs poorly as a baseline on LLM fine-tuning, it becomes highly effective after rank reduction. These insights motivate our method: Low-rank Informed Sparse Fine-Tuning (LIFT). LIFT only updates the top 5% Principal Weights throughout training and consistently achieves better performance on reasoning tasks than Full FT, while maintaining memory efficiency on par with popular parameter-efficient fine-tuning methods. In addition to strong performance on target domains such as arithmetic reasoning, LIFT also retains up to 20% more source-domain knowledge, compared to Full FT and LoRA. Our code is available at: https://github.com/zihanghliu/LIFT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。