miCSE: Mutual Information Contrastive Learning for Low-shot Sentence Embeddings
Tassilo Klein, Moin Nabi
Abstract
This paper presents miCSE, a mutual information-based contrastive learning framework that significantly advances the stateof-the-art in few-shot sentence embedding. The proposed approach imposes alignment between the attention pattern of different views during contrastive learning. Learning sentence embeddings with miCSE entails enforcing the structural consistency across augmented views for every sentence, making contrastive self-supervised learning more sample efficient. As a result, the proposed approach shows strong performance in the few-shot learning domain. While it achieves superior results compared to state-of-the-art methods on multiple benchmarks in few-shot learning, it is comparable in the full-shot scenario. This study opens up avenues for efficient self-supervised learning methods that are more robust than current contrastive methods for sentence embedding. 1
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 papers2
- Headless Language Models: Learning without Predicting with Contrastive Weight TyingNathan Godey, Éric Villemonte de la Clergerie, Benoît SagotICLR 2024 · 5 citations
- Robust Similarity Learning with Difference Alignment RegularizationShuo Chen, Gang Niu, Chen Gong, Okan Koc et al.ICLR 2024
Builds on17
- SimCSE: Simple Contrastive Learning of Sentence EmbeddingsTianyu Gao, Xingcheng Yao, Danqi ChenEMNLP 2021 · 2,496 citations
- On the Stability of Fine-tuning BERT: Misconceptions, Explanations, and Strong BaselinesMarius Mosbach, Maksym Andriushchenko, Dietrich KlakowICLR 2021 · 448 citations
- MixText: Linguistically-Informed Interpolation of Hidden Space for Semi-Supervised Text ClassificationJiaao Chen, Zichao Yang, Diyi YangACL 2020 · 340 citations
- Debiased Contrastive Learning of Unsupervised Sentence RepresentationsKun Zhou, Beichen Zhang, Wayne Xin Zhao, Ji-Rong WenACL 2022 · 128 citations
- An Unsupervised Sentence Embedding Method by Mutual Information MaximizationYan Zhang, Ruidan He, Zuozhu Liu, Kwan Hui Lim et al.EMNLP 2020 · 126 citations
Related papers
- Contrastive Learning of Sentence Embeddings from ScratchJunlei Zhang, Zhenzhong Lan, Junxian HeEMNLP 2023 · 9 citations
- English Contrastive Learning Can Learn Universal Cross-lingual Sentence EmbeddingsYau-Shian Wang, Ashley Wu, Graham NeubigEMNLP 2022 · 18 citations
- WhitenedCSE: Whitening-based Contrastive Learning of Sentence EmbeddingsWenjie Zhuo, Yifan Sun, Xiaohan Wang, Linchao Zhu et al.ACL 2023 · 17 citations
- Self-Guided Contrastive Learning for BERT Sentence RepresentationsTaeuk Kim, Kang Min Yoo, Sang-goo LeeACL 2021
- TNCSE: Tensor Norm Constraints for Unsupervised Contrastive Learning of Sentence EmbeddingsTianyu Zong, Bingkang Shi, Hongzhu Yi, Jungang XuAAAI 2025 · 2 citations
