Distilling Self-Supervised Vision Transformers for Weakly-Supervised Few-Shot Classification & Segmentation
AI-generated Key Points
⚠The license of the paper does not allow us to build upon its content and the key points are generated using the paper metadata rather than the full article.
- Paper title: "Distilling Self-Supervised Vision Transformers for Weakly-Supervised Few-Shot Classification & Segmentation"
- Addresses the challenge of weakly-supervised few-shot image classification and segmentation
- Proposes a method that leverages a Vision Transformer (ViT) pretrained with self-supervision
- Uses token representations from the self-supervised ViT and exploits their correlations through self-attention
- Generates separate task heads for classification and segmentation predictions
- Uses attention maps created from tokens as pixel-level pseudo-labels instead of relying on pixel-level labels
- Explores a practical setup called "mixed" supervision, where only a small number of training images have ground truth pixel level labels
- Introduces a pseudo label enhancer to improve pseudo labels in the mixed setup, trained using available ground truth pixel level labels
- Experimental results on Pascal 5i and COCO 20i datasets show significant performance gains across various supervision settings, especially with limited or no pixel level labels available
- Shows promise in addressing weakly supervised few shot image classification and segmentation tasks by leveraging self supervision in Vision Transformers.
Authors: Dahyun Kang, Piotr Koniusz, Minsu Cho, Naila Murray
Abstract: We address the task of weakly-supervised few-shot image classification and segmentation, by leveraging a Vision Transformer (ViT) pretrained with self-supervision. Our proposed method takes token representations from the self-supervised ViT and leverages their correlations, via self-attention, to produce classification and segmentation predictions through separate task heads. Our model is able to effectively learn to perform classification and segmentation in the absence of pixel-level labels during training, using only image-level labels. To do this it uses attention maps, created from tokens generated by the self-supervised ViT backbone, as pixel-level pseudo-labels. We also explore a practical setup with ``mixed" supervision, where a small number of training images contains ground-truth pixel-level labels and the remaining images have only image-level labels. For this mixed setup, we propose to improve the pseudo-labels using a pseudo-label enhancer that was trained using the available ground-truth pixel-level labels. Experiments on Pascal-5i and COCO-20i demonstrate significant performance gains in a variety of supervision settings, and in particular when little-to-no pixel-level labels are available.
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.
Assess the quality of the AI-generated content by voting
Score: 0
Why do we need votes?
Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.
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)
By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while 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.