Patch-level Representation Learning for Self-supervised Vision Transformers
Authors: Sukmin Yun, Hankook Lee, Jaehyung Kim, Jinwoo Shin
AI-generated Key Points
- Recent self-supervised learning (SSL) methods have made significant progress in learning visual representations from unlabeled images.
- Leveraging the architectural advantages of the underlying neural network can further improve the performance of SSL methods.
- Vision Transformers (ViTs) are considered a better architectural choice for SSL, outperforming convolutional networks in various visual tasks.
- ViTs process images as a sequence of disjoint patches and internally process patch-level representations.
- The authors propose a visual pretext task called SelfPatch to learn better patch-level representations using ViTs.
- In SelfPatch, each patch is enforced to be invariant against itself and its neighboring patches, treating similar neighboring patches as positive samples.
- Training ViTs with SelfPatch improves semantically meaningful relations among patches without human-annotated labels.
- The proposed method achieves significant improvements in the performance of existing SSL methods for object detection, instance segmentation, and semantic segmentation tasks.
- When applied to the self-supervised ViT model called DINO, SelfPatch achieves impressive results including +1.3 AP on COCO object detection, +1.2 AP on COCO instance segmentation, and +2.9 mIoU on ADE20K semantic segmentation.
- Considering architectural advantages when designing visual pretext tasks for SSL leads to enhanced patch-level representations and improved performance on downstream visual tasks.
Abstract: Recent self-supervised learning (SSL) methods have shown impressive results in learning visual representations from unlabeled images. This paper aims to improve their performance further by utilizing the architectural advantages of the underlying neural network, as the current state-of-the-art visual pretext tasks for SSL do not enjoy the benefit, i.e., they are architecture-agnostic. In particular, we focus on Vision Transformers (ViTs), which have gained much attention recently as a better architectural choice, often outperforming convolutional networks for various visual tasks. The unique characteristic of ViT is that it takes a sequence of disjoint patches from an image and processes patch-level representations internally. Inspired by this, we design a simple yet effective visual pretext task, coined SelfPatch, for learning better patch-level representations. To be specific, we enforce invariance against each patch and its neighbors, i.e., each patch treats similar neighboring patches as positive samples. Consequently, training ViTs with SelfPatch learns more semantically meaningful relations among patches (without using human-annotated labels), which can be beneficial, in particular, to downstream tasks of a dense prediction type. Despite its simplicity, we demonstrate that it can significantly improve the performance of existing SSL methods for various visual tasks, including object detection and semantic segmentation. Specifically, SelfPatch significantly improves the recent self-supervised ViT, DINO, by achieving +1.3 AP on COCO object detection, +1.2 AP on COCO instance segmentation, and +2.9 mIoU on ADE20K semantic segmentation.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.