论文

REGER:用语义ID建模物品间关系的生成式推荐框架

REGER: Modeling inter-item Relations with Semantic IDs for Generative Recommendation

精选理由

这篇论文做了个叫 REGER 的生成式推荐框架,让模型在预训练阶段就学会物品间的语义和协作关系,还配了套前缀树偏好蒸馏的理论,实验比现有 GR 基线都好,做推荐系统的话很值得看看。

论文提出 REGER,一个针对生成式推荐(GR)的统一框架,通过自回归生成物品的离散语义 ID(SID)进行推荐。针对随机初始化 SID token 嵌入导致模型无法继承物品语义关系的问题,REGER 通过生成式语义检索预训练,让模型学习生成语义邻居的 SID,从而继承物品间语义相似性。框架将全局注意力与对齐 SID 边界的局部注意力结合,捕捉历史上下文和相邻物品间的有向转移。作者还建立了基于 SID 前缀树的层次偏好分解理论框架,证明协作偏好分布的精确分解,并据此设计了从判别式教师模型蒸馏全条件分布的偏好蒸馏目标。真实数据集上的实验显示 REGER 一致优于现有 GR 基线。

原文 · arXiv cs.AI

REGER: Modeling inter-item Relations with Semantic IDs for Generative Recommendation

Personalized recommendation depends on learning semantic and collaborative relations among items from online content and user interactions. Generative recommendation (GR) offers a promising paradigm by autoregressively generating discrete semantic IDs (SIDs) of items. However, randomly initialized SID token embeddings and GR models trained from scratch do not directly inherit the semantic relations inherent in items, while inter-item collaborative relations are learned largely implicitly through token-level attention and next-token supervision. To address this limitation, we propose REGER, a unified framework that explicitly incorporates inter-item relations into semantic pretraining, history encoding, and SID generation. We pretrain the GR model through generative semantic retrieval, enabling it to inherit inter-item semantic similarity by learning to generate semantic neighbors' SIDs. We combine global attention with local attention aligned with SID boundaries to capture historical context and directed transitions between adjacent items. We further develop a theoretical framework for hierarchical preference decomposition over the SID prefix tree, proving an exact factorization of collaborative preference distributions and a weighted decomposition of item-level KL divergence into prefix-conditional KL divergences. Building on this analysis, we distill a discriminative teacher's full conditional distributions over all legal next tokens at ground-truth prefixes along the target SID path, jointly optimizing this objective with cross-entropy. Extensive experiments on real-world datasets demonstrate consistent improvements over existing GR baselines and support the effectiveness of the proposed preference distillation.