PulseAugur
EN
LIVE 11:17:05

InfoNCE loss generalization in contrastive learning analyzed

A new paper by Nick Whiteley explores the generalization capabilities of similarity search in contrastive learning, specifically focusing on the InfoNCE loss function. The research demonstrates that the population risk associated with InfoNCE, when using k negative samples, is closely related to an expected cross-entropy measure. This measure quantifies the deviation between similarity searches performed on unseen data using learned embeddings and an idealized search using implicit similarity from the positive sample generator. The study also introduces a novel continuity bound for InfoNCE loss, which helps stabilize generalization error as the number of negative samples increases. AI

IMPACT Provides theoretical insights into the generalization properties of contrastive learning methods, potentially informing future model training strategies.

RANK_REASON The cluster contains an academic paper published on arXiv detailing novel research findings.

Read on arXiv stat.ML →

AI-generated summary · Google Gemini · from 2 sources. How we write summaries →

InfoNCE loss generalization in contrastive learning analyzed

COVERAGE [2]

  1. arXiv stat.ML TIER_1 English(EN) · Nick Whiteley ·

    Similarity search generalisation in contrastive learning with InfoNCE loss

    arXiv:2607.09405v1 Announce Type: cross Abstract: Similarity search is a primary application of embedding models trained by contrastive learning. For one of the most popular contrastive learning loss functions, InfoNCE, we show that the population risk with $k$ negative samples i…

  2. arXiv stat.ML TIER_1 English(EN) · Nick Whiteley ·

    Similarity search generalisation in contrastive learning with InfoNCE loss

    Similarity search is a primary application of embedding models trained by contrastive learning. For one of the most popular contrastive learning loss functions, InfoNCE, we show that the population risk with $k$ negative samples is $O(1/k)$ close to an expected cross-entropy whic…