Provable Accuracy Collapse of Embedding-Based Representations under Dimensionality Mismatch
Dionysis Arvanitakis, Vaggos Chatziafratis, Yiyuan Luo
Abstract
Embedding-based representations in Euclidean space are a cornerstone of modern machine learning, where a major goal is to use the smallest dimension that faithfully captures data relations. In this work, we prove sharp dimension--accuracy tradeoffs and identify a fundamental information-theoretic limitation: unless the embedding dimension is chosen close to the ground-truth dimension , accuracy undergoes a sudden collapse. Our main result shows that this phenomenon arises even in standard contrastive learning settings, where supervision is limited to a set of anchor--positive--negative triplets encoding distance comparisons . Specifically, given triplets realizable by an unknown ground-truth embedding in dimensions, we prove that there exists constant , such that every embedding of dimension at most violates half of the triplets, yielding accuracy as low as a trivial one-dimensional solution that ignores the input. We complement our information-theoretic bounds with strong computational hardness results: under the Unique Games Conjecture, even if the given triplets are nearly realizable in dimension, no polynomial-time algorithm---regardless of its dimension---can achieve accuracy above the trivial 50% baseline.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ec77c706-4f05-40b6-a955-d16401ec375bBuilds on8
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Matryoshka Representation LearningAditya Kusupati, Gantavya Bhatt, Aniket Rege, Matthew Wallingford et al.NeurIPS 2022 · 364 citations
- On the Theoretical Limitations of Embedding-Based RetrievalOrion Weller, Michael Boratko, Iftekhar Naim, Jinhyuk LeeICLR 2026 · 138 citations
- Understanding Contrastive Learning Requires Incorporating Inductive BiasesNikunj Saunshi, Jordan T. Ash, Surbhi Goel, Dipendra Misra et al.ICML 2022 · 130 citations
- Optimal Sample Complexity of Contrastive LearningNoga Alon, Dmitrii Avdiukhin, Dor Elboim, Orr Fischer et al.ICLR 2024 · 15 citations
Related papers
- Embedding Dimension of Contrastive Learning and k-Nearest NeighborsDmitrii Avdiukhin, Vaggos Chatziafratis, Orr Fischer, Grigory YaroslavtsevNeurIPS 2024 · 5 citations
- The Complexity of Finding Local Optima in Contrastive LearningJingming Yan, Yiyuan Luo, Vaggos Chatziafratis, Ioannis Panageas et al.NeurIPS 2025 · 2 citations
- Multi-modal contrastive learning adapts to intrinsic dimensions of shared latent variablesYu Gui, Cong Ma, Zongming MaNeurIPS 2025 · 9 citations
- Effective post-training embedding compression via temperature control in contrastive trainingGeorgiana Dinu, Corey D. Barrett, Yi Xiang, Miguel Romero Calvo et al.ICLR 2025
- Triplet Reconstruction and all other Phylogenetic CSPs are Approximation ResistantVaggos Chatziafratis, Konstantin MakarychevFOCS 2023 · 2 citations
