CASTing Your Model: Learning to Localize Improves Self-Supervised Representations
Not on the map. Why?
Abstract (arXiv)
Recent advances in self-supervised learning (SSL) have largely closed the gap with supervised ImageNet pretraining. Despite their success these methods have been primarily applied to unlabeled ImageNet images, and show marginal gains when trained on larger sets of uncurated images. We hypothesize that current SSL methods perform best on iconic images, and struggle on complex scene images with many objects. Analyzing contrastive SSL methods shows that they have poor visual grounding and receive poor supervisory signal when trained on scene images. We propose Contrastive Attention-Supervised Tuning(CAST) to overcome these limitations. CAST uses unsupervised saliency maps to intelligently sample crops, and to provide grounding supervision via a Grad-CAM attention loss. Experiments on COCO show that CAST significantly improves the features learned by SSL methods on scene images, and further experiments show that CAST-trained models are more robust to changes in backgrounds.
Key facts. Cited by 63 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Dec 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 13; 2022: 21; 2023: 13; 2024: 9; 2025: 4; 2026: 3.
63 papers.
Sort:
By area:
- 2021Efficient Visual Pretraining with Contrastive Detectioncited by 150
- 2021Unsupervised Object-Level Representation Learning from Scene Imagescited by 72
- 2022Self-Supervised Visual Representation Learning with Semantic Groupingcited by 70
- 2022Crafting Better Contrastive Views for Siamese Representation Learningcited by 67
- 2022Hyperbolic Contrastive Learning for Visual Representations beyond Objectscited by 58
- 2021Revisiting Contrastive Methods for Unsupervised Learning of Visual Representationscited by 48
- 2022Unsupervised Hierarchical Semantic Segmentation with Multiview Cosegmentation and Clustering Transformerscited by 39
- 2021Object-aware Contrastive Learning for Debiased Scene Representationcited by 34
- 2023CLIP-S4: Language-Guided Self-Supervised Semantic Segmentationcited by 34
- 2023Learning Visual Representations via Language-Guided Samplingcited by 32
- 2022Architecture-Agnostic Masked Image Modeling -- From ViT back to CNNcited by 31
- 2022UniVIP: A Unified Framework for Self-Supervised Visual Pre-trainingcited by 30
38 more
- 2024EAGLE: Eigen Aggregation Learning for Object-Centric Unsupervised Semantic Segmentationcited by 27
- 2023Anatomical Invariance Modeling and Semantic Alignment for Self-supervised Learning in 3D Medical Image Analysiscited by 25
- 2021Augmentations in Graph Contrastive Learning: Current Methodological Flaws & Towards Better Practicescited by 24
- 2022Improving Visual Grounding by Encouraging Consistent Gradient-based Explanationscited by 24
- 2021Object-Aware Cropping for Self-Supervised Learningcited by 22
- 2023CrOC: Cross-View Online Clustering for Dense Visual Representation Learningcited by 21
- 2023Boosting Video Object Segmentation via Space-time Correspondence Learningcited by 21
- 2021Look at What I'm Doing: Self-Supervised Spatial Grounding of Narrations in Instructional Videoscited by 18
- 2022Learning Hierarchical Image Segmentation For Recognition and By Recognitioncited by 16
- 2023Learning Common Rationale to Improve Self-Supervised Representation for Fine-Grained Visual Recognition Problemscited by 14
- 2022Semantic-Aware Fine-Grained Correspondencecited by 13
- 2022Exploring the Coordination of Frequency and Attention in Masked Image Modelingcited by 12
- 2023Correlational Image Modeling for Self-Supervised Visual Pre-Trainingcited by 11
- 2024PooDLe: Pooled and dense self-supervised learning from naturalistic videoscited by 11
- 2024Understanding Hyperbolic Metric Learning through Hard Negative Samplingcited by 10
- 2021Exploring Localization for Self-supervised Fine-grained Contrastive Learningcited by 9
- 2021Characterizing and Improving the Robustness of Self-Supervised Learning through Background Augmentationscited by 8
- 2022Sim2Real Object-Centric Keypoint Detection and Descriptioncited by 8
- 2022Dense Siamese Network for Dense Unsupervised Learningcited by 8
- 2022Can domain adaptation make object recognition work for everyone?cited by 8
- 2024GenView: Enhancing View Quality with Pretrained Generative Model for Self-Supervised Learningcited by 8
- 2023Understanding Self-Supervised Features for Learning Unsupervised Instance Segmentationcited by 7
- 2023Mining Gaze for Contrastive Learning toward Computer-Assisted Diagnosiscited by 7
- 2021GenURL: A General Framework for Unsupervised Representation Learningcited by 6
- 2022Multi-Task Learning for Visual Scene Understandingcited by 6
- 2022Towards Efficient and Effective Self-Supervised Learning of Visual Representationscited by 6
- 2022Refine and Represent: Region-to-Object Representation Learningcited by 5
- 2021Semi-weakly Supervised Contrastive Representation Learning for Retinal Fundus Imagescited by 4
- 2023Learning Better Contrastive View from Radiologist's Gazecited by 4
- 2023Mitigating Bias: Enhancing Image Classification by Improving Model Explanationscited by 3
- 2021ExCon: Explanation-driven Supervised Contrastive Learning for Image Classificationcited by 2
- 2021PreViTS: Contrastive Pretraining with Video Tracking Supervisioncited by 2
- 2022MSR: Making Self-supervised learning Robust to Aggressive Augmentationscited by 2
- 2023Saliency Guided Contrastive Learning on Scene Imagescited by 2
- 2024On Learning Discriminative Features from Synthesized Data for Self-Supervised Fine-Grained Visual Recognitioncited by 2
- 2024Self-Supervised Learning for Real-World Object Detection: a Surveycited by 2
- 2022CYBORGS: Contrastively Bootstrapping Object Representations by Grounding in Segmentationcited by 1
- 2022On the robustness of self-supervised representations for multi-view object classificationcited by 1
50 most cited of 63 All 63 (JSON)
- 2026PhysAttNet: Enhancing Predictive Performance in Industrial and Astrophysical Time Series via Physics-Informed Attention7 Aug 2026
- 2026Lesion Detection in CT with Frozen Self-Distilled Features: SALT, a Spatially Adaptive Label-Guided Temperature5 Aug 2026
- 2026Hierarchical, Interpretable, Label-Free Concept Bottleneck Model2 Apr 2026
- 2025GenView++: Unifying Adaptive Generative Augmentation and Quality-Driven Supervision for Contrastive Representation Learning28 Sep 2025
- 2025Beyond Instance Consistency: Investigating View Diversity in Self-supervised Learning14 Sep 2025
- 2025Component Adaptive Clustering for Generalized Category Discovery2 Jul 2025
- 2025Multiple Object Stitching for Unsupervised Representation Learning9 Jun 2025
- 2024PP-SSL : Priority-Perception Self-Supervised Learning for Fine-Grained Recognition28 Nov 2024
- 2024CNN Explainability with Multivector Tucker Saliency Maps for Self-Supervised Models30 Oct 2024
- 2024Self-Supervised Learning for Real-World Object Detection: a Survey9 Oct 2024
- 2024PooDLe: Pooled and dense self-supervised learning from naturalistic videos20 Aug 2024
- 2024On Learning Discriminative Features from Synthesized Data for Self-Supervised Fine-Grained Visual Recognition19 Jul 2024
38 more
- 2024Understanding Hyperbolic Metric Learning through Hard Negative Sampling23 Apr 2024
- 2024GenView: Enhancing View Quality with Pretrained Generative Model for Self-Supervised Learning18 Mar 2024
- 2024EAGLE: Eigen Aggregation Learning for Object-Centric Unsupervised Semantic Segmentation3 Mar 2024
- 2024Attention to detail: inter-resolution knowledge distillation11 Jan 2024
- 2023Mining Gaze for Contrastive Learning toward Computer-Assisted Diagnosis11 Dec 2023
- 2023Understanding Self-Supervised Features for Learning Unsupervised Instance Segmentation24 Nov 2023
- 2023Mitigating Bias: Enhancing Image Classification by Improving Model Explanations4 Jul 2023
- 2023Learning with Difference Attention for Visually Grounded Self-supervised Representations26 Jun 2023
- 2023Learning Better Contrastive View from Radiologist's Gaze15 May 2023
- 2023CLIP-S4: Language-Guided Self-Supervised Semantic Segmentation1 May 2023
- 2023Boosting Video Object Segmentation via Space-time Correspondence Learning13 Apr 2023
- 2023CrOC: Cross-View Online Clustering for Dense Visual Representation Learning23 Mar 2023
- 2023Correlational Image Modeling for Self-Supervised Visual Pre-Training22 Mar 2023
- 2023Learning Common Rationale to Improve Self-Supervised Representation for Fine-Grained Visual Recognition Problems3 Mar 2023
- 2023Learning Visual Representations via Language-Guided Sampling23 Feb 2023
- 2023Saliency Guided Contrastive Learning on Scene Images22 Feb 2023
- 2023Anatomical Invariance Modeling and Semantic Alignment for Self-supervised Learning in 3D Medical Image Analysis11 Feb 2023
- 2022Contrastive View Design Strategies to Enhance Robustness to Domain Shifts in Downstream Object Detection9 Dec 2022
- 2022Hyperbolic Contrastive Learning for Visual Representations beyond Objects1 Dec 2022
- 2022Exploring the Coordination of Frequency and Attention in Masked Image Modeling28 Nov 2022
- 2022Copy-Pasting Coherent Depth Regions Improves Contrastive Learning for Urban-Scene Segmentation25 Nov 2022
- 2022Towards Efficient and Effective Self-Supervised Learning of Visual Representations18 Oct 2022
- 2022Learning Hierarchical Image Segmentation For Recognition and By Recognition1 Oct 2022
- 2022Refine and Represent: Region-to-Object Representation Learning25 Aug 2022
- 2022On the robustness of self-supervised representations for multi-view object classification27 Jul 2022
- 2022Semantic-Aware Fine-Grained Correspondence21 Jul 2022
- 2022Improving Visual Grounding by Encouraging Consistent Gradient-based Explanations30 Jun 2022
- 2022MSR: Making Self-supervised learning Robust to Aggressive Augmentations4 Jun 2022
- 2022Self-Supervised Visual Representation Learning with Semantic Grouping30 May 2022
- 2022Architecture-Agnostic Masked Image Modeling -- From ViT back to CNN27 May 2022
- 2022Unsupervised Hierarchical Semantic Segmentation with Multiview Cosegmentation and Clustering Transformers25 Apr 2022
- 2022Can domain adaptation make object recognition work for everyone?23 Apr 2022
- 2022Multi-Task Learning for Visual Scene Understanding28 Mar 2022
- 2022Dense Siamese Network for Dense Unsupervised Learning21 Mar 2022
- 2022CYBORGS: Contrastively Bootstrapping Object Representations by Grounding in Segmentation17 Mar 2022
- 2022UniVIP: A Unified Framework for Self-Supervised Visual Pre-training14 Mar 2022
- 2022Crafting Better Contrastive Views for Siamese Representation Learning7 Feb 2022
- 2022Sim2Real Object-Centric Keypoint Detection and Description1 Feb 2022
50 newest of 63 All 63 (JSON)
- 2021Efficient Visual Pretraining with Contrastive Detectioncited by 150
- 2021Unsupervised Object-Level Representation Learning from Scene Imagescited by 72
- 2022Self-Supervised Visual Representation Learning with Semantic Groupingcited by 70
- 2022Crafting Better Contrastive Views for Siamese Representation Learningcited by 67
- 2022Hyperbolic Contrastive Learning for Visual Representations beyond Objectscited by 58
- 2021Revisiting Contrastive Methods for Unsupervised Learning of Visual Representationscited by 48
- 2022Unsupervised Hierarchical Semantic Segmentation with Multiview Cosegmentation and Clustering Transformerscited by 39
- 2021Object-aware Contrastive Learning for Debiased Scene Representationcited by 34
- 2023Learning Visual Representations via Language-Guided Samplingcited by 32
- 2022Architecture-Agnostic Masked Image Modeling -- From ViT back to CNNcited by 31
- 2022UniVIP: A Unified Framework for Self-Supervised Visual Pre-trainingcited by 30
- 2024EAGLE: Eigen Aggregation Learning for Object-Centric Unsupervised Semantic Segmentationcited by 27
12 most cited of 61 All 61 (JSON)
- 2026PhysAttNet: Enhancing Predictive Performance in Industrial and Astrophysical Time Series via Physics-Informed Attention7 Aug 2026
- 2026Lesion Detection in CT with Frozen Self-Distilled Features: SALT, a Spatially Adaptive Label-Guided Temperature5 Aug 2026
- 2026Hierarchical, Interpretable, Label-Free Concept Bottleneck Model2 Apr 2026
- 2025GenView++: Unifying Adaptive Generative Augmentation and Quality-Driven Supervision for Contrastive Representation Learning28 Sep 2025
- 2025Beyond Instance Consistency: Investigating View Diversity in Self-supervised Learning14 Sep 2025
- 2025Component Adaptive Clustering for Generalized Category Discovery2 Jul 2025
- 2025Multiple Object Stitching for Unsupervised Representation Learning9 Jun 2025
- 2024PP-SSL : Priority-Perception Self-Supervised Learning for Fine-Grained Recognition28 Nov 2024
- 2024CNN Explainability with Multivector Tucker Saliency Maps for Self-Supervised Models30 Oct 2024
- 2024Self-Supervised Learning for Real-World Object Detection: a Survey9 Oct 2024
- 2024PooDLe: Pooled and dense self-supervised learning from naturalistic videos20 Aug 2024
- 2024On Learning Discriminative Features from Synthesized Data for Self-Supervised Fine-Grained Visual Recognition19 Jul 2024
12 newest of 61 All 61 (JSON)