跳到主要导航 跳到搜索 跳到主要内容

AQ-DETR: Low-Bit Quantized Detection Transformer with Auxiliary Queries

  • Runqi Wang
  • , Huixin Sun
  • , Linlin Yang*
  • , Shaohui Lin
  • , Chuanjian Liu
  • , Yan Gao
  • , Yao Hu
  • , Baochang Zhang
  • *此作品的通讯作者
  • Beihang University
  • Communication University of China
  • Huawei Technologies Co., Ltd.
  • Xiaohongshu
  • Zhongguancun Laboratory
  • Nanchang Institute of Technology

科研成果: 期刊稿件会议文章同行评审

摘要

DEtection TRansformer (DETR) and its variants have achieved remarkable performance. However, they are accompanied by a large computation overhead cost, which significantly prevents their applications on resource-limited devices. Prior arts attempt to reduce the computational burden of DETR using low-bit quantization, while these methods sacrifice a severe significant performance on weight-activation-attention low-bit quantization. We observe that the number of matching queries and positive samples affects much on the representation capacity of queries in DETR, while quantifying queries of DETR further reduces its representational capacity, thus leading to a severe performance drop. We introduce a new quantization strategy based on Auxiliary Queries for DETR (AQ-DETR), aiming to enhance the capacity of quantized queries. In addition, a layer-by-layer distillation is proposed to reduce the quantization error between quantized attention and full-precision counterpart. Through our extensive experiments on large-scale open datasets, the performance of the 4-bit quantization of DETR and Deformable DETR models is comparable to full-precision counterparts.

源语言英语
页(从-至)15598-15606
页数9
期刊Proceedings of the AAAI Conference on Artificial Intelligence
38
14
DOI
出版状态已出版 - 25 3月 2024
活动38th AAAI Conference on Artificial Intelligence, AAAI 2024 - Vancouver, 加拿大
期限: 20 2月 202427 2月 2024

指纹

探究 'AQ-DETR: Low-Bit Quantized Detection Transformer with Auxiliary Queries' 的科研主题。它们共同构成独一无二的指纹。

引用此