arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.09315cs.CV

End-to-End Object Detection with Adaptive Clustering Transformer

Peking University(北京大学);Shanghai AI Laboratory(上海人工智能实验室);CUHK-SenseTime Joint Laboratory(香港中文大学-商汤科技联合实验室);The Chinese University of Hong Kong(香港中文大学)

Minghang Zheng, Peng Gao, Renrui Zhang, Kunchang Li, Xiaogang Wang, Hongsheng Li, Hao Dong

英文摘要:

End-to-end Object Detection with Transformer (DETR)proposes to perform object detection with Transformer and achieve comparable performance with two-stage object detection like Faster-RCNN. However, DETR needs huge computational resources for training and inference due to the high-resolution spatial input. In this paper, a novel variant of transformer named Adaptive Clustering Transformer(ACT) has been proposed to reduce the computation cost for high-resolution input. ACT cluster the query features adaptively using Locality Sensitive Hashing (LSH) and ap-proximate the query-key interaction using the prototype-key interaction. ACT can reduce the quadratic O(N2) complexity inside self-attention into O(NK) where K is the number of prototypes in each layer. ACT can be a drop-in module replacing the original self-attention module without any training. ACT achieves a good balance between accuracy and computation cost (FLOPs). The code is available as supplementary for the ease of experiment replication and verification. Code is released at \url{https://github.com/gaopengcuhk/SMCA-DETR/}

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

↑