arXiv:2511.10707cs.LGcs.AI2025-11中稿 · AAAI被引 2

改进推理前缀生成,让轻量微调更擅长数学推理

Bias-Restrained Prefix Representation Finetuning for Mathematical Reasoning

  • 通过截断训练数据优化初始推理前缀生成
  • 干预早期推理阶段防止错误累积,提升数学推理准确率
  • 约束干预向量幅度,避免干扰数值编码,适合数学任务

参数高效微调(PEFT)通过更新少量参数提升下游任务性能。表示微调(ReFT)进一步提高效率,冻结模型权重并优化内部表示,参数量少于PEFT,且在多个任务上表现更优。然而,ReFT在数学推理任务上表现显著下降。本文发现,ReFT在数学任务中表现差主要源于其在推理初期难以生成有效推理前缀,且在思维链(CoT)阶段扰乱数值编码并导致误差累积。为此,提出偏置受限的前缀表示微调(BREP ReFT),通过截断训练数据优化初始推理前缀生成,干预早期推理阶段防止误差积累,并约束干预向量幅度以避免干扰数值编码。大量实验表明,BREP在多种模型架构下均优于标准ReFT和基于权重的PEFT方法,在数学推理任务上展现出更强的有效性、效率与鲁棒泛化能力。源代码见 https://github.com/LiangThree/BREP。

原文摘要 · Abstract (English)

Parameter-Efficient finetuning (PEFT) enhances model performance on downstream tasks by updating a minimal subset of parameters. Representation finetuning (ReFT) methods further improve efficiency by freezing model weights and optimizing internal representations with fewer parameters than PEFT, outperforming PEFT on several tasks. However, ReFT exhibits a significant performance decline on mathematical reasoning tasks. To address this problem, the paper demonstrates that ReFT's poor performance on mathematical tasks primarily stems from its struggle to generate effective reasoning prefixes during the early inference phase. Moreover, ReFT disturbs the numerical encoding and the error accumulats during the CoT stage. Based on these observations, this paper proposes Bias-REstrained Prefix Representation FineTuning (BREP ReFT), which enhances ReFT's mathematical reasoning capability by truncating training data to optimize the generation of initial reasoning prefixes, intervening on the early inference stage to prevent error accumulation, and constraining the intervention vectors' magnitude to avoid disturbing numerical encoding. Extensive experiments across diverse model architectures demonstrate BREP's superior effectiveness, efficiency, and robust generalization capability, outperforming both standard ReFT and weight-based PEFT methods on the task of mathematical reasoning. The source code is available at https://github.com/LiangThree/BREP.

数学推理轻量微调表示微调

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。