DN-DETR: Accelerate DETR Training by Introducing Query DeNoising

• Published in: Proceedings (IEEE Computer Society Conference on Computer Vision and Pattern Recognition. Online) pp. 13609 - 13617
• Main Authors: Li, Feng, Zhang, Hao, Liu, Shilong, Guo, Jian, Ni, Lionel M., Zhang, Lei
• Format: Conference Proceeding
• Language: English
• Published: IEEE 01.06.2022
• Subjects: categorization; Codes; Decoding; Machine vision; Noise reduction; Pattern recognition; Recognition: detection; retrieval; Machine learning; Vision applications and systems; Training; Transformers
• ISSN: 1063-6919
• DOI: 10.1109/CVPR52688.2022.01325

We present in this paper a novel denoising training method to speedup DETR (DEtection TRansformer) training and offer a deepened understanding of the slow convergence issue of DETR-like methods. We show that the slow convergence results from the instability of bipartite graph matching which causes inconsistent optimization goals in early training stages. To address this issue, except for the Hungarian loss, our method additionally feeds ground-truth bounding boxes with noises into Transformer decoder and trains the model to reconstruct the original boxes, which effectively reduces the bipartite graph matching difficulty and leads to a faster convergence. Our method is universal and can be easily plugged into any DETR-like methods by adding dozens of lines of code to achieve a remarkable improvement. As a result, our DN-DETR results in a remarkable improvement (+1.9AP) under the same setting and achieves the best result (AP 43.4 and 48.6 with 12 and 50 epochs of training respectively) among DETR-like methods with ResNet-50 backbone. Compared with the baseline under the same setting, DN-DETR achieves comparable performance with 50% training epochs. Code is available at https://github.com/FengLi-ust/DN-DETR.