arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.07670cs.LGcs.AI
收藏

惊讶引导的选择:为执行导向代码生成的计算最优测试时间策略

Surprisal-Guided Selection: Compute-Optimal Test-Time Strategies for Execution-Grounded Code Generation

Arc Intelligence(Arc智能)

Jarrod Barnes

AI总结:

本文提出惊讶引导选择策略,通过提升样本多样性和智能选择,优化执行导向代码生成任务的测试时间策略,实现计算最优的高成功率。

摘要:

测试时间训练(TTT)通过在推理时基于梯度的更新来适应语言模型。但适应是否是正确的策略?我们研究了可验证的执行导向(VEG)任务的计算最优测试时间策略,如GPU内核优化等领域,其中确定性评估器提供密集的连续奖励信号。使用KernelBench作为我们的测试平台和一个1200亿参数模型(GPT-OSS-120B带有LoRA适应),我们发现搜索优于最小适应(1-5个梯度步骤):在K=64的情况下,最佳N采样在完整的KernelBench L1评估集上实现了90%的任务成功率(18/20任务),而TTT的最佳检查点仅达到30.6%(3种子均值),TTT的“等效K”低于1,比单样本推理更差。失败模式是过度锐化:梯度更新将多样性压缩向中等解决方案,而不是发现最优解。我们的主要贡献是惊讶引导的选择:选择最高惊讶(最低信心)的正确样本可实现80%的成功率,而选择最自信样本仅为50%,提高了30%。扩展到惊讶引导的top3在100%的Oracle性能上表现优异。这种零成本策略通过长度控制分析验证,恢复了Oracle性能。对于密集奖励VEG任务,计算应分配给样本多样性和智能选择,而不是梯度适应。惊讶引导的选择原则可能推广到其他执行导向领域,其中最优解占据分布尾部。

英文摘要:

Test-time training (TTT) adapts language models through gradient-based updates at inference. But is adaptation the right strategy? We study compute-optimal test-time strategies for verifiable execution-grounded (VEG) tasks, domains like GPU kernel optimization where a deterministic evaluator provides dense, continuous reward signals. Using KernelBench as our testbed and a 120B-parameter model (GPT-OSS-120B with LoRA adaptation), we find that search outperforms minimal adaptation (1-5 gradient steps): Best-of-N sampling achieves 90% task success (18/20 tasks) at K=64 across the full KernelBench L1 eval set while TTT's best checkpoint reaches only 30.6% (3-seed mean), with TTT's "equivalent K" falling below 1, worse than single-sample inference. The failure mode is over-sharpening: gradient updates collapse diversity toward mediocre solutions rather than discovering optimal ones. Our main contribution is surprisal-guided selection: selecting the highest-surprisal (lowest-confidence) correct sample yields 80% success vs. 50% for most-confident selection, a 30% improvement. Extending to surprisal-guided-top3 matches oracle performance at 100%. This zero-cost strategy, validated through length-controlled analysis, recovers oracle performance. For dense-reward VEG tasks, compute should be allocated to sample diversity and intelligent selection rather than gradient adaptation. The surprisal-guided selection principle may generalize to other execution-grounded domains where optimal solutions occupy the distribution tail.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑