SuperAPI AI 工具集

predicting-improving-test-time-scaling

智能体与模型

查看项目

更新于 10-16 01:34

技能介绍

用途:优化大模型推理时的测试时算力分配。传统 best-of-N 对每条候选路径均等采样,浪费算力;SLG 搜索先对若干中间状态做少量试点采样,用 GPD 拟合奖励分布上尾,估计尾部形状参数 xi,再预测追加预算可达到的最优奖励,从而把剩余算力集中到最有潜力的状态。

主要能力:1)尾部预测:对每个状态取前 20% 奖励,用 scipy.stats.genpareto.fit 拟合 GPD,得到 xi 与 sigma,用 V(s,N)=baseline+sigma/(1-xi)*N^xi 预测收益;2)自动超参:试点数 m=max(20,floor(log(N)^3/5)),探索状态数 K=floor(N/(2m)),剩余预算用于利用;3)流程:Expand 生成 K 个中间状态 → Estimate 各做 m 次试点打分 → Select 选预测值最高者 → Exploit 将剩余预算全部投入该状态 → 返回全局最优。

使用方式:需提供生成函数与标量奖励模型,按上述步骤实现,并记录预测值与实际最优以校准。参考实现见 github.com/PotatoJnny/Scaling-Law-Guided-search。

适用场景:数学推理、代码生成、规划等有连续奖励信号的多候选任务;也可用于在不做穷举评估的情况下预测 N 增大时的性能曲线。

限制:依赖连续标量奖励,二值奖励需改用贝叶斯方法;总预算低于约 50 次时 BoN 更实用;默认单层展开,多层扩展复杂度与延迟更高;奖励模型质量差会放大误差;GPD 拟合增加延迟且需 scipy;假设奖励分布稳定。

论文:Li, Qian, Mou(2026),arXiv:2602.01485。