让大模型推理线程直接通信,提速且省资源。
Message Passing Enables Efficient Reasoning
- 引入消息传递机制,线程间轻量通信避免重复上下文
- 支持提前终止无望分支,3-SAT任务效率显著提升
- 适配现有大模型,长文本问答表现媲美主流并行方法
尽管推理时扩展提升了大语言模型的推理能力,但生成长链条思维(CoT)仍是计算瓶颈。与序列式扩展不同,近年并行扩展采用分叉-合并(FJ)原语将任务分派给多个模型线程,但线程通常临时存在且无法点对点通信,限制了可扩展性。为此,我们提出消息传递语言模型(MPLM),通过轻量级发送/接收原语实现线程间直接通信。MPLM通过两个关键机制实现高效扩展:(1)降低通信开销,通过避免冗余上下文共享;(2)支持抢占,使线程能基于同伴部分信息提前终止。我们在三类任务上验证其潜力:在数独问题中,MPLM所需上下文渐近小于串行CoT和并行FJ;微调单模型可解25×25数独,仍挑战传统CoT与FJ及无工具前沿模型。在3-SAT问题中,抢占能力可终止无望分支,提升效率。最后,经适当提示的大预训练模型可遵循MPLM协议,在长上下文问答任务中表现媲美主流分叉-合并方法。
原文摘要 · Abstract (English)
While inference-time scaling has improved the reasoning abilities of large language models (LLMs), the need to generate long chains-of-thought (CoTs) is a computational bottleneck. Thus, in contrast to sequential scaling methods like CoT, recent parallel scaling techniques instead use fork and join (FJ) primitives to divide work across multiple LLM threads. However, in the fork-join paradigm, threads are typically transient and do not communicate pointwise with one another which limits scalability. To tackle this, we introduce Message Passing Language Models (MPLMs), a framework for LLM reasoning in which threads communicate directly via lightweight send and receive primitives. MPLMs enable efficient scaling through two key mechanisms: (1) reduced communication costs, achieved by avoiding redundant context sharing, and (2) preemption, which allows threads to terminate early based on partial information from their peers. We demonstrate the promise of MPLMs on 3 classes of tasks. First, on Sudoku puzzles, we show that MPLMs require an asymptotically smaller context than both serial CoT and parallel FJ. We then fine-tune a single model to solve 25 x 25 puzzles that remain challenging for standard CoT and FJ approaches, as well as frontier reasoning models without tools. Second, on 3-SAT puzzles, the capability of preemption allows termination of unpromising branches, which results in improved efficiency. Finally, we show that appropriately prompted large pre-trained models follow the MPLM protocol, achieving competitive results on long-context question answering relative to popular fork-join approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。