arXiv:2509.15148cs.CL2025-09中稿 · ICLR被引 8

ATTS通过异步推理实现测试时加速,显著降低延迟与内存开销。

ATTS: Asynchronous Test-Time Scaling via Conformal Prediction

  • 基于置信度校准的异步推理框架,支持并行与串行双重扩展
  • 在多个数据集上实现最高56.7倍加速和4.14倍吞吐提升
  • 适用于追求高效率推理的大型语言模型部署场景

大型语言模型在测试时扩展中受益,但常受高推理延迟制约。推测解码虽可加速扩展过程,但在并行与串行维度同时扩展时面临严重内存占用和同步开销问题。本文提出ATTS(异步测试时扩展),一种基于置信区间检验的统计保障自适应扩展框架。通过重审算术强度,识别出同步为首要瓶颈。该方法通过在线校准实现异步推理,并提出一种序数分类算法,支持三阶段拒绝采样流水线,实现并行与串行双维扩展。在MATH、AMC23、AIME24和AIME25数据集上,针对多种草稿-目标模型组合的实验表明,ATTS可实现最高56.7倍的测试时扩展加速,吞吐提升4.14倍,同时精准控制拒绝率,降低延迟与内存开销,且无准确率损失。通过双维扩展,1.5B/70B草稿-目标模型组合在AIME数据集上达到o3-mini(high)状态领先推理模型性能。代码已开源:https://github.com/menik1126/asynchronous-test-time-scaling。

原文摘要 · Abstract (English)

Large language models (LLMs) benefit from test-time scaling but are often hampered by high inference latency. Speculative decoding is a natural way to accelerate the scaling process; however, scaling along both the parallel and sequential dimensions poses significant challenges, including substantial memory-bound execution and synchronization overhead. We introduce ATTS (Asynchronous Test-Time Scaling), a statistically guaranteed adaptive scaling framework that follows the hypothesis testing process to address these challenges. By revisiting arithmetic intensity, ATTS identifies synchronization as the primary bottleneck. It enables asynchronous inference through online calibration and proposes an ordinal classification algorithm that supports a three-stage rejection sampling pipeline, scaling along both the sequential and parallel axes. Across experiments on the MATH, AMC23, AIME24, and AIME25 datasets and across multiple draft-target model families, we show that ATTS delivers up to 56.7x speedup in test-time scaling and a 4.14x throughput improvement, while maintaining accurate control of the rejection rate, reducing latency and memory overhead, and incurring no accuracy loss. By scaling both in parallel and sequential dimensions, we enable the 1.5B/70B draft/target model combination to achieve the performance of the state-of-the-art reasoning model o3-mini (high) on the AIME dataset. We have released the code at https://github.com/menik1126/asynchronous-test-time-scaling.

测试时扩展异步推理语言模型优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。