让微调更聪明:按输入重要性动态调整更新强度
GateRA: Token-Aware Modulation for Parameter-Efficient Fine-Tuning
- 引入自适应门控机制,根据每个标记的重要性动态调节微调力度
- 在常识推理任务上显著优于或持平现有轻量微调方法
- 适合需要高效精准微调的场景,尤其对生成式模型有帮助
参数高效微调(PEFT)方法如LoRA、DoRA和HiRA通过低秩更新实现大模型的轻量级适配。然而,现有方法对所有标记施加静态、与输入无关的更新,忽视了不同输入的重要性和难易差异,导致在自回归生成中对简单内容过拟合、对关键区域适配不足。本文提出GateRA,一个统一框架,引入标记感知调制,动态调节PEFT更新强度。通过在标准PEFT分支中加入自适应门控,实现选择性、标记级别的适配,保留对已良好建模输入的知识,将能力集中于困难样本。实验显示,该方法在预填充阶段自动抑制冗余标记的更新,在解码阶段强化适应。进一步引入基于熵的正则化,促使近二值化的门控决策,避免扩散式更新,实现可解释、稀疏的适配且无需硬阈值。理论分析表明,GateRA在PEFT路径上产生软梯度屏蔽效应,实现连续可微的适配控制。在多个常识推理基准测试中,GateRA持续优于或匹配现有PEFT方法。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning (PEFT) methods, such as LoRA, DoRA, and HiRA, enable lightweight adaptation of large pre-trained models via low-rank updates. However, existing PEFT approaches apply static, input-agnostic updates to all tokens, disregarding the varying importance and difficulty of different inputs. This uniform treatment can lead to overfitting on trivial content or under-adaptation on more informative regions, especially in autoregressive settings with distinct prefill and decoding dynamics. In this paper, we propose GateRA, a unified framework that introduces token-aware modulation to dynamically adjust the strength of PEFT updates. By incorporating adaptive gating into standard PEFT branches, GateRA enables selective, token-level adaptation, preserving pre-trained knowledge for well-modeled inputs while focusing capacity on challenging cases. Empirical visualizations reveal phase-sensitive behaviors, where GateRA automatically suppresses updates for redundant prefill tokens while emphasizing adaptation during decoding. To promote confident and efficient modulation, we further introduce an entropy-based regularization that encourages near-binary gating decisions. This regularization prevents diffuse update patterns and leads to interpretable, sparse adaptation without hard thresholding. Finally, we present a theoretical analysis showing that GateRA induces a soft gradient-masking effect over the PEFT path, enabling continuous and differentiable control over adaptation. Experiments on multiple commonsense reasoning benchmarks demonstrate that GateRA consistently outperforms or matches prior PEFT methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。