arXiv:2409.10522cs.IRcs.AI2024-09被引 16

用薛定谔桥改进扩散模型,让推荐更懂用户动态变化。

Bridging User Dynamics: Transforming Sequential Recommendations with Schrödinger Bridge and Diffusion Models

  • 引入薛定谔桥替换扩散模型的高斯先验,直接建模用户状态到推荐目标的演变过程。
  • 在多个公开数据集上优于主流方法,尤其在长序列推荐中提升显著。
  • 适合关注用户兴趣演化与生成式推荐的研究者或工程师。

序列推荐因能精准捕捉用户兴趣的动态变化而受到越来越多关注。我们注意到,生成模型尤其是扩散模型在图像、音频等领域取得显著成果,有望应用于序列推荐。然而,现有基于扩散模型的推荐方法受限于固定的高斯先验分布,难以融入个性化用户信息,导致信息损失。为此,本文将薛定谔桥引入扩散模型,提出SdifRec模型,以用户当前状态替代高斯先验,直接建模从用户当前状态到目标推荐的演化过程。为进一步利用协同信息,提出扩展版本con-SdifRec,利用用户聚类信息作为引导条件,增强后验分布。大量实验在多个公开基准数据集上验证了SdifRec和con-SdifRec的有效性,相比多种先进方法表现更优。深入分析也证实了其高效性与鲁棒性。

原文摘要 · Abstract (English)

Sequential recommendation has attracted increasing attention due to its ability to accurately capture the dynamic changes in user interests. We have noticed that generative models, especially diffusion models, which have achieved significant results in fields like image and audio, hold considerable promise in the field of sequential recommendation. However, existing sequential recommendation methods based on diffusion models are constrained by a prior distribution limited to Gaussian distribution, hindering the possibility of introducing user-specific information for each recommendation and leading to information loss. To address these issues, we introduce the Schrödinger Bridge into diffusion-based sequential recommendation models, creating the SdifRec model. This allows us to replace the Gaussian prior of the diffusion model with the user's current state, directly modeling the process from a user's current state to the target recommendation. Additionally, to better utilize collaborative information in recommendations, we propose an extended version of SdifRec called con-SdifRec, which utilizes user clustering information as a guiding condition to further enhance the posterior distribution. Finally, extensive experiments on multiple public benchmark datasets have demonstrated the effectiveness of SdifRec and con-SdifRec through comparison with several state-of-the-art methods. Further in-depth analysis has validated their efficiency and robustness.

序列推荐扩散模型用户建模生成式推荐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。