cs.LGMar 27, 2026

Contrastive Conformal Sets

Authors: Yahya AlkhatibWee Peng Tay

Organizations: School of Electrical and Electronic Engineering, Nanyang Technological University, Singapore

Abstract

Contrastive learning produces coherent semantic feature embeddings by encouraging positive samples to cluster closely while separating negative samples. However, existing contrastive learning methods lack a principled construction of geometric sets in the semantic feature space with distribution-free guarantees at any user-specified coverage level. We extend conformal prediction to this setting by introducing covering sets equipped with learnable generalized hyper-ball constraints. We propose a method that constructs conformal sets guaranteeing user-specified coverage of positive samples while maximizing negative sample exclusion. We theoretically motivate volume minimization as a proxy for negative exclusion, enabling our approach to operate effectively even when negative pairs are unavailable. The positive inclusion guarantee inherits the distribution-free coverage property of conformal prediction, while negative exclusion is maximized through learned set geometry optimized on a held-out training split. Experiments on simulated and real-world image datasets demonstrate improved inclusion-exclusion trade-offs compared to standard distance-based conformal baselines.

Explore similar work

CardsList
  1. Conformal Robust Set Estimation

    Apr 20, 2026Alejandro Cholaquidis, Emilien Joly, Leonardo MorenoConformal PredictionConformal Calibration

  2. Cluster Frequency Conformal Prediction for Local Coverage

    May 24, 2026Tomer Lavi, Bracha Shapira, Nadav RappoportConformal Prediction