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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Zihao, Yin, Hang, Liu, Lihui, Tong, Hanghang, Song, Yangqiu, Wong, Ginny, See, Simon
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915760413081600
author Wang, Zihao
Yin, Hang
Liu, Lihui
Tong, Hanghang
Song, Yangqiu
Wong, Ginny
See, Simon
author_facet Wang, Zihao
Yin, Hang
Liu, Lihui
Tong, Hanghang
Song, Yangqiu
Wong, Ginny
See, Simon
contents This paper studies the minimal dimension required to embed subset memberships ($m$ elements and ${m\choose k}$ subsets of at most $k$ elements) into vector spaces, denoted as Minimal Embeddable Dimension (MED). The tight bounds of MED are derived theoretically and supported empirically for various notions of "distances" or "similarities," including the $\ell_2$ metric, inner product, and cosine similarity. In addition, we conduct numerical simulation in a more achievable setting, where the ${m\choose k}$ subset embeddings are chosen as the centroid of the embeddings of the contained elements. Our simulation easily realizes a logarithmic dependency between the MED and the number of elements to embed. These findings imply that embedding-based retrieval limitations stem primarily from learnability challenges, not geometric constraints, guiding future algorithm design.
format Preprint
id arxiv_https___arxiv_org_abs_2601_20844
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle $\mathbb{R}^{2k}$ is Theoretically Large Enough for Embedding-based Top-$k$ Retrieval
Wang, Zihao
Yin, Hang
Liu, Lihui
Tong, Hanghang
Song, Yangqiu
Wong, Ginny
See, Simon
Machine Learning
Artificial Intelligence
Information Retrieval
This paper studies the minimal dimension required to embed subset memberships ($m$ elements and ${m\choose k}$ subsets of at most $k$ elements) into vector spaces, denoted as Minimal Embeddable Dimension (MED). The tight bounds of MED are derived theoretically and supported empirically for various notions of "distances" or "similarities," including the $\ell_2$ metric, inner product, and cosine similarity. In addition, we conduct numerical simulation in a more achievable setting, where the ${m\choose k}$ subset embeddings are chosen as the centroid of the embeddings of the contained elements. Our simulation easily realizes a logarithmic dependency between the MED and the number of elements to embed. These findings imply that embedding-based retrieval limitations stem primarily from learnability challenges, not geometric constraints, guiding future algorithm design.
title $\mathbb{R}^{2k}$ is Theoretically Large Enough for Embedding-based Top-$k$ Retrieval
topic Machine Learning
Artificial Intelligence
Information Retrieval
url https://arxiv.org/abs/2601.20844