Lune

ICML2026Top-tier venue

R2k\mathbb{R}^{2k} is Theoretically Large Enough for Embedding-based Top-kk Retrieval

Zihao Wang, Hang Yin, Lihui Liu, Hanghang Tong, Yangqiu Song, Ginny Wong, Simon See

2026Year
1Top-tier citations

Abstract

This paper studies the Minimal Embeddable Dimension (MED): the least dimension in which there exists a configuration of mm object vectors so that every subset of size at most kk is exactly retrieved by score comparison. Our result shows MED is Θ(k)\Theta(k), independent of mm, for inner product, Euclidean distance, and cosine similarity. We then consider Robust MED (RMED), where all vectors are unit normed and an ϵ\epsilon gap of scores is required. We derive the mm-dependent feasibility ceiling ϵ⋆(m,k)=m/k(m−1)(m−k)\epsilon_\star(m,k)=m/\sqrt{k(m-1)(m-k)}, which approaches 1/k1/\sqrt{k} when m≫km\gg k, and a Gaussian centroid construction gives a robust witness upper bound in the feasible margin regime. Numerical simulation on synthetic top-22 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.

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.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext a5101794-8e5e-481f-8473-3ce453dbfe54

Cited by top-tier papers1

Ask how each one uses it

Builds on2

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines