ArXiv TLDR

Harmonizing Generative Retrieval and Ranking in Chain-of-Recommendation

🐦 Tweet
2604.25787

Yu Liu, Jiangxia Cao

cs.IR

TLDR

RecoChain unifies generative retrieval and ranking in a single Transformer, bridging the gap between generation and ranking for improved Top-K recommendations.

Key contributions

  • Addresses the gap between generative item prediction and effective ranking in recommender systems.
  • Proposes RecoChain, a unified framework integrating candidate generation and ranking in a single Transformer.
  • Employs hierarchical semantic ID prediction for candidates and SIM-based ranking for click probability.
  • Demonstrates improved Top-K recommendation performance on large-scale real-world datasets.

Why it matters

Generative recommenders struggle with ranking the many items they generate. RecoChain solves this by unifying generation and ranking, leading to more accurate Top-K recommendations. This advancement could significantly improve user experience in various recommendation scenarios.

Original Abstract

Generative recommender systems have recently emerged as a promising paradigm by formulating next-item prediction as an auto-regressive semantic IDs generation, such as OneRec series works. However, with the next-item-agnostic prediction paradigm, its could beam out some next potential items via Semantic IDs but hard to estimate which items are better from them, e.g., select the top-10 from beam-256 items, leading to a gap between generation and ranking performance. To fulfill this gap, we propose RecoChain, a unified generative retrieval and ranking framework that integrates candidate generation and ranking within a single Transformer backbone. Specifically, in inference, the model first generates candidate items via hierarchical semantic ID prediction, then performs the SIM-based ranking process to estimate the click possibility of corresponding item candidate continuously. Extensive experiments on large-scale real-world datasets demonstrate that our approach effectively bridges the gap between generative retrieval and ranking, achieving improved Top-K recommendation performance while maintaining strong generative capability.

📬 Weekly AI Paper Digest

Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.