arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.05097cs.CVcs.LGeess.IV
收藏

用于图像分类和基于块压缩的1Mb混合精度量化编码器

A 1Mb mixed-precision quantized encoder for image classification and patch-based compression

CEA-LETI;University Grenoble Alpes(格勒诺布尔阿尔卑斯大学)

Van Thien Nguyen, William Guicquero, Gilles Sicard

AI总结:

本文提出一种1Mb可重构混合精度ASIC编码器,通过权重/激活量化、结构化剪枝、量化均衡与位移归一化,在CIFAR-10上达到87.5%准确率并支持低块效应的逐块图像压缩。

摘要:

尽管专用集成电路(ASIC)已被证明是在边缘端集成推理的一种相关选择,但其在适用性方面往往受到限制。本文证明,专用于图像处理的ASIC神经网络加速器可应用于不同层级的多个任务:图像分类和压缩,同时只需要非常有限的硬件。关键组件是一个可重构的混合精度(3b/2b/1b)编码器,它利用适当的权重和激活量化,并结合卷积层结构化剪枝,以降低硬件相关约束(存储器和计算)。我们引入了线性对称量化器缩放因子的自动自适应,以执行量化级别均衡,旨在稳定五值和三值权重训练。此外,所提出的层共享位移归一化显著简化了硬件代价高昂的批量归一化的实现。在编码器设计仅需1Mb的特定配置下,其在CIFAR-10上的分类准确率达到87.5%。此外,我们还表明,该量化编码器可用于逐图像块压缩,而重建可由专用的全帧解码器远程执行。该方案通常可实现几乎没有任何块效应的端到端压缩,优于采用块恒定比特率的基于块的最先进技术。

英文摘要:

Even if Application-Specific Integrated Circuits (ASIC) have proven to be a relevant choice for integrating inference at the edge, they are often limited in terms of applicability. In this paper, we demonstrate that an ASIC neural network accelerator dedicated to image processing can be applied to multiple tasks of different levels: image classification and compression, while requiring a very limited hardware. The key component is a reconfigurable, mixed-precision (3b/2b/1b) encoder that takes advantage of proper weight and activation quantizations combined with convolutional layer structural pruning to lower hardware-related constraints (memory and computing). We introduce an automatic adaptation of linear symmetric quantizer scaling factors to perform quantized levels equalization, aiming at stabilizing quinary and ternary weights training. In addition, a proposed layer-shared Bit-Shift Normalization significantly simplifies the implementation of the hardware-expensive Batch Normalization. For a specific configuration in which the encoder design only requires 1Mb, the classification accuracy reaches 87.5% on CIFAR-10. Besides, we also show that this quantized encoder can be used to compress image patch-by-patch while the reconstruction can performed remotely, by a dedicated full-frame decoder. This solution typically enables an end-to-end compression almost without any block artifacts, outperforming patch-based state-of-the-art techniques employing a patch-constant bitrate.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑