about
Hierarchical Retrieval: The Geometry and a Pretrain-Finetune Recipe (arxiv.org)
1 point by JnBrymn on Sep 25, 2025 | hide | past | pdf | discuss on HN

In plain words: A search model that turns queries and documents into vectors can handle tree-shaped document sets if the vectors grow with tree depth, but struggles to find matches far up the tree. Training on the whole tree first lifts far-match recall from 19% to 76%.

Abstract

Dual encoder (DE) models, where a pair of matching query and document are embedded into similar vector representations, are widely used in information retrieval due to their simplicity and scalability. However, the Euclidean geometry of the embedding space limits the expressive power of DEs, which may compromise their quality. This paper investigates such limitations in the context of hierarchical retrieval (HR), where the document set has a hierarchical structure and the matching documents for a query are all of its ancestors. We first prove that DEs are feasible for HR as long as the embedding dimension is linear in the depth of the hierarchy and logarithmic in the number of documents. Then we study the problem of learning such embeddings in a standard retrieval setup where DEs are trained on samples of matching query and document pairs. Our experiments reveal a lost-in-the-long-distance phenomenon, where retrieval accuracy degrades for documents further away in the hierarchy. To address this, we introduce a pretrain-finetune recipe that significantly improves long-distance retrieval without sacrificing performance on closer documents. We experiment on a realistic hierarchy from WordNet for retrieving documents at various levels of abstraction, and show that pretrain-finetune boosts the recall on long-distance pairs from 19% to 76%. Finally, we demonstrate that our method improves retrieval of relevant products on a shopping queries dataset.

Chong You, Rajesh Jayaram, Ananda Theertha Suresh, Robin Nittka, Felix Yu, Sanjiv Kumar
arXiv:2509.16411 · cs.IR, cs.CL, cs.LG, stat.ML · submitted Sep 19, 2025
abstract · pdf · html · NeurIPS 2025

add comment on HN