Differentiable Geometric Indexing for End-to-End Generative Retrieval

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Xujing, Chen, Yufeng, Zhang, Boxuan, Zhao, Jie, Wei, Chao, Xu, Cai, Guan, Ziyu, Zhao, Wei, Zhang, Weiru, Zeng, Xiaoyi
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908878358183936
author Wang, Xujing
Chen, Yufeng
Zhang, Boxuan
Zhao, Jie
Wei, Chao
Xu, Cai
Guan, Ziyu
Zhao, Wei
Zhang, Weiru
Zeng, Xiaoyi
author_facet Wang, Xujing
Chen, Yufeng
Zhang, Boxuan
Zhao, Jie
Wei, Chao
Xu, Cai
Guan, Ziyu
Zhao, Wei
Zhang, Weiru
Zeng, Xiaoyi
contents Generative Retrieval (GR) has emerged as a promising paradigm to unify indexing and search within a single probabilistic framework. However, existing approaches suffer from two intrinsic conflicts: (1) an Optimization Blockage, where the non-differentiable nature of discrete indexing creates a gradient blockage, decoupling index construction from the downstream retrieval objective; and (2) a Geometric Conflict, where standard unnormalized inner-product objectives induce norm-inflation instability, causing popular "hub" items to geometrically overshadow relevant long-tail items. To systematically resolve these misalignments, we propose Differentiable Geometric Indexing (DGI). First, to bridge the optimization gap, DGI enforces Operational Unification. It employs Soft Teacher Forcing via Gumbel-Softmax to establish a fully differentiable pathway, combined with Symmetric Weight Sharing to effectively align the quantizer's indexing space with the retriever's decoding space. Second, to restore geometric fidelity, DGI introduces Isotropic Geometric Optimization. We replace inner-product logits with scaled cosine similarity on the unit hypersphere to effectively decouple popularity bias from semantic relevance. Extensive experiments on large-scale industry search datasets and online e-commerce platform demonstrate that DGI outperforms competitive sparse, dense, and generative baselines. Notably, DGI exhibits superior robustness in long-tail scenarios, validating the necessity of harmonizing structural differentiability with geometric isotropy.
format Preprint
id arxiv_https___arxiv_org_abs_2603_10409
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Differentiable Geometric Indexing for End-to-End Generative Retrieval
Wang, Xujing
Chen, Yufeng
Zhang, Boxuan
Zhao, Jie
Wei, Chao
Xu, Cai
Guan, Ziyu
Zhao, Wei
Zhang, Weiru
Zeng, Xiaoyi
Information Retrieval
Generative Retrieval (GR) has emerged as a promising paradigm to unify indexing and search within a single probabilistic framework. However, existing approaches suffer from two intrinsic conflicts: (1) an Optimization Blockage, where the non-differentiable nature of discrete indexing creates a gradient blockage, decoupling index construction from the downstream retrieval objective; and (2) a Geometric Conflict, where standard unnormalized inner-product objectives induce norm-inflation instability, causing popular "hub" items to geometrically overshadow relevant long-tail items. To systematically resolve these misalignments, we propose Differentiable Geometric Indexing (DGI). First, to bridge the optimization gap, DGI enforces Operational Unification. It employs Soft Teacher Forcing via Gumbel-Softmax to establish a fully differentiable pathway, combined with Symmetric Weight Sharing to effectively align the quantizer's indexing space with the retriever's decoding space. Second, to restore geometric fidelity, DGI introduces Isotropic Geometric Optimization. We replace inner-product logits with scaled cosine similarity on the unit hypersphere to effectively decouple popularity bias from semantic relevance. Extensive experiments on large-scale industry search datasets and online e-commerce platform demonstrate that DGI outperforms competitive sparse, dense, and generative baselines. Notably, DGI exhibits superior robustness in long-tail scenarios, validating the necessity of harmonizing structural differentiability with geometric isotropy.
title Differentiable Geometric Indexing for End-to-End Generative Retrieval
topic Information Retrieval
url https://arxiv.org/abs/2603.10409