The Complexity of Finding Local Optima in Contrastive Learning

Vaggos Chatziafratis (University of California, Santa Cruz) · Ioannis Panageas (UC Irvine) · Jingming Yan (University of California, Irvine) · Yiyuan Luo (UC Santa Cruz) · Parnian Shahkar (University of California, Irvine) · Stelios Stavroulakis (UCI)
cls-hardnesscontinuous settingscontrastive informationcontrastive learningdata representationsdiscrete settingsembeddingsexponential timeglobal optimalocal optimalocal search dynamicsnp-hardpls-hardnesspolynomial time algorithmtriplet lossweighted triplets

Contrastive learning is a powerful technique for discovering meaningful data representations by optimizing objectives based on $\textit{contrastive information}$, often given as a set of weighted triplets $\{(x_i, y_i^+, z_{i}^-)\}_{i = 1}^m$ indicating that an "anchor" $x_i$ is more similar to a "positive" example $y_i$ than to a "negative" example $z_i$. The goal is to find representations (e.g., embeddings in $\mathbb{R}^d$ or a tree metric) where anchors are placed closer to positive than to negative examples. While finding $\textit{global}$ optima of contrastive objectives is $\mathsf{NP}$-hard, the complexity of finding $\text{\textit{local}}$ optima