English

Visual Transformer for Object Detection

Computer Vision and Pattern Recognition 2022-06-14 v1

Abstract

Convolutional Neural networks (CNN) have been the first choice of paradigm in many computer vision applications. The convolution operation however has a significant weakness which is it only operates on a local neighborhood of pixels, thus it misses global information of the surrounding neighbors. Transformers, or Self-attention networks to be more specific, on the other hand, have emerged as a recent advance to capture long range interactions of the input, but they have mostly been applied to sequence modeling tasks such as Neural Machine Translation, Image captioning and other Natural Language Processing tasks. Transformers has been applied to natural language related tasks and achieved promising results. However, its applications in visual related tasks are far from being satisfying. Taking into consideration of both the weaknesses of Convolutional Neural Networks and those of the Transformers, in this paper, we consider the use of self-attention for discriminative visual tasks, object detection, as an alternative to convolutions. In this paper, we propose our model: DetTransNet. Extensive experiments show that our model leads to consistent improvements in object detection on COCO across many different models and scales, including ResNets, while keeping the number of parameters similar. In particular, our method achieves a 1.2% Average Precision improvement on COCO object detection task over other baseline models.

Keywords

Cite

@article{arxiv.2206.06323,
  title  = {Visual Transformer for Object Detection},
  author = {Michael Yang},
  journal= {arXiv preprint arXiv:2206.06323},
  year   = {2022}
}

Comments

In preparation for short paper of conferences. I am using the name Michael Yang

R2 v1 2026-06-24T11:49:25.544Z