Preprint
Machine Learning

On the Theoretical Limitations of Embedding-based Retrieval

Orion Weller, Michael Boratko, Iftekhar Naim, Jinhyuk Lee
August 28, 2025arXiv.org121 citations

121

Citations

13

Influential Citations

arXiv.org

Venue

2025

Year

Abstract

Vector embeddings have been tasked with an ever-increasing set of retrieval tasks over the years, with a nascent rise in using them for reasoning, instruction-following, coding, and more. These new benchmarks push embeddings to work for any query and any notion of relevance that could be given. While prior works have pointed out theoretical limitations of vector embeddings, there is a common assumption that these difficulties are exclusively due to unrealistic queries, and those that are not can be overcome with better training data and larger models. In this work, we demonstrate that we may encounter these theoretical limitations in realistic settings with extremely simple queries. We connect known results in learning theory, showing that the number of top-k subsets of documents capable of being returned as the result of some query is limited by the dimension of the embedding. We empirically show that this holds true even if we directly optimize on the test set with free parameterized embeddings. Using free embeddings, we then demonstrate that returning all pairs of documents requires a relatively high dimension. We then create a realistic dataset called LIMIT that stress tests embedding models based on these theoretical results, and observe that even state-of-the-art models fail on this dataset despite the simple nature of the task. Our work shows the limits of embedding models under the existing single vector paradigm and calls for future research to develop new techniques that can resolve this fundamental limitation.

Analysis

Why This Paper Matters

This paper strikes at a core assumption in modern AI: that vector embeddings can serve as a universal representation for any retrieval task, given enough data and compute. The authors show that this assumption is mathematically false—there is a hard, dimension-dependent limit on the number of distinct retrieval results an embedding can produce. What makes this work particularly impactful is that it demonstrates this limitation not just in theory, but in practice with simple, realistic queries. The LIMIT dataset they release will likely become a standard benchmark for testing the true retrieval capacity of embedding models.

For practitioners, this means that current single-vector embedding models may be fundamentally incapable of handling certain retrieval tasks, no matter how much they are trained. This is a sobering result for the many production systems that rely on embeddings for search, recommendation, and question answering. The paper provides a clear theoretical framework to understand when and why embeddings will fail.

Technical Contributions

  • Theoretical bound: The authors prove that a d-dimensional embedding can represent at most O(d * log(d)) distinct top-k subsets, connecting this to known results in learning theory about the number of linear separators.
  • Empirical validation with free embeddings: They show that even when embeddings are directly optimized on the test set (i.e., given infinite training data and capacity), the bound holds—proving it is a fundamental limitation, not a training issue.
  • Pairwise retrieval limitation: They demonstrate that to return all possible pairs of documents, the embedding dimension must grow at least quadratically in the number of documents.
  • LIMIT dataset: A carefully constructed set of queries that stress-test the theoretical limits, showing that even state-of-the-art models like OpenAI's text-embedding-3-large fail on simple tasks.

Results

The paper does not report traditional accuracy metrics on standard benchmarks; instead, it measures the number of distinct top-k subsets that models can produce. For example, with a 10-dimensional embedding, the theoretical maximum is around 23 distinct top-1 subsets, and the authors confirm that even free embeddings cannot exceed this. On the LIMIT dataset, state-of-the-art models achieve near-zero accuracy on queries designed to require more distinct subsets than their dimension allows.

Significance

This work has profound implications for the field of information retrieval and representation learning. It provides a rigorous mathematical foundation for understanding the limits of embeddings, which have become a default tool in AI. The paper calls for new research into multi-vector representations, learned hash functions, or other architectures that can overcome this fundamental bottleneck. For practitioners, it offers a clear diagnostic: if your retrieval task requires distinguishing many different relevance notions, you may need to move beyond single-vector embeddings.