English

Sparse DETR: Efficient End-to-End Object Detection with Learnable Sparsity

Computer Vision and Pattern Recognition 2022-03-07 v2 Machine Learning

Abstract

DETR is the first end-to-end object detector using a transformer encoder-decoder architecture and demonstrates competitive performance but low computational efficiency on high resolution feature maps. The subsequent work, Deformable DETR, enhances the efficiency of DETR by replacing dense attention with deformable attention, which achieves 10x faster convergence and improved performance. Deformable DETR uses the multiscale feature to ameliorate performance, however, the number of encoder tokens increases by 20x compared to DETR, and the computation cost of the encoder attention remains a bottleneck. In our preliminary experiment, we observe that the detection performance hardly deteriorates even if only a part of the encoder token is updated. Inspired by this observation, we propose Sparse DETR that selectively updates only the tokens expected to be referenced by the decoder, thus help the model effectively detect objects. In addition, we show that applying an auxiliary detection loss on the selected tokens in the encoder improves the performance while minimizing computational overhead. We validate that Sparse DETR achieves better performance than Deformable DETR even with only 10% encoder tokens on the COCO dataset. Albeit only the encoder tokens are sparsified, the total computation cost decreases by 38% and the frames per second (FPS) increases by 42% compared to Deformable DETR. Code is available at https://github.com/kakaobrain/sparse-detr

Keywords

Cite

@article{arxiv.2111.14330,
  title  = {Sparse DETR: Efficient End-to-End Object Detection with Learnable Sparsity},
  author = {Byungseok Roh and JaeWoong Shin and Wuhyun Shin and Saehoon Kim},
  journal= {arXiv preprint arXiv:2111.14330},
  year   = {2022}
}

Comments

ICLR 2022. Code is available at https://github.com/kakaobrain/sparse-detr

R2 v1 2026-06-24T07:55:12.645Z