论文精选

训练 draft 模型在关键点被拒:用投机解码提升长程智能体推理

Adversarial Speculative Reasoning Improves Long Horizon Agents

精选理由

这篇论文把投机解码从单纯提速工具变成精度分配器,只训 0.06% 的参数就能提升长程智能体成功率,做 agent 的话值得看看思路。

arXiv 论文 2611.05345 指出投机解码在长程智能体任务中不只是加速手段:数值噪声会在验证窗口内累积,draft 模型的接受结果可在 30 步后改变模型输出分布,随机影响推理准确率。作者训练 draft 模型在关键决策点上偏好对抗性拒绝,使关键推理 token 改由 target 模型在低噪声条件下采样。该方法只训练 0.06% 至 0.28% 的托管参数,无需改动 target 模型或其输出分布,即可在复杂智能体编程基准上提高端到端成功率。

原文 · arXiv cs.AI

Adversarial Speculative Reasoning Improves Long Horizon Agents

Speculative decoding has been treated purely as a systems optimization that speeds up large language model decoding without changing the model behavior. Under this view, the growing effort to train better draft models has been focused solely on acceptance rate. We argue that this view does not hold true for long-horizon agent reasoning. Measuring frontier open-weight models on complex agentic tasks with different speculative decoding approaches, we found that numerical noise grows within verification windows, and the draft acceptance results of the thinking process can change the output distribution of the model after 30 steps, thus noticeably influencing the reasoning accuracy of long-horizon agents in random directions. In this work, we improve the reasoning capability of target models by controlling such noise. While we cannot change the overall token distribution verified by the target model, we can encode critical reasoning with highest precision by training the draft model to be rejected. We identify these points in agent traces and train the draft model to favor adversarial preference. At inference the trained draft and the target part ways precisely at these decision points, verification restarts, so that critical thinking tokens are sampled from the target model and encoded under its least noisy conditions. This approach makes critical reasoning better indexed and more durable over a long horizon. On complex agentic coding benchmarks, our approach raises end-to-end success rate with no change to the target model or its output distribution. Our results recast the speculative decoding mechanism from a pure accelerator into an active allocator for context precision that improves the accuracy of long-horizon agents by training 0.06% to 0.28% of hosted parameters.