Contrastive loss in self-supervised learning is the objective that pulls embeddings of positive pairs together while pushing embeddings of negatives apart in representation space - it builds discriminative features by explicitly teaching what should match and what should remain separate.
What Is Contrastive Loss?
- Definition: A metric-learning objective such as InfoNCE applied to augmented views of images.
- Positive Pair: Two views of the same source image.
- Negative Pair: Views from different images in the batch or memory bank.
- Optimization Target: Maximize similarity for positives and relative margin against negatives.
Why Contrastive Loss Matters
- Discriminative Embeddings: Produces strong instance-level separation.
- Retrieval Strength: Excellent for nearest-neighbor search and metric tasks.
- Theoretical Clarity: Objective directly encodes separation constraints.
- Wide Adoption: Foundation of many influential self-supervised methods.
- Transfer Performance: Strong linear probe results when trained with adequate negatives.
How Contrastive Training Works
Step 1:
- Generate two or more augmentations per image and encode all views.
- Normalize embeddings and compute pairwise similarity matrix.
Step 2:
- Apply InfoNCE-style loss where each anchor selects one positive and many negatives.
- Use temperature scaling to control hardness of similarity discrimination.
Practical Guidance
- Batch Size: Larger effective negative pool usually improves results.
- Memory Banks: Queues can extend negative count when batch is limited.
- Augmentations: Strong and diverse transforms are required to avoid shortcut matching.
Contrastive loss in self-supervised learning is a direct and effective way to shape representation geometry through attraction and repulsion forces - its success depends on careful management of negatives, temperature, and augmentation strength.
Explore 500+ Semiconductor & AI Topics
From EUV lithography to CUDA optimization — search the full knowledge base or chat with our AI assistant.