Vision Transformers Need Registers

By Timothée Darcet, Maxime Oquab, Maxime Oquab, Julien Mairal, Piotr Bojanowski. International Conference on Learning Representations, 2023

917 citations recorded by the source corpus.

Transformers have recently emerged as a powerful tool for learning visual representations. In this paper, we identify and characterize artifacts in feature maps of both supervised and self-supervised ViT networks. The artifacts correspond to high-norm tokens appearing during inference primarily in low-informative background areas of images, that are repurposed for internal computations. We propose a simple yet effective solution based on providing additional tokens to the input sequence of the Vision Transformer to fill that role. We show that this solution fixes that problem entirely for both supervised and self-supervised models, sets a new state of the art for self-supervised visual models on dense visual prediction tasks, enables object discovery methods with larger models, and most importantly leads to smoother feature maps and attention maps for downstream visual processing.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/10bd38673951f5d7729568284093cbd80482ab16

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Matthieu Cord
  2. Christian Rupprecht
  3. Sanjiv Kumar
  4. Brandon Duderstadt
  5. Andrea Vedaldi
  6. Zach Nussbaum
  7. Ignacio Rocco
  8. Victor Sanh

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Andrea Vedaldi
  3. Dacheng Tao
  4. Xiaogang Wang
  5. Andrew Zisserman
  6. Jitendra Malik
  7. Trevor Darrell
  8. Ming-Hsuan Yang

Citations in the corpus