arXiv:2605.31494cs.CLcs.LG2026-05

用可部署模型替代多轮推理,提升大模型后训练效果

Consolidating Rewarded Perturbations for LLM Post-Training

论文配图:Consolidating Rewarded Perturbations for LLM Post-Training
图 1 · 摘自论文原文
  • 将奖励过的扰动合并为单一更新,无需梯度
  • 平均提升8.1分,仅需1次前向传播
  • 适合追求高效推理的部署场景

语言模型后训练通常采用样本-评分-更新的循环,由梯度下降实现。近期工作如RandOpt将该循环移至权重空间,对预训练模型采样高斯扰动,并在推理时集成表现最优的K个专家。尽管在相同计算量下性能可比PPO和GRPO,但这种预测级集成需对每个测试样本进行K次前向传播,且难以扩展至自由生成任务。本文探讨是否可将受奖励的扰动群体整合为单一可部署模型,以替换推理时的集成。对25个模型-任务组合的半数分析揭示了稳定的低秩结构。基于此几何特性,提出CoRP(Consolidating Rewarded Perturbations),一种无梯度算子,结合奖励加权聚合、兼容性感知重加权与保留验证门控机制,不经过语言模型梯度。在五个0.5B至8B参数的语言模型上,覆盖数学、代码和创意写作五项任务,CoRP平均提升基线模型8.1分。仅使用十分之一的RandOpt扰动预算,其性能超过单次前向的RandOpt达6.5分,且恢复了50次多数投票集成超过一半的增益,同时仅需一次前向传播。

原文摘要 · Abstract (English)

Post-training of language models is commonly framed as a sample-score-update loop implemented by gradient descent. A recent line of work, exemplified by RandOpt, relocates this loop to weight space, sampling Gaussian perturbations around a pretrained model and ensembling the top-K rewarded specialists at inference. While competitive with PPO and GRPO under matched training compute, this prediction-level ensemble incurs K forward passes per test example and does not extend cleanly to free-form generation. We ask whether the rewarded population can instead be folded into a single deployable model, replacing the inference-time ensemble with one consolidated update. A split-half analysis over 25 model-task pairs reveals reproducible low-rank structure in every case. We turn this geometry into CoRP (Consolidating Rewarded Perturbations), a gradient-free operator that combines reward-weighted aggregation, compatibility-aware reweighting, and a held-out validation gate, with no gradient flowing through the language model. Across five language models from 0.5B to 8B and five tasks covering math, code, and creative writing, CoRP improves the base model by 8.1 points on average. Using one tenth of RandOpt's perturbation budget, CoRP exceeds single-inference RandOpt by 6.5 points and recovers more than half of the gain of the 50-pass majority-vote ensemble, at one forward pass per test example.

后训练模型集成高效推理

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