arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.21677cs.IRcs.LG

基于Trie的生成推荐变压器

Trie-Aware Transformers for Generative Recommendation

Zhejiang University(浙江大学)

Zhenxiang Xu, Jiawei Chen, Sirui Chen, Yong He, Jieyu Yang, Chuan Yuan, Ke Ding, Can Wang

AI总结:

TrieRec通过引入基于Trie的结构归纳偏差,改进生成推荐中的自回归建模,提升推荐效果。

摘要:

生成推荐(GR)通过将下一项预测视为令牌级生成而非基于分数的排序,与生成AI的进步相吻合。大多数GR方法采用两阶段流程:(i)物品令牌化,将每个物品映射到一个离散且层次组织的令牌序列;(ii)自回归生成,根据用户交互历史的令牌预测下一项的令牌。尽管层次令牌化在物品上诱导出一个前缀树(trie),但标准自回归建模与传统Transformer通常将物品令牌扁平化为线性流,并忽略了底层拓扑。为此,我们提出了TrieRec,一种基于Trie的生成推荐方法,通过两种位置编码为Transformer引入结构归纳偏差。首先,一种基于Trie的绝对位置编码将令牌的(节点的)局部结构上下文(例如深度、祖先和后代)聚合到令牌表示中。其次,一种拓扑感知的相对位置编码将成对结构关系注入到自注意力中,以捕捉拓扑诱导的语义相关性。TrieRec也是模型无关、高效且无超参数的。在我们的实验中,我们在三个代表性的GR骨干网络中实现了TrieRec,平均在四个真实世界数据集上实现了8.83%的显著提升。

英文摘要:

Generative recommendation (GR) aligns with advances in generative AI by casting next-item prediction as token-level generation rather than score-based ranking. Most GR methods adopt a two-stage pipeline: (i) \textit{item tokenization}, which maps each item to a sequence of discrete, hierarchically organized tokens; and (ii) \textit{autoregressive generation}, which predicts the next item's tokens conditioned on the tokens of user's interaction history. Although hierarchical tokenization induces a prefix tree (trie) over items, standard autoregressive modeling with conventional Transformers often flattens item tokens into a linear stream and overlooks the underlying topology. To address this, we propose TrieRec, a trie-aware generative recommendation method that augments Transformers with structural inductive biases via two positional encodings. First, a \textit{trie-aware absolute positional encoding} aggregates a token's (node's) local structural context (\eg depth, ancestors, and descendants) into the token representation. Second, a \textit{topology-aware relative positional encoding} injects pairwise structural relations into self-attention to capture topology-induced semantic relatedness. TrieRec is also model-agnostic, efficient, and hyperparameter-free. In our experiments, we implement TrieRec within three representative GR backbones, achieving notably improvements of 8.83\% on average across four real-world datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑