Positive Unlabeled Contrastive Learning

Anish Acharya, S Sanghavi, L Jing, B Bhushanam, D Choudhary, M Rabbat, I Dhillon

International Conference on Machine Learning (ICML), 2022 · Principles of Distribution Shift (PODS) Workshop

Extends contrastive pretraining to positive-unlabeled data with an unbiased, lower-variance loss that uses the few labeled positives.

ProceedingsarXiv

Extended version: Understanding Contrastive Representation Learning from Positive Unlabeled (PU) Data — arXiv:2402.06038 · Short version: Simons Institute, UC Berkeley — Data-Driven Decision Processes Workshop

All publications · Research