arXiv:2511.06838cs.ARcs.LG2025-11中稿 · the 53rd IEEE/ACM …被引 4

P3-LLM通过混合精度量化与低精度存算一体设计,提升边缘端大模型推理速度与能效。

P3-LLM: An Integrated NPU-PIM Accelerator for Edge LLM Inference Using Hybrid Numerical Formats

  • 采用混合数值格式的灵活量化方案,压缩模型且损失小。
  • 在同面积约束下,低精度存算单元实现4.9倍加速。
  • 适合边缘设备部署,尤其对算力受限场景友好。

大语言模型(LLMs)对内存带宽和计算资源的高需求给高效推理带来挑战。现有研究采用神经处理器(NPU)与基于DRAM的存内计算(PIM)异构系统加速,但高精度PIM计算单元在DRAM中占用大量面积与功耗,限制了计算吞吐量。本文提出P3-LLM,一种面向边缘端大模型推理的NPU-PIM集成加速器。首先,提出灵活的混合精度量化方案,利用混合数值格式对不同模型组件进行量化,实现高压缩率与低精度损失。其次,设计支持混合格式的高效PIM加速器,通过合理选择数值格式,协同优化低精度计算单元,在相同面积下显著提升计算吞吐。第三,通过操作融合优化低精度数据流,减少运行时反量化开销。在多种大模型与任务上的评估表明,P3-LLM的精度高于现有KV缓存量化与权值-激活量化方法。结合量化方案与低精度架构协同设计,相比HBM-PIM、Ecco、Pimba等先进加速器,平均分别提速4.9倍、2.0倍、3.4倍。代码已开源:https://github.com/yc2367/P3-LLM。

原文摘要 · Abstract (English)

The substantial memory bandwidth and computational demands of large language models (LLMs) present critical challenges for efficient inference. To tackle this, the literature has explored heterogeneous systems that combine neural processing units (NPUs) with DRAM-based processing-in-memory (PIM) for LLM acceleration. However, the high-precision PIM compute units incur significant area and power overhead in DRAM technology, limiting the effective computation throughput. In this paper, we introduce P3-LLM, a novel NPU-PIM integrated accelerator for edge LLM inference. Our approach is threefold: First, we propose a flexible mixed-precision quantization scheme, which leverages hybrid numerical formats to quantize different LLM operands with high compression efficiency and minimal accuracy loss. Second, we architect an efficient PIM accelerator for P3-LLM, featuring enhanced compute units to support hybrid numerical formats. Our careful choice of numerical formats allows to co-design low-precision PIM compute units that significantly boost the computation throughput under iso-area constraints. Third, we optimize the low-precision dataflow of different LLM modules by applying operator fusion to minimize the overhead of runtime dequantization. Evaluations on diverse LLMs and tasks demonstrate that P3-LLM achieves higher accuracy than state-of-the-art KV-cache quantization and weight-activation quantization algorithms. Combining the proposed quantization scheme with low-precision PIM architecture co-design, P3-LLM yields an average of $4.9\times$, $2.0\times$, and $3.4\times$ speedups over state-of-the-art LLM accelerators HBM-PIM, Ecco, and Pimba, respectively. Code is available at https://github.com/yc2367/P3-LLM.

边缘计算存算一体混合精度

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