AI 推理的算力,终于知道该花在哪了
现在的大模型在解题时,算力分配极其浪费:要么把所有可能的思路都并行跑一遍,内存直接爆掉;要么砍掉不行的思路,但硬件又闲着。这篇提出 Gambit,让模型像下棋一样,每走几步就淘汰没希望的思路,只从最有希望的分支继续展开,同时保持硬件一直满载。结果在同样算力下,准确率最高提升 6.7%,token 消耗最多省 68.5%。它不是你明天能用上的,但这是推理模型从「拼算力」转向「拼分配」的关键一步。
📄 原文摘要(英文)
Test-time compute scaling is a primary driver of performance in large reasoning models (LRMs), but extreme inefficiency bounds current approaches, shifting the critical question from how much compute to spend, to where to allocate it. We formalize test-time reasoning as a constrained compute allocation problem over partial trajectories. Under a fixed hardware budget, existing paradigms fail to actively allocate the compute to the most promising partial progress: traditional parallel sampling treats traces independently and induces severe memory bottlenecks, while subtractive pruning starves hardware and fails to actively and sufficiently shift the output distribution. To overcome this dichotomy, we introduce Gambit, an inference algorithm that executes thought-level beam search. By periodically pruning unpromising trajectories and immediately branching from high-quality prefixes, Gambit dynamically concentrates compute onto the most promising reasoning traces via a light-weight scorer probing hidden states while maintaining continuous high hardware utilization. Extensive evaluations across multiple models and benchmarks demonstrate that Gambit strictly dominates existing baselines. Under identical hardware constraints, our method yields up to a +6.7\% absolute accuracy gain on HMMT-24 and +3.3\% on AIME-25 over pruning baselines, delivers >2times higher throughput on trace completion, and reduces total token consumption by up to 68.5\% relative to standard parallel sampling.