arXiv:2412.10261cs.CVcs.AR2024-12中稿 · ASPLOS '25被引 8

MVQ通过掩码量化提升模型压缩效率,显著降低功耗与存储开销。

MVQ:Towards Efficient DNN Compression and Acceleration with Masked Vector Quantization

  • 结合剪枝与掩码k-means,仅对重要权重进行向量聚类
  • 在相同压缩比下精度更高,且减少55%计算阵列面积
  • 适合部署于边缘设备的高效DNN加速器设计

向量量化(VQ)是一种硬件友好的深度神经网络压缩方法,可降低存储成本和加速器的权重量化位宽。但传统VQ方法因未能有效保留重要权重而导致显著精度损失。为此,本文提出新型方法MVQ,旨在用有限码本更精确地逼近重要权重。算法层面,先通过N:M剪枝移除次要权重,再利用掩码k-means算法最小化剩余权重与码本间的向量聚类误差,仅计算未剪枝权重与码本间距离并用于更新码本。架构层面,在EWS(增强权值站式)CNN加速器上实现向量量化,并设计稀疏卷积阵列以最大化掩码向量量化的优势。在图像分类、目标检测与分割任务上的实验表明,MVQ在相近压缩比下优于传统向量量化方法,同时减少计算量。ASIC评估显示,该加速器相比基线EWS加速器提升2.3倍能效,且计算阵列面积减少55%;相较此前稀疏加速器,能效提高1.73倍。

原文摘要 · Abstract (English)

Vector quantization(VQ) is a hardware-friendly DNN compression method that can reduce the storage cost and weight-loading datawidth of hardware accelerators. However, conventional VQ techniques lead to significant accuracy loss because the important weights are not well preserved. To tackle this problem, a novel approach called MVQ is proposed, which aims at better approximating important weights with a limited number of codewords. At the algorithm level, our approach removes the less important weights through N:M pruning and then minimizes the vector clustering error between the remaining weights and codewords by the masked k-means algorithm. Only distances between the unpruned weights and the codewords are computed, which are then used to update the codewords. At the architecture level, our accelerator implements vector quantization on an EWS (Enhanced weight stationary) CNN accelerator and proposes a sparse systolic array design to maximize the benefits brought by masked vector quantization.\\ Our algorithm is validated on various models for image classification, object detection, and segmentation tasks. Experimental results demonstrate that MVQ not only outperforms conventional vector quantization methods at comparable compression ratios but also reduces FLOPs. Under ASIC evaluation, our MVQ accelerator boosts energy efficiency by 2.3$\times$ and reduces the size of the systolic array by 55\% when compared with the base EWS accelerator. Compared to the previous sparse accelerators, MVQ achieves 1.73$\times$ higher energy efficiency.

模型压缩向量量化加速器设计稀疏计算

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