LLM 改进 LLM:用于测试时缩放的智能体自动发现 LLMs Improving LLMs: Agentic Discovery for Test-Time Scaling
AutoTTS 把 TTS 策略设计重构为控制器合成,5 轮发现即超越手工基线。
前置知识
测试时缩放 (Test-Time Scaling, TTS)
在推理阶段为单个问题分配更多计算以提升 LLM 表现的技术族,包括 Self-Consistency 式的宽度采样、Chain-of-Thought 式的深度延伸以及混合的剪枝/分支控制。
本文的核心研究对象就是 TTS 策略,理解何时分支、何时剪枝、何时停止是读懂 AutoTTS 设计空间的前提。
控制器合成 (Controller Synthesis)
在给定状态空间、动作集合和反馈信号的环境里,自动搜索出把状态映射到动作的策略 π(a|s),常用于强化学习与形式化方法。
AutoTTS 把 TTS 重新表述为控制器合成问题,整个框架的状态/动作/反馈定义都源自这一范式。
离线重放环境 (Offline Replay Environment)
在算法搜索前预先收集并存储模型对每条问题的多条推理轨迹和中间探测信号,让搜索阶段的策略评估无需再调用 LLM。
这是 AutoTTS 把发现成本压到 $39.9 的关键设计,没有重放环境每次评估都需重新采样会极昂贵。
Self-Consistency / Parallel-Probe 等基线
SC@64 采 64 条推理轨迹做多数投票;ASC 按阈值自适应停止;ESC 用滑动窗口检测稳定性;Parallel-Probe 通过跨分支探测动态剪枝/停止。
这些是 Table 1 中的强手工基线,AutoTTS 需要在 Pareto 前沿上同时击败它们才能证明发现的价值。
研究动机
现有 TTS 策略几乎全部由研究者手工设计:他们凭直觉为"何时分支、何时剪枝、何时停止"设定阈值与规则,再调参验证。例如 SC@64 在 Qwen3-8B/AIME25 上虽可达 76.7% 准确率却要消耗 1124.4k tokens,ASC 用 406.2k tokens 取得相同准确率;又如 Parallel-Probe 等近年方法依赖多个超参数协同调优。这种范式存在两大问题:第一,搜索空间庞大但人类只能覆盖其中少量点;第二,评估候选策略时若在线调用 LLM,每改一次阈值都要重新采样,开销难以承受,最终导致大量手工设计的策略虽然创新但成本-精度权衡并非最优。
本文的目标是本文提出 AutoTTS,目标是把 TTS 策略设计从"手工雕琢启发式"转向"环境驱动的自动发现"——让人类只需定义好状态/动作/反馈/目标所构成的可重放环境,由基于 Claude Code 的智能体在 5 轮迭代中自动合成能在准确性-成本 Pareto 前沿上击败 SC@64、ASC、ESC、Parallel-Probe 的代码化控制器 π(·|s, β),并通过单标量 β 在推理时灵活控制预算。
与已有工作不同的是,已有的 FunSearch、AlphaEvolve、ADAS 等 LLM 驱动程序搜索多用于数学/组合优化,没有把 TTS 视作搜索问题;TTT-Discover、ThetaEvolve 虽做测试时发现但需要更新模型权重。本文独特之处在于三点切入:(1) 把宽度-深度 TTS 重述为离线重放 MDP 上的控制器合成;(2) 引入 β 参数化把高维超参搜索塌缩成一维扫描以防止在搜索集上过拟合;(3) 用执行轨迹而非纯标量准确率作为反馈,让智能体能诊断失败模式——三者结合使得"环境即接口"的新范式首次在 TTS 上落地。
核心方法
AutoTTS 的总体思路是"搭环境、放智能体":先用被测模型在每个 (模型, 问题) 对上预采样 $N=128$ 条推理轨迹,按 $\Delta=500$ tokens 的固定间隔切片,得到分支前缀 $z_{i,1}, z_{i,2},\dots$ 和探测答案 $\omega_{i,k}$,构造离线重放环境;然后让 Claude Code 智能体在 5 轮内读取历史 $\mathcal{H}$(含过往控制器的代码、准确率、token 数与执行轨迹)并直接编辑控制器代码,新控制器在搜索集 $\mathcal{Q}_{\text{search}}$(AIME24 跨四个 Qwen3 模型)上沿 β 一维扫描求 Pareto 前沿并写回 $\mathcal{H}$。关键约束是控制器必须把全部超参表达为关于 β 的单调函数 map(β),从而把搜索空间从 $O(10^{10})$ 降到 1D,避免智能体陷入"靠搜索集特化阈值"的陷阱。
与已有方法相比,本文的本质区别在于"环境-策略分离":传统手工 TTS 把策略设计与环境耦合,AutoTTS 则把环境做成可重放的确定化 MDP,让策略搜索变成在固定数据上反复模拟;同时用 β 参数化把"高维黑盒调参"变成"一维预算旋钮",用执行轨迹替代纯标量反馈让智能体能定位失败原因(剪枝过激?停止过早?分支过少?)。这三者合在一起,把 TTS 研究从"凭直觉调阈值"变成"在受约束空间中做程序合成",是 Meta-Harness 范式在推理资源分配问题上的首次实例化。
方法步骤详情
方法分为四步。Step 1(数据预采集):对每个模型-问题对在温度 $T=0.7$ 下采样 $N=128$ 条完整推理轨迹,按 $\Delta=500$ tokens 分片记录前缀 $z_{i,k}$ 与中间答案 $\omega_{i,k}$,离线存储为可查询表。Step 2(环境构造):定义状态 $s_t=(q, m_t, I_t, \ell_t, Z_t, \Omega_t)$,动作集合 $\mathcal{A}(s)=\{\text{BRANCH}, \text{CONTINUE}(i), \text{PROBE}(i), \text{PRUNE}(i), \text{ANSWER}\}$,成本 $\text{Cost}(s_t)=\sum_{i=1}^{m_t}\ell_{t,i}+\kappa_\text{probe}|\Omega_t|$,聚合规则 Agg 由控制器自带。Step 3(β 参数化):智能体编写的控制器只能暴露一个标量 β,并实现单调函数 $\text{map}(\beta)\to$ 所有内部超参,使推理时只需扫一遍 β 即可获得 Pareto 曲线。Step 4(5 轮发现循环):每轮 Claude Code 读取 $\mathcal{H}$,诊断上次控制器在执行轨迹上的失败模式(例如过早剪枝或分支从未深度展开),改写控制器代码;评估时在 $\mathcal{Q}_{\text{search}}$ 上对每个候选沿 β 扫描 64 次采样平均,写回准确率-成本对与完整轨迹。发现完成后选在 $\mathcal{Q}_{\text{search}}$ 上准确率最高的 (π, β) 用于后续迁移评测。
技术新颖性
技术新颖性体现在四个层面。第一,把 TTS 显式建模为离线重放 MDP,状态空间含激活分支集合 $I_t$、深度 $\ell_t$、已揭示探测 $\Omega_t$ 和剩余预算,动作包含分支/继续/探测/剪枝/回答五类——这是对 Parallel-Probe 等已有算法的统一超集。第二,β 参数化与单调 map 函数约束同时存在,把"控制器发现"从组合爆炸的超参搜索降维成预算-精度曲线扫描,工程上等价于把"算法设计"和"预算控制"解耦。第三,执行轨迹反馈比纯标量准确率信息密度高得多,智能体可定位"哪一步决策导致精度塌方",这是从 Meta-Harness 借鉴但首次用于 TTS 资源分配的设计。第四,所发现控制器呈现四个非平凡机制——基于 EMA 动量的趋势停止、共享证据信号耦合宽度-深度、对齐感知的深度分配、保守分支放弃——这种协同复杂度手工难以设计,但通过环境+智能体组合自动涌现。
实验结果
核心发现分四块。(1) 在 4 个 Qwen3 模型 × 3 个数学基准的 12 组实验里,AutoTTS 在 $\beta=1.0$ 时于 5/8 设置下准确率超过所有手工基线,例如 Qwen3-8B/AIME24 上从 81.5% (Parallel-Probe) 提升到 85.8%,Qwen3-1.7B/AIME25 从 44.7% 提到 49.0%。(2) $\beta=0.5$ 的低预算变体相对 SC@64 平均减少 69.5% tokens 而精度持平(45.3 vs 45.2),并在 Qwen3-0.6B/AIME25 把 token 从 890.5k 砍到 250.3k,准确率 28.6% vs 28.9%,证明 Pareto 前沿整体被推进。(3) 跨模型/任务迁移上,DeepSeek-R1-Distill-Llama-8B/HMMT25 下 $\beta=1$ 取得 27.2%(超过 SC@64 的 26.7%)同时把 985.7k tokens 压到 533.9k;Qwen3-1.7B/GPQA-Diamond 非数学任务上 $\beta=0.5$ 把 510.0k tokens 降到 151.0k 且精度 41.6% 与 SC@64 持平。(4) 消融实验显示去掉 β 参数化会让准确率从 53.1 掉到 49.0(剪枝过激过拟合),去掉执行轨迹反馈则 token 从 575.5k 涨到 824.3k 且精度也降到 51.6,两者均证实设计选择的必要性。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Qwen3-8B / AIME24 (搜索集) | 准确率 (%) / 总 tokens (k) | AutoTTS β=1.0: 85.8% / 467.4k | Parallel-Probe: 81.5% / 730.8k | 准确率 +4.3pp,tokens -36.0% |
| Qwen3-1.7B / AIME25 (留出) | 准确率 (%) / 总 tokens (k) | AutoTTS β=1.0: 49.0% / 612.6k | Parallel-Probe: 44.7% / 775.8k | 准确率 +4.3pp,tokens -21.0% |
| Qwen3-0.6B / AIME25 (留出) | 准确率 (%) / 总 tokens (k) | AutoTTS β=0.5: 28.6% / 250.3k | SC@64: 28.9% / 890.5k | 精度 -0.3pp 但 tokens -71.9% |
| DeepSeek-R1-Distill-Llama-8B / HMMT25 | 准确率 (%) / 总 tokens (k) | AutoTTS β=1: 27.2% / 533.9k | SC@64: 26.7% / 985.7k | 准确率 +0.5pp,tokens -45.8% |
| Qwen3-1.7B / GPQA-Diamond (非数学) | 准确率 (%) / 总 tokens (k) | AutoTTS β=0.5: 41.6% / 151.0k | SC@64: 41.3% / 510.0k | 精度 +0.3pp,tokens -70.4% |
| 消融:去掉 β 参数化 (四模型平均) | 留出集平均准确率 (%) / 总 tokens (k) | 完整方法: 53.1 / 575.5k | w/o Beta: 49.0 / 93.3k | β 参数化用 6.2 倍 token 换 +4.1pp 精度,避免过拟合 |
局限与改进
作者承认的局限有四:第一,离线重放要求每个 (模型, 问题) 预采 128 条轨迹,单一模型的 replay matrix 构建成本已经不小,把它迁移到新模型需重做数据;第二,搜索集仅用 AIME24,虽然留出 AIME25/HMMT25 表现稳健,但 $\mathcal{Q}_{\text{search}}$ 单一时智能体的探索多样性可能受限;第三,β 参数化强制单调约束限制了非单调策略的可能形态,例如"小预算深分支、大预算浅分支"这种非单调 map 无法表达;第四,控制器代码是智能体自由编写的,无形式化正确性保证,可能包含边界 bug。我的额外观察是:(a) 控制器只在四个 Qwen3 规模 + DeepSeek-R1-Distill 蒸馏模型上测试,对超大模型(如 70B+)和弱模型 (<0.6B) 行为未知;(b) Table 2 中 DeepSeek-R1-Distill 上的 token 节省比例明显低于 Qwen3 系列,可能反映蒸馏模型推理质量差异;(c) β 参数化在 $\beta=0$ 与 $\beta=1$ 之间的中间值是否平滑连续、能否给出置信区间未做分析;(d) 论文未给出控制器代码的循环复杂度上界,部署在延迟敏感场景时推理时间可能成为新瓶颈。
独立分析的弱点
独立分析存在三点可改进的弱点。(1) 离线重放的可移植性差:每换一个模型或基准都要重新采样 $128 \times |Q|$ 条轨迹,对小实验室极不友好;改进方向是构造可跨模型迁移的代理 replay,或用重要性采样在已有轨迹上估计新模型行为。(2) β 参数化虽避免过拟合但牺牲了灵活性,例如 Parallel-Probe 的多阈值结构无法表达;改进方向是引入"复合 β"或分段 map,在保持单调约束的同时允许局部精细调节。(3) 5 轮发现迭代的方差未量化:Table 3 显示不同 ablations 搜索成本差异显著($30.9 vs $46.4),但没报告多次独立 5 轮运行的方差,结论的统计显著性存疑;改进方向是报告 ≥3 次独立运行的均值±标准差并配显著性检验。
未来方向
作者提出的未来方向是把"环境驱动发现"范式扩展到更广的 TTS 算法族,例如结合树搜索、MCTS、verifier-guided refinement 的更丰富控制空间。基于成果可以延伸的研究包括:(a) 把 replay 矩阵跨模型做迁移学习,避免每个新模型都重采 128 条;(b) 把 β 参数化扩展为多维预算向量(如同时控制宽度预算与深度预算),并约束其联合单调性;(c) 把控制器发现用于非数学推理任务(代码生成、Agent 决策),验证环境范式的通用性;(d) 把发现过程本身做成在线学习,让控制器在使用中持续演化;(e) 与 RL 微调结合,让 TTS 策略与模型权重协同优化;(f) 探索执行轨迹的自动摘要技术,降低智能体的上下文开销以把发现成本进一步压到 100 美元以下。
复现评估
复现评估整体良好:作者承诺代码与数据开源(论文末尾声明 "Our data, and code will be open-source at https://github.com/zhengkid/AutoTTS",但截止到本文写作时需自行核对实际仓库状态);关键技术细节如采样温度 0.7、128 条轨迹、间隔 500 tokens、64 次采样平均、4 个 Qwen3 模型、AIME24/25/HMMT25/GPQA-Diamond 评测集、Claude Code 作为探索 LLM 都在 §4 中明确给出;发现总成本 $39.9、160 分钟意味着复现门槛极低。但风险点在于:(a) Claude Code 的具体 prompt 模板只在 Appendix C 给出摘要,全文依赖该提示的设计;(b) 智能体迭代的随机性意味着同样 5 轮可能发现不同控制器,需要多次运行取最优;(c) replay matrix 体积大(每模型 30 题 × 128 条 × 全部 token),存储与 IO 是工程难点;(d) 不同版本 Claude Code 行为差异可能导致结果波动。综合而言,复现难度中等,主要门槛在 Claude Code 访问与 replay 数据准备。
论文图表
在 width×depth 二维平面上画出 SC@64、ASC、ESC、Answer-Consistency、ST-BON、Parallel-Probe 等六种代表性算法的轨迹:SC@64 占据满预算角落,ASC/ESC 只沿宽度轴扩展,Answer-Consistency 只沿深度轴,ST-BON 先宽后窄再深,Parallel-Probe 由宽渐窄渐深。
揭示本文的核心洞察——"许多 TTS 算法可视为宽度-深度空间中的不同路径",为后续"用控制器统一描述它们"奠定基础。