arXiv:2412.14633cs.CVcs.AI2024-12被引 1

提出渐进式粗粒度重建方法,显著提升视觉Transformer低比特量化精度。

Progressive Fine-to-Coarse Reconstruction for Accurate Low-Bit Post-Training Quantization in Vision Transformers

  • 分层重建:从注意力与前馈模块开始,逐步合并为粗粒度块进行迭代重建。
  • 3比特量化下ViT-B在ImageNet上达75.61%准确率,优于现有方法。
  • 适合需要高效压缩视觉模型的研究者,尤其关注低比特部署场景。

由于高效性,后训练量化(PTQ)被广泛用于压缩视觉变换器(ViTs)。然而,在低比特表示下,其性能常显著低于全精度模型。为解决此问题,已有研究在PTQ框架中引入重建方法以提升低比特量化性能。但现有方法预设重建粒度,未探索不同粒度间的渐进关系,导致ViTs量化效果不优。为此,本文提出渐进式细到粗重建(PFCR)方法,显著提升低比特量化视觉变换器的性能。具体而言,将多头自注意力和多层感知机模块及其残差连接定义为最细粒度重建单元;先重建这两个细粒度单元,再组合形成更粗块,并在更粗粒度层级重建。通过迭代组合与重建,实现渐进式细到粗重建。此外,引入渐进优化策略(POS)缓解训练难度,进一步提升模型性能。ImageNet实验表明,所提方法在同类方法中取得最佳Top-1准确率,尤其在3比特量化下ViT-B达到75.61%。此外,COCO数据集上的量化结果验证了该方法在目标检测与实例分割等任务中的有效性与泛化能力。

原文摘要 · Abstract (English)

Due to its efficiency, Post-Training Quantization (PTQ) has been widely adopted for compressing Vision Transformers (ViTs). However, when quantized into low-bit representations, there is often a significant performance drop compared to their full-precision counterparts. To address this issue, reconstruction methods have been incorporated into the PTQ framework to improve performance in low-bit quantization settings. Nevertheless, existing related methods predefine the reconstruction granularity and seldom explore the progressive relationships between different reconstruction granularities, which leads to sub-optimal quantization results in ViTs. To this end, in this paper, we propose a Progressive Fine-to-Coarse Reconstruction (PFCR) method for accurate PTQ, which significantly improves the performance of low-bit quantized vision transformers. Specifically, we define multi-head self-attention and multi-layer perceptron modules along with their shortcuts as the finest reconstruction units. After reconstructing these two fine-grained units, we combine them to form coarser blocks and reconstruct them at a coarser granularity level. We iteratively perform this combination and reconstruction process, achieving progressive fine-to-coarse reconstruction. Additionally, we introduce a Progressive Optimization Strategy (POS) for PFCR to alleviate the difficulty of training, thereby further enhancing model performance. Experimental results on the ImageNet dataset demonstrate that our proposed method achieves the best Top-1 accuracy among state-of-the-art methods, particularly attaining 75.61% for 3-bit quantized ViT-B in PTQ. Besides, quantization results on the COCO dataset reveal the effectiveness and generalization of our proposed method on other computer vision tasks like object detection and instance segmentation.

量化视觉TransformerPTQ渐进重建

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