Learning Query Encoders Can Be Hard Even When Vector Retrieval Is Geometrically Easy
Organizations: Stanford University · Amazon AWS · Tel Aviv University
Abstract
Efficient vector retrieval requires both a corpus geometry that supports retrieving the right documents through vector similarity, and a query encoder that can embed queries near their desired documents in the embedding space. Recent work has studied geometric capacity through the lens of the minimum embedding dimension needed to realize all top- answer sets of documents. We study a different notion of geometric capacity--the maximum recall achievable for a frozen document index--and explore whether learned query encoders can reach this ceiling. On several real-world retrieval benchmarks, we show that retrieval quality of single-vector query encoders often lies far below what the document indices can support. Motivated by this observation, we give theoretical evidence that learning query encoders can be computationally hard. In particular, we construct a retrieval task that (1) admits a query encoder with perfect recall which is representable by a small one-hidden-layer ReLU network, but (2) any statistical-query learner (a class capturing learners that access training data through aggregate statistics) provably requires exponentially many statistical queries to achieve non-trivial recall advantage over the random baseline . Taken together, our results suggest substantial unrealized geometric capacity in retrieval benchmarks and establish query encoder learnability as a possible barrier in embedding-based retrieval.
Figures & tables
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| DistilBERT | Qwen3-Embedding-4B | GTE-ModernColBERT | |
| Checkpoint | sentence-transformers/ | Qwen/ | lightonai/ |
| msmarco-distilbert- | Qwen3-Embedding-4B | GTE-ModernColBERT- | |
| base-v3 | v1 | ||
| Representation | single-vector | single-vector | multi-vector (MaxSim) |
| Pooling | mean | last token | none (per token) |
| Truncation | 350 tok. | 512 tok. | 300 tok. (queries 32) |
| DistilBERT | Qwen3-Embedding-4B | ||||||
| Category | train | eval | train | eval | |||
| LIMIT | limit | 1.000 | 1.000 | 1.000 | 1.000 | ||
| BRIGHT | biology | 0.979 | 0.962 | 0.980 | 0.954 | ||
| earth science | 0.980 | 0.997 | 0.985 | 0.999 | |||
| economics | 0.983 | 1.000 | 0.984 | 0.999 | |||
| psychology | 0.999 | 0.996 | 0.995 | 0.990 | |||
| Category | BM25 | DistilBERT | Qwen3-Embedding-4B | GTE-ModernColBERT | |
|---|---|---|---|---|---|
| LIMIT | limit | 0.917 | 0.002 | 0.015 | 0.525 |
| BRIGHT | biology | 0.156 | 0.016 | 0.073 | 0.078 |
| earth science | 0.201 | 0.078 | 0.233 | 0.168 | |
| economics | 0.186 | 0.074 | 0.179 | 0.143 | |
| psychology | 0.087 | 0.073 | 0.210 | 0.102 | |
| robotics | 0.046 | 0.045 | 0.038 | 0.104 |
| DistilBERT | Qwen3-Embedding-4B | GTE-ModernColBERT | ||||||||
| Category | train | eval | train | eval | train | eval | ||||
| LIMIT | limit | 0.439 | 0.045 | 1.000 | 0.068 | 1.000 | 0.998 | |||
| BRIGHT | biology | 0.972 | 0.105 | 0.972 | 0.189 | 0.975 | 0.240 | |||
| earth science | 0.973 | 0.178 | 0.982 | 0.274 | 0.910 | 0.243 | ||||
| economics | 0.984 | 0.103 | 0.995 | 0.276 | 0.971 | 0.197 | ||||
| psychology | 0.992 | 0.021 | 0.996 | 0.151 | 0.970 | 0.119 | ||||
Dense Retrieval
Dense retrieval uses learned vector representations to find relevant documents, passages, products, or tools by semantic similarity.
See all 152 papers