arXiv:2502.04628cs.CV2025-02AAAI被引 11

针对视觉Transformer量化性能下降问题,提出新方法提升低比特压缩效果。

AIQViT: Architecture-Informed Post-Training Quantization for Vision Transformers

  • 引入架构感知的低秩补偿机制修复权重量化损失
  • 动态聚焦量化器优化注意力后激活分布不均问题
  • 在五类视觉任务中优于现有方法,尤其适合低比特场景

后训练量化(PTQ)是降低视觉变换器(ViTs)存储与计算开销的有前景方案。现有方法多关注处理ViTs特有的激活特性,但普遍低估权重量化带来的信息损失,导致低比特下性能显著下降。此外,对ViTs后Softmax激活采用对数变换的常见做法,反而优先保留零附近信息量较低的值,引入额外冗余,影响量化效果。为此,本文提出专为ViTs设计的创新PTQ方法AIQViT(Architecture-Informed Post-training Quantization for ViTs)。首先,设计架构感知的低秩补偿机制,通过可学习的低秩权重补偿权重量化引起的性能退化;其次,设计动态聚焦量化器,适应后Softmax激活的非均衡分布,动态选择最具价值区间进行更高分辨率量化。在图像分类、目标检测、实例分割、点云分类及点云部件分割五项视觉任务上的大量实验表明,AIQViT优于当前最先进的PTQ方法。

原文摘要 · Abstract (English)

Post-training quantization (PTQ) has emerged as a promising solution for reducing the storage and computational cost of vision transformers (ViTs). Recent advances primarily target at crafting quantizers to deal with peculiar activations characterized by ViTs. However, most existing methods underestimate the information loss incurred by weight quantization, resulting in significant performance deterioration, particularly in low-bit cases. Furthermore, a common practice in quantizing post-Softmax activations of ViTs is to employ logarithmic transformations, which unfortunately prioritize less informative values around zero. This approach introduces additional redundancies, ultimately leading to suboptimal quantization efficacy. To handle these, this paper proposes an innovative PTQ method tailored for ViTs, termed AIQViT (Architecture-Informed Post-training Quantization for ViTs). First, we design an architecture-informed low rank compensation mechanism, wherein learnable low-rank weights are introduced to compensate for the degradation caused by weight quantization. Second, we design a dynamic focusing quantizer to accommodate the unbalanced distribution of post-Softmax activations, which dynamically selects the most valuable interval for higher quantization resolution. Extensive experiments on five vision tasks, including image classification, object detection, instance segmentation, point cloud classification, and point cloud part segmentation, demonstrate the superiority of AIQViT over state-of-the-art PTQ methods.

量化视觉Transformer后训练量化低秩补偿

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