arXiv:2603.22370cs.LGcs.AI2026-03被引 3

针对NVFP4格式设计自适应舍入,显著降低大模型在边缘设备上的量化误差。

FAAR: Format-Aware Adaptive Rounding for NVFP4

  • 根据NVFP4非均匀数值网格动态调整舍入策略,提升量化精度。
  • 在Llama3-1B上将WikiText-2困惑度从14.28降至12.60,性能优于现有方法。
  • 仅需4小时GPU训练开销,适合资源受限的边缘部署场景。

将大语言模型(LLMs)部署于边缘设备需要极低比特量化。超低精度格式如NVFP4可有效减少内存占用并加速计算。然而,现有量化方法通常依赖传统舍入策略,未能考虑NVFP4数值网格的非均匀性,导致次优舍入决策和放大量化误差。为此,我们提出格式感知自适应舍入(FAAR),一种专为NVFP4格式设计的可学习舍入策略。与传统范式不同,FAAR将NVFP4非均匀网格显式纳入优化过程,通过损失梯度引导自适应调整舍入决策,有效逼近理论最优量化。为配合FAAR,我们引入两阶段格式对齐(2FA)微调方案,逐层对齐模型参数至NVFP4数值空间,进一步缩小性能差距。值得注意的是,该可学习优化仅带来4 GPU小时的额外训练开销。大量实验表明,相比四舍五入(RTN),我们的方法在Llama3-1B上将WikiText-2困惑度从14.28降至12.60,在Qwen3-1.7B上从23.06降至21.27。此外,该方法在多个零样本下游任务中持续超越当前最先进水平。

原文摘要 · Abstract (English)

Deploying large language models (LLMs) on edge devices requires extremely low-bit quantization. Ultra-low precision formats such as NVFP4 offer a promising solution for reducing memory footprint and accelerating computation. However, existing quantization methods typically rely on conventional rounding strategies and fail to account for the non-uniformity of the NVFP4 numerical grid, resulting in suboptimal rounding decisions and amplified quantization errors. To address this, we propose Format-Aware Adaptive Rounding (FAAR), a learnable rounding strategy tailored for the NVFP4 format. Unlike conventional quantization paradigms, FAAR explicitly incorporates the non-uniform NVFP4 grid into the optimization process. By adaptively adjusting rounding decisions guided by loss gradients, our method effectively approximates the theoretically optimal quantization. To complement FAAR, we introduce a 2-stages Format Alignment (2FA) fine-tuning scheme that aligns LLM parameters layer-by-layer to the NVFP4 numerical space, further narrowing the performance gap. Remarkably, this learnable optimization incurs a minimal training overhead of only 4 GPU hours on Llama3-1B. Extensive experiments demonstrate the effectiveness of our approach. Compared with Round-to-Nearest (RTN), our method reduces perplexity on WikiText-2 from 14.28 to 12.60 on Llama3-1B and from 23.06 to 21.27 on Qwen3-1.7B. Additionally, our method consistently outperforms state-of-the-art approaches across various zero-shot downstream tasks.

低比特量化边缘部署NVFP4自适应舍入

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