通过动态调温提升低精度推理的准确率和速度
ReSET: Accurate Latency-Critical NVFP4 Reasoning via Step-Aware Temperature Scaling

- 根据推理步骤熵值在线调整温度,缓解量化带来的采样偏差
- 在多个基准上使低精度推理准确率提升约2个百分点
- 专为小批量自回归解码优化,实现2.5倍内核加速
大型推理模型(LRMs)通过生成长序列中间推理路径来提升复杂问题求解能力,但显著增加了推理开销。NVFP4推理通过硬件支持的低精度执行,有望降低计算与内存成本。然而,直接应用于LRMs时存在两大实际限制:量化导致推理准确率下降,且现有NVFP4内核在小批量自回归解码中未能充分发挥延迟优势。本文分析了NVFP4量化对推理过程中词元级不确定性的影。结果表明,量化会加剧低熵符号词元的错误采样,同时在高不确定性推理步骤中造成对少数词元的过度集中。基于此,提出ReSET——一种基于推理步熵的温度调节方法,可在线估计步骤级不确定性,并结合词元级与步骤级熵信号动态调整解码温度。为进一步缩小延迟差距,设计了一个针对小规模矩阵乘法(small-$M$)的CUDA核心级NVFP4内核,专用于延迟敏感的自回归解码。在多个推理基准和模型规模下,ReSET相比基线NVFP4推理准确率提升最高达约2个百分点。所提内核相较NVFP4 vLLM实现高达2.5倍的内核级加速,端到端解码速度提升约2倍。代码已开源。
原文摘要 · Abstract (English)
Large reasoning models (LRMs) improve complex problem-solving by generating long intermediate reasoning traces, but this substantially increases inference costs. NVFP4 inference offers a promising approach to reduce both computational and memory costs through hardware-supported low-precision execution. However, directly applying NVFP4 to LRMs introduces two practical limitations: reasoning accuracy degrades under quantization, and existing NVFP4 kernels do not fully realize latency benefits in small-batch autoregressive decoding. In this work, we analyze the effect of NVFP4 quantization on token-level uncertainty during reasoning. We show that quantization increases incorrect sampling at low-entropy symbolic tokens, while causing over-concentration on a small set of tokens in high-uncertainty reasoning steps. Based on this observation, we propose \textbf{ReSET}, a reasoning-step entropy-based temperature-scaling method that estimates step-level uncertainty online and adapts the decoding temperature using both token-level and step-level entropy signals. To address the latency gap, we further design a CUDA-core small-$M$ NVFP4 kernel for latency-critical autoregressive decoding. Across reasoning benchmarks and model scales, ReSET improves NVFP4 reasoning accuracy by up to $\sim\!$2 points over the NVFP4 baseline. Our CUDA-core small-$M$ kernel further improves latency-critical decoding, delivering up to $2.5\!\times$ kernel-level speedup over NVFP4 vLLM and approximately $2\!\times$ end-to-end decoding speedup over BF16. Code is available at https://github.com/aiha-lab/ReSET.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。