arXiv:2605.20295cs.LGcs.AI2026-05

让手机NPU高效运行大模型,全静态量化不依赖运行时计算

Quant.npu: Enabling Efficient Mobile NPU Inference for on-device LLMs via Fully Static Quantization

论文配图:Quant.npu: Enabling Efficient Mobile NPU Inference for on-device LLMs via Fully Static Quantization
图 1 · 摘自论文原文
  • 全静态量化+可学习参数与旋转矩阵,无需运行时重算
  • 在真实手机NPU上推理延迟降低15.1%,精度接近顶尖方法
  • 适合移动端部署低比特大模型,尤其对NPU硬件友好

大型语言模型(LLMs)正越来越多地部署在移动设备上,而神经网络处理单元(NPUs)要求完全静态量化以实现最优推理效率。然而,现有后训练量化(PTQ)方法多依赖动态激活量化,无法满足NPU硬件约束。为此,我们提出Quant.npu,一个纯整数、全静态量化框架。该框架引入可学习量化参数与旋转矩阵,实现低比特激活-权重量化且无需运行时重新计算量化参数。关键发现是:量化参数的初始化与选择性优化对优化稳定性至关重要;不当初始化与联合优化会导致梯度不稳定,破坏旋转矩阵的优化。为此,我们设计了针对不同激活分布的旋转与位宽感知初始化,以及针对旋转/未旋转张量的分布感知选择性优化(两阶段量化流程)。此外,提出基于敏感度引导的自适应混合精度方案,在精度与推理效率间取得平衡。在真实移动NPU上的大量实验表明,Quant.npu在精度上媲美最先进方法,同时推理延迟最高降低15.1%。

原文摘要 · Abstract (English)

Large language models (LLMs) are increasingly deployed on mobile devices, where Neural Processing Units (NPUs) necessitate fully static quantization for optimal inference efficiency. However, existing post-training quantization (PTQ) methods predominantly rely on dynamic activation quantization, rendering them incompatible with NPU hardware constraints. To bridge the gap between high-fidelity PTQ and NPU-constrained inference, we propose Quant.npu, a integer-only fully static quantization framework. It incorporates learnable quantization parameters and rotation matrices, enabling low-bit activation-weight quantization without runtime quantization parameters re-computation. Crucially, we identify that initialization and selective optimization of quantization parameters is pivotal for optimization stability, as improper initialization and naive joint optimization induce gradient instability that disrupts the optimization of rotation matrices. To address this, we propose a rotation-and-bit-width-aware initialization tailored to diverse activation profiles and a distribution-aware selective optimization (two-stage quantization pipeline) tailored to rotated and unrotated tensors. Furthermore, we introduce a sensitivity-guided adaptive mixed-precision scheme to balance accuracy with inference efficiency. Extensive experiments on real-world mobile NPUs demonstrate that Quant.npu achieves comparable accuracy to state-of-the-art methods, while reducing inference latency by up to 15.1%.

大模型部署NPU优化量化静态量化

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