CapsID: Soft-Routed Variable-Length Semantic IDs for Generative Recommendation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cheng, Wenzhuo, Gong, Menghang, Guo, Qixin, Zheng, Hang, Yang, Zhaobin, Lou, Jianguo, Zheng, Zhengwei
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918486306979840
author Cheng, Wenzhuo
Gong, Menghang
Guo, Qixin
Zheng, Hang
Yang, Zhaobin
Lou, Jianguo
Zheng, Zhengwei
author_facet Cheng, Wenzhuo
Gong, Menghang
Guo, Qixin
Zheng, Hang
Yang, Zhaobin
Lou, Jianguo
Zheng, Zhengwei
contents Generative recommendation maps each item to a sequence of Semantic IDs (SIDs) and recasts retrieval as autoregressive token generation. In this paradigm the main bottleneck is the tokenizer rather than the Transformer: residual vector quantization with a hard nearest-neighbor assignment at every layer collapses multi-faceted item semantics at cluster boundaries and propagates early errors to later SID positions. A common workaround is to append a dense vector or attribute prefix to the SID, but this dual-representation design inflates inference cost and gives up the simplicity of a generative interface. We address the bottleneck at the tokenizer itself. CAPSID replaces hard residual quantization with capsule routing: at each layer an item probabilistically routes to several semantic capsules, the residual is updated by the routed reconstruction rather than by a single winning code, and the SID terminates once the active capsule's confidence is high enough. On top of CAPSID, SEMANTICBPE composes adjacent SID tokens into reusable subwords by combining their co-occurrence with their embedding compatibility. On Amazon Beauty, Sports, Toys, and a 35M-item proprietary industrial catalog, CAPSID+SEMANTICBPE improves Recall at 10 by 9.6% on average over ReSID, the strongest single-representation baseline, and matches or exceeds a COBRA-style sparse-dense system on every public benchmark while running at 51% of its inference latency. Ablations show that soft routing, iterative agreement, and confidence-driven length each contribute independently, and the gains are largest on tail items where boundary semantics dominate.
format Preprint
id arxiv_https___arxiv_org_abs_2605_05096
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle CapsID: Soft-Routed Variable-Length Semantic IDs for Generative Recommendation
Cheng, Wenzhuo
Gong, Menghang
Guo, Qixin
Zheng, Hang
Yang, Zhaobin
Lou, Jianguo
Zheng, Zhengwei
Information Retrieval
Generative recommendation maps each item to a sequence of Semantic IDs (SIDs) and recasts retrieval as autoregressive token generation. In this paradigm the main bottleneck is the tokenizer rather than the Transformer: residual vector quantization with a hard nearest-neighbor assignment at every layer collapses multi-faceted item semantics at cluster boundaries and propagates early errors to later SID positions. A common workaround is to append a dense vector or attribute prefix to the SID, but this dual-representation design inflates inference cost and gives up the simplicity of a generative interface. We address the bottleneck at the tokenizer itself. CAPSID replaces hard residual quantization with capsule routing: at each layer an item probabilistically routes to several semantic capsules, the residual is updated by the routed reconstruction rather than by a single winning code, and the SID terminates once the active capsule's confidence is high enough. On top of CAPSID, SEMANTICBPE composes adjacent SID tokens into reusable subwords by combining their co-occurrence with their embedding compatibility. On Amazon Beauty, Sports, Toys, and a 35M-item proprietary industrial catalog, CAPSID+SEMANTICBPE improves Recall at 10 by 9.6% on average over ReSID, the strongest single-representation baseline, and matches or exceeds a COBRA-style sparse-dense system on every public benchmark while running at 51% of its inference latency. Ablations show that soft routing, iterative agreement, and confidence-driven length each contribute independently, and the gains are largest on tail items where boundary semantics dominate.
title CapsID: Soft-Routed Variable-Length Semantic IDs for Generative Recommendation
topic Information Retrieval
url https://arxiv.org/abs/2605.05096