about
Plain Vision Transformer Backbones for Object Detection (arxiv.org)
3 points by georgehill on Mar 31, 2022 | hide | past | pdf | discuss on HN

In plain words: They show a flat vision network, which reads the whole image at one size, can find objects using a simple multi-scale view and mostly local patch attention. It matches layered networks, scoring 61.3 on the standard object-finding test after pre-training on one image set.

Abstract · Exploring Plain Vision Transformer Backbones for Object Detection

We explore the plain, non-hierarchical Vision Transformer (ViT) as a backbone network for object detection. This design enables the original ViT architecture to be fine-tuned for object detection without needing to redesign a hierarchical backbone for pre-training. With minimal adaptations for fine-tuning, our plain-backbone detector can achieve competitive results. Surprisingly, we observe: (i) it is sufficient to build a simple feature pyramid from a single-scale feature map (without the common FPN design) and (ii) it is sufficient to use window attention (without shifting) aided with very few cross-window propagation blocks. With plain ViT backbones pre-trained as Masked Autoencoders (MAE), our detector, named ViTDet, can compete with the previous leading methods that were all based on hierarchical backbones, reaching up to 61.3 AP_box on the COCO dataset using only ImageNet-1K pre-training. We hope our study will draw attention to research on plain-backbone detectors. Code for ViTDet is available in Detectron2.

Yanghao Li, Hanzi Mao, Ross Girshick, Kaiming He
arXiv:2203.16527 · cs.CV · submitted Mar 30, 2022 · updated Jun 10, 2022
abstract · pdf · html · Tech report. arXiv v2: add RetinaNet results

add comment on HN