is Theoretically Large Enough for Embedding-based Top- Retrieval
Zihao Wang, Hang Yin, Lihui Liu, Hanghang Tong, Yangqiu Song, Ginny Wong, Simon See
Abstract
This paper studies the Minimal Embeddable Dimension (MED): the least dimension in which there exists a configuration of object vectors so that every subset of size at most is exactly retrieved by score comparison. Our result shows MED is , independent of , 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 -dependent feasibility ceiling , which approaches when , and a Gaussian centroid construction gives a robust witness upper bound in the feasible margin regime. Numerical simulation on synthetic top- 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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a5101794-8e5e-481f-8473-3ce453dbfe54Cited by top-tier papers1
Ask how each one uses itBuilds on2
- On the Theoretical Limitations of Embedding-Based RetrievalOrion Weller, Michael Boratko, Iftekhar Naim, Jinhyuk LeeICLR 2026 · 138 citations
- Hierarchical Retrieval: The Geometry and a Pretrain-Finetune RecipeChong You, Rajesh Jayaram, Ananda Theertha Suresh, Robin Nittka et al.NeurIPS 2025 · 4 citations
Related papers
- On Coresets for Clustering in Small Dimensional Euclidean spacesLingxiao Huang, Ruiyuan Huang, Zengfeng Huang, Xuan WuICML 2023 · 7 citations
- Coreset for Robust Geometric Median: Eliminating Size Dependency on OutliersZiyi Fang, Lingxiao Huang, Runkai YangNeurIPS 2025 · 1 citation
- Near-optimal Coresets for Robust ClusteringLingxiao Huang, Shaofeng H.-C. Jiang, Jianing Lou, Xuan WuICLR 2023 · 1 citation
- Robust Sparsification via SensitivityChansophea Wathanak In, Yi Li, David P. Woodruff, Xuan WuICML 2025
- Towards optimal lower bounds for k-median and k-means coresetsVincent Cohen-Addad, Kasper Green Larsen, David Saulpic, Chris SchwiegelshohnSTOC 2022 · 20 citations
