arXiv:2606.23595cs.AI2026-06被引 1

让大模型在推理时并行生成多个思考路径并整合,显著提升推理效率和准确率。

SPIRAL: Learning to Search and Aggregate

论文配图:SPIRAL: Learning to Search and Aggregate
图 1 · 摘自论文原文
  • 通过并行生成多条思维链,再统一聚合结果,实现端到端优化。
  • 在推理计算资源增加时,性能比GRPO高15%,效率提升最高达11倍。
  • 适合需要高精度推理的场景,如复杂逻辑题、数学问题求解。

语言模型的推理能力可通过扩展推理计算资源来显著提升,包括单条思维链的逐步推理、独立并行生成多条推理路径,以及将多条路径聚合为最终答案。然而,后训练阶段的语言模型仅针对单条路径的逐步推理进行优化。我们提出序列-并行-聚合强化学习(SPIRAL)框架,使语言模型在统一推理计算流程中同时使用三种计算原语。具体而言,模型先并行采样多个独立的思维链(每条通过逐步链式思考生成),再基于这些路径生成最终聚合路径;所有组件均以最终聚合响应的奖励为目标进行端到端优化。训练采用集合强化学习,指导模型生成对聚合器有用的路径集合;并结合标准强化学习,指导模型将路径集合并生成更优的最终回答。实验表明,当三类计算原语均被扩展时,SPIRAL能有效随推理计算规模增长,相比GRPO在推理效率上最高提升11倍,性能高出15%。

原文摘要 · Abstract (English)

Language model reasoning can be substantially improved at test time via scaffolds that scale inference compute across different primitives -- sequential reasoning within a trace, independently sampled parallel traces, and aggregation of multiple reasoning traces into a final response. During post-training, however, language models are optimized only for sequential reasoning within a single trace. We introduce Sequential-Parallel-Aggregative Reinforcement Learning (SPIRAL), a framework in which a language model is trained to use all three primitives, as part of a unified inference compute pipeline. Concretely, the language model first samples a set of independent traces in parallel, each produced through sequential chain-of-thought reasoning, and then generates a final aggregation trace conditioned on those traces; all components are optimized end-to-end against the reward of the final aggregated response. To train this system, SPIRAL uses set reinforcement learning to teach models to produce a set of traces that are collectively useful for an aggregator and standard reinforcement learning to teach models to aggregate the set into improved final responses. Our experiments on reasoning tasks show that SPIRAL effectively scales with inference compute, outperforming GRPO by up to 11$\times$ scaling efficiency and 15% higher performance when all three compute primitives are scaled.

推理增强强化学习多路径聚合思维链

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