GR2: Mitigating Granularity Collapse via Adaptive Query Routing in Generative Retrieval

Soona Hong, Dohyeon Lee, Jaeyoung Kim, Seung-Won Hwang


Abstract
Generative Retrieval (GR) aims to directly generate document identifiers (DocIDs) without external indices. We identify granularity collapse: during training, GR models develop a preference for generating DocIDs with a specific granularity, causing retrieval failures when it differs from that of the target document’s DocID. We propose to quantify this mismatch using "isotropy" and confirm that retrieval accuracy degrades as such granularity mismatch increases. To address this, we propose Granularity Routing for Generative Retrieval (GR2), which trains multiple expert models specialized for different granularity levels and routes each query to the appropriate expert. Experiments on NQ320k demonstrate that improves retrieval performance, with larger gains on queries that suffer more from granularity collapse.
Anthology ID:
2026.emnlp-main.1928
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
43891–43902
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1928/
DOI:
Bibkey:
Cite (ACL):
Soona Hong, Dohyeon Lee, Jaeyoung Kim, and Seung-Won Hwang. 2026. GR2: Mitigating Granularity Collapse via Adaptive Query Routing in Generative Retrieval. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 43891–43902, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
GR2: Mitigating Granularity Collapse via Adaptive Query Routing in Generative Retrieval (Hong et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1928.pdf