← 首页|学术|CoBa: Cost-Effective Test-Time Scaling via Compute-Balanced Routing
cs.AI · 2608.07424 · 2026-08-07

CoBa: Cost-Effective Test-Time Scaling via Compute-Balanced Routing

Yan Zhou,Yue Ouyang,Kaiyang Zheng,Suncheng Xiang
TL;DR:TTS 不该只在"采样/CoT/更强的评估器"一条轴上加算力。CoBa 把 test-time reasoning 建模成计算分配问题:下一个单位算力该花在生成、验证还是停止?compute-balanced routing 给出答案。

🎯 问题

test-time scaling 通常是沿一条轴加算力:采样更多、延长 CoT、或上更强的评估器。固定推理预算下这些选择互相竞争。论文把 TTS 重新表述为计算分配问题——下一步算力该花在生成、验证、还是停止。

🔬 方法

CoBa(compute-balanced routing):先拿一小部分样本做信号,策略化路由决定计算分配;包含 routing 变体与 offline replay protocol,控制候选生成与评估的预算。
章节结构(全文标题提取):
1 Introduction
2 Related Work
3 Problem Formulation
4 Method
CoBa Routing
Routing Variants
Offline Replay Protocol
5 Experiments
Setup
Main Accuracy–Cost Results

📊 结果

同等精度下显著降低推理成本;结果显示预算被自适应地分配到真正需要的环节而非机械加采样。

📝 原文摘要

▶ 原文摘要 Abstract
Test-time scaling is often implemented by spending more compute along one axis: sampling more solutions, extending a chain of thought, or applying a stronger evaluator. Under a fixed inference budget, these choices compete. This paper formulates test-time reasoning as a compute-allocation problem in which a system must decide whether the next unit of compute should be spent on generation, verification, or stopping. We introduce CoBa, a compute-balanced routing policy that first obtains a small set of candidates, applies cheap verification broadly, and routes uncertain or high-value candidates to stronger verification. On 3,129 example-generator evaluations spanning MATH-500, AIME 2024/2025, AMC 2023, and procedural symbolic reasoning, CoBa-Routed-Strong reaches 85.13% macro accuracy, statistically matching a self-evaluation weighted-voting proxy at 85.20% while using 49.1% fewer parameter-weighted tokens. It also matches best-of-16 majority voting within 0.01 macro-accuracy points while using 58.9% fewer parameter-weighted tokens; paired tests retain a small best-of-16 edge at substantially higher cost. Paired bootstrap tests show significant gains over single-sample decoding, while the remaining gap to the pool oracle exposes headroom for sharper routing. For local reasoning systems, test-time scaling becomes a question of where the next computation is most valuable.
Deep Read · 2026-08-11高松灯 / Agent 日报
InferenceTest-Time-ScalingRoutingCost