arXiv:2502.16140cs.IR2025-02中稿 · DASFAA 2025

用混合高斯先验建模用户多重兴趣,提升序列推荐效果

Semantic Gaussian Mixture Variational Autoencoder for Sequential Recommendation

  • 采用高斯混合分布作为用户行为序列的先验,捕捉多兴趣特性
  • 在多个公开数据集上优于现有方法,显著提升推荐准确率
  • 适合需要建模复杂用户偏好的推荐系统研究者

基于变分自编码器(VAE)的序列推荐模型通过学习用户-物品交互序列的连续分布,具备较强的抗数据稀疏能力并取得显著性能。然而,现有模型假设序列表示的先验为单峰高斯分布,限制了对复杂用户兴趣的捕捉能力,尤其当用户具有多个异质兴趣时表现受限。考虑到用户通常拥有多种兴趣,本文提出新型VAE-based序列推荐模型SIGMA,其先验分布为高斯混合分布,每个成分语义对应一种兴趣。SIGMA引入概率化多兴趣提取模块,根据隐式物品超类别学习每个兴趣的单峰高斯分布;同时构建多兴趣感知的变分下界(ELBO),兼容高斯混合先验。大量实验表明SIGMA在多个公开数据集上均有效,代码已开源。

原文摘要 · Abstract (English)

Variational AutoEncoder (VAE) for Sequential Recommendation (SR), which learns a continuous distribution for each user-item interaction sequence rather than a determinate embedding, is robust against data deficiency and achieves significant performance. However, existing VAE-based SR models assume a unimodal Gaussian distribution as the prior distribution of sequence representations, leading to restricted capability to capture complex user interests and limiting recommendation performance when users have more than one interest. Due to that it is common for users to have multiple disparate interests, we argue that it is more reasonable to establish a multimodal prior distribution in SR scenarios instead of a unimodal one. Therefore, in this paper, we propose a novel VAE-based SR model named SIGMA. SIGMA assumes that the prior of sequence representation conforms to a Gaussian mixture distribution, where each component of the distribution semantically corresponds to one of multiple interests. For multi-interest elicitation, SIGMA includes a probabilistic multi-interest extraction module that learns a unimodal Gaussian distribution for each interest according to implicit item hyper-categories. Additionally, to incorporate the multimodal interests into sequence representation learning, SIGMA constructs a multi-interest-aware ELBO, which is compatible with the Gaussian mixture prior. Extensive experiments on public datasets demonstrate the effectiveness of SIGMA. The code is available at https://github.com/libeibei95/SIGMA.

序列推荐变分自编码器多兴趣建模

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