Awesome Contrastive Learning Papers&Codes

A comprehensive list of awesome Contrastive Learning Papers&Codes.

Contrastive learning papers at the top conferences, research include, but are not limited to: CV, NLP, Audio, Video, Multimodal, Graph, Language, etc.

Content

Must-read Papers

Survey Papers

1. A Survey on Contrastive Self-supervised Learning.
   Authors:Ashish Jaiswal, Ashwin Ramesh Babu, Mohammad Zaki Zadeh, Debapriya Banerjee, Fillia Makedon. paper

Problems

Computer Vision

1. [PCL] Prototypical Contrastive Learning of Unsupervised Representations. ICLR2021.
   Authors:Junnan Li, Pan Zhou, Caiming Xiong, Steven C.H. Hoi. paper code
2. [BalFeat] Exploring Balanced Feature Spaces for Representation Learning. ICLR2021.
   Authors:Bingyi Kang, Yu Li, Sa Xie, Zehuan Yuan, Jiashi Feng. paper
3. [MiCE] MiCE: Mixture of Contrastive Experts for Unsupervised Image Clustering. ICLR2021.
   Authors:Tsung Wei Tsai, Chongxuan Li, Jun Zhu. paper code
4. [i-Mix] i-Mix: A Strategy for Regularizing Contrastive Representation Learning. ICLR2021.
   Authors:Kibok Lee, Yian Zhu, Kihyuk Sohn, Chun-Liang Li, Jinwoo Shin, Honglak Lee. paper code
5. Contrastive Learning with Hard Negative Samples. ICLR2021.
   Authors:Joshua Robinson, Ching-Yao Chuang, Suvrit Sra, Stefanie Jegelka. paper coder
6. [LooC] What Should Not Be Contrastive in Contrastive Learning. ICLR2021.
   Authors:Tete Xiao, Xiaolong Wang, Alexei A. Efros, Trevor Darrell. paper
7. [MoCo] Momentum Contrast for Unsupervised Visual Representation Learning. CVPR2020.
   Authors:Kaiming He, Haoqi Fan, Yuxin Wu, Saining Xie, Ross Girshick. paper code
8. [MoCo v2] Improved Baselines with Momentum Contrastive Learning. Authors:Xinlei Chen, Haoqi Fan, Ross Girshick, Kaiming He. paper code
9. [SimCLR] A Simple Framework for Contrastive Learning of Visual Representations. ICML2020.
   Authors:Ting Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey Hinton. paper code
10. [SimCLR v2] Big Self-Supervised Models are Strong Semi-Supervised Learners. NIPS2020.
    Authors:Ting Chen, Simon Kornblith, Kevin Swersky, Mohammad Norouzi, Geoffrey Hinton. paper code
11. [BYOL] Bootstrap your own latent: A new approach to self-supervised Learning.
    Authors:Jean-Bastien Grill, Florian Strub, Florent Altché, Corentin Tallec, Pierre H. Richemond, Elena Buchatskaya, Carl Doersch, Bernardo Avila Pires, Zhaohan Daniel Guo, Mohammad Gheshlaghi Azar, Bilal Piot, Koray Kavukcuoglu, Rémi Munos, Michal Valko. paper code
12. [SwAV] Unsupervised Learning of Visual Features by Contrasting Cluster Assignments. NIPS2020.
    Authors:Mathilde Caron, Ishan Misra, Julien Mairal, Priya Goyal, Piotr Bojanowski, Armand Joulin. paper code
13. [SimSiam] Exploring Simple Siamese Representation Learning. CVPR2021.
    Authors:Xinlei Chen, Kaiming He. paper code
14. Hard Negative Mixing for Contrastive Learning. NIPS2020.
    Authors:Yannis Kalantidis, Mert Bulent Sariyildiz, Noe Pion, Philippe Weinzaepfel, Diane Larlus. paper
15. Supervised Contrastive Learning. NIPS2020.
    Authors:Prannay Khosla, Piotr Teterwak, Chen Wang, Aaron Sarna, Yonglong Tian, Phillip Isola, Aaron Maschinot, Ce Liu, Dilip Krishnan. paper
16. [LoCo] LoCo: Local Contrastive Representation Learning. NIPS2020.
    Authors:Yuwen Xiong, Mengye Ren, Raquel Urtasun. paper
17. What Makes for Good Views for Contrastive Learning?. NIPS2020.
    Authors:Yonglong Tian, Chen Sun, Ben Poole, Dilip Krishnan, Cordelia Schmid, Phillip Isola. paper
18. [ContraGAN] ContraGAN: Contrastive Learning for Conditional Image Generation. NIPS2020.
    Authors:Minguk Kang, Jaesik Park. paper code
19. [SpCL] Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID. NIPS2020.
    Authors:Yixiao Ge, Feng Zhu, Dapeng Chen, Rui Zhao, Hongsheng Li. paper code

Audio

1. wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representations.
   Authors:Alexei Baevski, Henry Zhou, Abdelrahman Mohamed, Michael Auli. paper code

Videos and Multimodal

1. Time-Contrastive Networks: Self-Supervised Learning from Video.
   Authors: Pierre Sermanet; Corey Lynch; Yevgen Chebotar; Jasmine Hsu; Eric Jang; Stefan Schaal; Sergey Levine. paper
2. Contrastive Multiview Coding.
   Authors:Yonglong Tian, Dilip Krishnan, Phillip Isola. paper code
3. Learning Video Representations using Contrastive Bidirectional Transformer.
   Authors:Chen Sun, Fabien Baradel, Kevin Murphy, Cordelia Schmid. paper
4. End-to-End Learning of Visual Representations from Uncurated Instructional Videos. CVPR2020.
   Authors:Antoine Miech, Jean-Baptiste Alayrac, Lucas Smaira, Ivan Laptev, Josef Sivic, Andrew Zisserman. paper code
5. Multi-modal Self-Supervision from Generalized Data Transformations.
   Authors:Mandela Patrick, Yuki M. Asano, Polina Kuznetsova, Ruth Fong, João F. Henriques, Geoffrey Zweig, Andrea Vedaldi. paper
6. Support-set bottlenecks for video-text representation learning. ICLR2021.
   Authors:Mandela Patrick, Po-Yao Huang, Yuki Asano, Florian Metze, Alexander Hauptmann, João Henriques, Andrea Vedaldi. paper
7. Contrastive Learning of Medical Visual Representations from Paired Images and Text. ICLR2021.
   Authors:Yuhao Zhang, Hang Jiang, Yasuhide Miura, Christopher D. Manning, Curtis P. Langlotz. paper
8. AVLnet: Learning Audio-Visual Language Representations from Instructional Videos.
   Authors:Andrew Rouditchenko, Angie Boggust, David Harwath, Brian Chen, Dhiraj Joshi, Samuel Thomas, Kartik Audhkhasi, Hilde Kuehne, Rameswar Panda, Rogerio Feris, Brian Kingsbury, Michael Picheny, Antonio Torralba, James Glass. paper
9. Self-Supervised MultiModal Versatile Networks. NIPS2020.
   Authors:Jean-Baptiste Alayrac, Adrià Recasens, Rosalia Schneider, Relja Arandjelović, Jason Ramapuram, Jeffrey De Fauw, Lucas Smaira, Sander Dieleman, Andrew Zisserman. paper
10. Memory-augmented Dense Predictive Coding for Video Representation Learning.
    Authors:Tengda Han, Weidi Xie, Andrew Zisserman. paper code
11. Spatiotemporal Contrastive Video Representation Learning.
    Authors:Rui Qian, Tianjian Meng, Boqing Gong, Ming-Hsuan Yang, Huisheng Wang, Serge Belongie, Yin Cui. paper code
12. Self-supervised Co-training for Video Representation Learning. NIPS2020.
    Authors:Tengda Han, Weidi Xie, Andrew Zisserman. paper

NLP

1. [CALM] Pre-training Text-to-Text Transformers for Concept-centric Common Sense. ICLR2021.
   Authors:Wangchunshu Zhou, Dong-Ho Lee, Ravi Kiran Selvam, Seyeon Lee, Xiang Ren. paper code
2. Residual Energy-Based Models for Text Generation. ICLR2021.
   Authors:Yuntian Deng, Anton Bakhtin, Myle Ott, Arthur Szlam, Marc'Aurelio Ranzato. paper
3. Contrastive Learning with Adversarial Perturbations for Conditional Text Generation. ICLR2021.
   Authors:Seanie Lee, Dong Bok Lee, Sung Ju Hwang. paper
4. [CoDA] CoDA: Contrast-enhanced and Diversity-promoting Data Augmentation for Natural Language Understanding. ICLR2021.
   Authors:Yanru Qu, Dinghan Shen, Yelong Shen, Sandra Sajeev, Jiawei Han, Weizhu Chen. paper
5. [FairFil] FairFil: Contrastive Neural Debiasing Method for Pretrained Text Encoders. ICLR2021.
   Authors:Pengyu Cheng, Weituo Hao, Siyang Yuan, Shijing Si, Lawrence Carin. paper
6. Towards Robust and Efficient Contrastive Textual Representation Learning. ICLR2021.
   Authors:Liqun Chen, Yizhe Zhang, Dianqi Li, Chenyang Tao, Dong Wang, Lawrence Carin. paper
7. Self-supervised Contrastive Zero to Few-shot Learning from Small, Long-tailed Text data. ICLR2021.
   Authors:Nils Rethmeier, Isabelle Augenstein. paper
8. Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrieval. ICLR2021.
   Authors:Lee Xiong, Chenyan Xiong, Ye Li, Kwok-Fung Tang, Jialin Liu, Paul Bennett, Junaid Ahmed, Arnold Overwijk. paper
9. Self-Supervised Contrastive Learning for Efficient User Satisfaction Prediction in Conversational Agents. NAACL2021.
   Authors:Mohammad Kachuee, Hao Yuan, Young-Bum Kim, Sungjin Lee. paper
10. SOrT-ing VQA Models : Contrastive Gradient Learning for Improved Consistency. NAACL2021.
    Authors:Sameer Dharur, Purva Tendulkar, Dhruv Batra, Devi Parikh, Ramprasaath R. Selvaraju. paper
11. Supporting Clustering with Contrastive Learning. NAACL2021.
    Authors:Dejiao Zhang, Feng Nan, Xiaokai Wei, Shangwen Li, Henghui Zhu, Kathleen McKeown, Ramesh Nallapati, Andrew Arnold, Bing Xiang. paper
12. Understanding Hard Negatives in Noise Contrastive Estimation. NAACL2021.
    Authors:Wenzheng Zhang, Karl Stratos. paper
13. Contextualized and Generalized Sentence Representations by Contrastive Self-Supervised Learning: A Case Study on Discourse Relation Analysis. NAACL2021.
    Authors:Hirokazu Kiyomaru, Sadao Kurohashi. paper
14. Fine-Tuning Pre-trained Language Model with Weak Supervision: A Contrastive-Regularized Self-Training Approach. NAACL2021.
    Authors:Yue Yu, Simiao Zuo, Haoming Jiang, Wendi Ren, Tuo Zhao, Chao Zhang. paper

Language Contrastive Learning

1. Distributed Representations of Words and Phrases and their Compositionality. 2013NIPS.
   Authors:Tomas Mikolov, Ilya Sutskever, Kai Chen, Greg Corrado, Jeffrey Dean. Paper

2. An efficient framework for learning sentence representations.
   Authors:Lajanugen Logeswaran, Honglak Lee. Paper

3. XLNet: Generalized Autoregressive Pretraining for Language Understanding.
   Authors:Zhilin Yang, Zihang Dai, Yiming Yang, Jaime Carbonell, Ruslan Salakhutdinov, Quoc V. Le. Paper

4. A Mutual Information Maximization Perspective of Language Representation Learning.
   Authors:Lingpeng Kong, Cyprien de Masson d'Autume, Wang Ling, Lei Yu, Zihang Dai, Dani Yogatama. Paper

5. InfoXLM: An Information-Theoretic Framework for Cross-Lingual Language Model Pre-Training.
   Authors:Zewen Chi, Li Dong, Furu Wei, Nan Yang, Saksham Singhal, Wenhui Wang, Xia Song, Xian-Ling Mao, Heyan Huang, Ming Zhou. Paper

Graph

1. [GraphCL] Graph Contrastive Learning with Augmentations. NIPS2020.
   Authors:Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen. paper
2. Contrastive Multi-View Representation Learning on Graphs. ICML2020.
   Authors:Kaveh Hassani, Amir Hosein Khasahmadi. Paper
3. [GCC] GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training. KDD2020.
   Authors:Jiezhong Qiu, Qibin Chen, Yuxiao Dong, Jing Zhang, Hongxia Yang, Ming Ding, Kuansan Wang, Jie Tang. Paper
4. [InfoGraph] InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization. ICLR2020.
   Authors:Fan-Yun Sun, Jordan Hoffmann, Vikas Verma, Jian Tang. Paper

Adversarial Learning

1. Contrastive Learning with Adversarial Examples. NIPS2020.
   Authors:Chih-Hui Ho, Nuno Vasconcelos. paper

Recommendation

1. Self-Supervised Hypergraph Convolutional Networks for Session-based Recommendation. AAAI2021.
   Authors:Xin Xia, Hongzhi Yin, Junliang Yu, Qinyong Wang, Lizhen Cui, Xiangliang Zhang. paper code
2. Self-Supervised Multi-Channel Hypergraph Convolutional Network for Social Recommendation. https://arxiv.org/abs/2101.06448 Yu, Hongzhi Yin, Jundong Li, Qinyong Wang, Nguyen Quoc Viet Hung, Xiangliang Zhang. paper code
3. Self-supervised Graph Learning for Recommendation. SIGIR2021.
   Authors:Jiancan Wu, Xiang Wang, Fuli Feng, Xiangnan He, Liang Chen, Jianxun Lian, Xing Xie. paper code

Applications

1. Contrastive Learning for Unpaired Image-to-Image Translation.
   Authors:Taesung ParkAlexei A. Efros, Richard ZhangJun-Yan Zhu. paper

About Me

你好，我是对白。目前是一名大厂算法工程师，也是一位本科独立创业并拿过数百万融资的小哥哥。日常在知乎和微信分享算法知识、创业心得和人生感悟。
欢迎关注我的微信公众号，后台回复关键词：对比学习，即可获取对比学习最新论文合集（包含以上所有论文哦~）