arXiv:2512.23367cs.LGcs.AI2025-12被引 1

将大模型量化至INT8/W4A8,实现高效推理部署

Post-Training Quantization of OpenPangu Models for Efficient Deployment on Atlas A2

  • 采用统一低比特推理框架,支持INT8与W4A8量化
  • INT8保持90%以上精度,预填充速度提升1.5倍
  • 适合在Atlas A2上部署需要高效推理的轻量模型

华为开源的openPangu-Embedded-1B和openPangu-Embedded-7B是专为Ascend NPU设计的轻量级大语言模型。7B版本支持slow_think、auto_think和no_think三种思维链(CoT)推理模式,1B版本仅运行no_think模式,采用压缩推理以提升效率。尽管CoT增强模型能力,但生成长推理轨迹带来显著内存与延迟开销,制约其在Ascend NPUs上的实际部署。本文通过低比特量化将FP16计算转为高效整数运算,提出针对openPangu-Embedded模型优化的统一低比特推理框架,支持INT8(W8A8)与W4A8量化。在HumanEval和MBPP代码生成基准上的评估显示,INT8量化保持超过90%的FP16基线精度,并在Atlas A2上实现1.5倍预填充加速;W4A8量化显著降低内存占用,虽精度略有下降,但整体仍维持较高模型保真度。结果表明,低比特量化可有效支撑Ascend NPUs上高效CoT推理。

原文摘要 · Abstract (English)

Huawei's openPangu-Embedded-1B and openPangu-Embedded-7B are variants of the openPangu large language model, designed for efficient deployment on Ascend NPUs. The 7B variant supports three distinct Chain-of-Thought (CoT) reasoning paradigms, namely slow_think, auto_think, and no_think, while the 1B variant operates exclusively in the no_think mode, which employs condensed reasoning for higher efficiency. Although CoT reasoning enhances capability, the generation of extended reasoning traces introduces substantial memory and latency overheads, posing challenges for practical deployment on Ascend NPUs. This paper addresses these computational constraints by leveraging low-bit quantization, which transforms FP16 computations into more efficient integer arithmetic. We introduce a unified low-bit inference framework, supporting INT8 (W8A8) and W4A8 quantization, specifically optimized for openPangu-Embedded models on the Atlas A2. Our comprehensive evaluation on code generation benchmarks (HumanEval and MBPP) demonstrates the efficacy of this approach. INT8 quantization consistently preserves over 90\% of the FP16 baseline accuracy and achieves a 1.5x prefill speedup on the Atlas A2. Furthermore, W4A8 quantization significantly reduces memory consumption, albeit with a moderate trade-off in accuracy. These findings collectively indicate that low-bit quantization effectively facilitates efficient CoT reasoning on Ascend NPUs, maintaining high model fidelity.

模型量化推理加速轻量模型Ascend NPU

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