mlsanityData to 7 Oct 2026

ℝ2k is Theoretically Large Enough for Embedding-based Top-k Retrieval

arXiv 2601.20844 ↗First posted 28 Jan 2026cs.LG cs.AI cs.IRcited by 0

Not on the map. Why?

Abstract (arXiv)

This paper studies the Minimal Embeddable Dimension (MED): the least dimension in which there exists a configuration of m object vectors so that every subset of size at most k is exactly retrieved by score comparison. Our result shows MED is Theta(k), independent of m, for inner product, Euclidean distance, and cosine similarity. We then consider Robust MED (RMED), where all vectors are unit normed and an ε gap of scores is required. We derive the m-dependent feasibility ceiling ε_star(m,k)=m/√k(m-1)(m-k), which approaches 1/√k when mgg k, and a Gaussian centroid construction gives a robust witness upper bound in the feasible margin regime. Numerical simulation on synthetic top-2 retrieval with cyclic polytope and centroid query optimization confirmed our theoretical claims. Experiments on LIMIT and LIMIT-small datasets also show that simple embedding-based retrieval baselines can overfit and outperform the reported single-vector LLM embedding baseline. Both theoretical and empirical findings rule out the lack of exact geometric capacity as the obstruction.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 9 of its 18 references that are in the map. First posted on arXiv 28 Jan 2026.

Builds on

9 of 18 references.

References · 9

  1. 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,216
  2. 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
  3. 2019Latent Retrieval for Weakly Supervised Open Domain Question Answeringcited by 691
  4. 2025On the Theoretical Limitations of Embedding-Based Retrievalcited by 98
  5. 2020The Curse of Dense Low-Dimensional Information Retrieval for Large Index Sizescited by 48
  6. 2018On the Dimensionality of Word Embeddingcited by 46
  7. 2025mFollowIR: a Multilingual Benchmark for Instruction Following in Retrievalcited by 7
  8. 2025Hierarchical Retrieval: The Geometry and a Pretrain-Finetune Recipecited by 4
  9. 2014Information-Theoretic Bounds for Adaptive Sparse Recoverycited by 1

Built on by

Not cited yet.