Max-Margin Contrastive Learning

    •  Shah, A., Sra, S., Chellappa, R., Cherian, A., "Max-Margin Contrastive Learning", AAAI Conference on Artificial Intelligence, February 2022.
      BibTeX TR2022-013 PDF
      • @inproceedings{Shah2022feb,
      • author = {Shah, Anshul and Sra, Suvrit and Chellappa, Rama and Cherian, Anoop},
      • title = {Max-Margin Contrastive Learning},
      • booktitle = {AAAI Conference on Artificial Intelligence},
      • year = 2022,
      • month = feb,
      • url = {}
      • }
  • MERL Contact:
  • Research Areas:

    Artificial Intelligence, Computer Vision, Machine Learning


Standard contrastive learning approaches usually require a large number of negatives for effective unsupervised learning and often have slow convergence. We suspect this is due to the sub-optimal selection of negatives that offer the most contrastiveness to the positives. To this end, we present max-margin contrastive learning (MMCL), inspired by support vector machines (SVM). We select useful negatives as the sparse support vectors via solving a quadratic optimization problem, and contrastiveness is enforced by maximizing the decision margin. As SVM optimization can be computationally demanding, especially in an end-to-end learning setting, we present simplifications to the problem formulation. We provide experiments using our approach on standard vision benchmark datasets, demonstrating better performances in unsupervised representation learning over state of the art, while having better convergence properties.