Contrastive Learning with Boosted Memorization
Zhihan Zhou, Jiangchao Yao, Yanfeng Wang, Bo Han, Ya Zhang
Abstract
Self-supervised learning has achieved a great success in the representation learning of visual and textual data. However, the current methods are mainly validated on the well-curated datasets, which do not exhibit the real-world long-tailed distribution. Recent attempts to consider self-supervised long-tailed learning are made by rebalancing in the loss perspective or the model perspective, resembling the paradigms in the supervised long-tailed learning. Nevertheless, without the aid of labels, these explorations have not shown the expected significant promise due to the limitation in tail sample discovery or the heuristic structure design. Different from previous works, we explore this direction from an alternative perspective, i.e., the data perspective, and propose a novel Boosted Contrastive Learning (BCL) method. Specifically, BCL leverages the memorization effect of deep neural networks to automatically drive the information discrepancy of the sample views in contrastive learning, which is more efficient to enhance the long-tailed learning in the label-unaware context. Extensive experiments on a range of benchmark datasets demonstrate the effectiveness of BCL over several state-of-the-art methods. Our code is available at https://github.com/MediaBrain-SJTU/BCL.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers18
- How Re-sampling Helps for Long-Tail Learning?Jiang-Xin Shi, Tong Wei, Yuke Xiang, Yufeng LiNeurIPS 2023 · 84 citations
- Uncovering Prototypical Knowledge for Weakly Open-Vocabulary Semantic SegmentationFei Zhang, Tianfei Zhou, Boyang Li, Hao He et al.NeurIPS 2023 · 46 citations
- Federated Learning with Bilateral Curation for Partially Class-Disjoint DataZiqing Fan, Ruipeng Zhang, Jiangchao Yao, Bo Han et al.NeurIPS 2023 · 28 citations
- Towards Distribution-Agnostic Generalized Category DiscoveryJianhong Bai, Zuozhu Liu, Hualiang Wang, Ruizhe Chen et al.NeurIPS 2023 · 24 citations
- Long-tailed Diffusion Models with Oriented CalibrationTianjiao Zhang, Huangjie Zheng, Jiangchao Yao, Xiangfeng Wang et al.ICLR 2024 · 22 citations
Builds on23
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Bootstrap Your Own Latent - A New Approach to Self-Supervised LearningJean-Bastien Grill, Florian Strub, Florent Altché, Corentin Tallec et al.NeurIPS 2020 · 9,171 citations
- ALBERT: A Lite BERT for Self-supervised Learning of Language RepresentationsZhenzhong Lan, Mingda Chen, Sebastian Goodman, Kevin Gimpel et al.ICLR 2020 · 7,418 citations
- Unsupervised Learning of Visual Features by Contrasting Cluster AssignmentsMathilde Caron, Ishan Misra, Julien Mairal, Priya Goyal et al.NeurIPS 2020 · 5,249 citations
Related papers
- Exploring Balanced Feature Spaces for Representation LearningBingyi Kang, Yu Li, Sa Xie, Zehuan Yuan et al.ICLR 2021 · 296 citations
- Self-Damaging Contrastive LearningZiyu Jiang, Tianlong Chen, Bobak J. Mortazavi, Zhangyang WangICML 2021 · 83 citations
- On the Effectiveness of Out-of-Distribution Data in Self-Supervised Long-Tail LearningJianhong Bai, Zuozhu Liu, Hualiang Wang, Jin Hao et al.ICLR 2023 · 6 citations
- Subclass-balancing Contrastive Learning for Long-tailed RecognitionChengkai Hou, Jieyu Zhang, Haonan Wang, Tianyi ZhouICCV 2023 · 50 citations
- Balanced Contrastive Learning for Long-Tailed Visual RecognitionJianggang Zhu, Zheng Wang, Jingjing Chen, Yi-Ping Phoebe Chen et al.CVPR 2022 · 194 citations
