arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.04541cs.IR
收藏

基于扩散模型的序列推荐

Sequential Recommendation with Diffusion Models

Hanwen Du, Huanhuan Yuan, Zhen Huang, Pengpeng Zhao, Xiaofang Zhou

AI总结:

针对现有生成模型在序列推荐中的后验坍缩和模型坍缩问题,提出基于扩散模型的DiffRec框架,通过修改扩散过程处理离散数据并仅对目标项目加噪,配合高效训练推理策略,在三个公开数据集上超越最先进方法。

摘要:

生成模型,如变分自编码器(VAE)和生成对抗网络(GAN),已成功应用于序列推荐。这些方法需要从概率分布中采样,并采用辅助损失函数来优化模型,从而能够捕捉用户行为的不确定性并缓解曝光偏差。然而,现有的生成模型仍然存在后验坍缩问题或模型坍缩问题,从而限制了它们在序列推荐中的应用。为了应对上述挑战,我们利用生成模型的新范式,即扩散模型,提出了基于扩散模型的序列推荐(DiffRec),该方法可以避免基于VAE和GAN的模型的问题,并展现出更好的性能。虽然扩散模型最初是为处理连续图像数据而提出的,但我们在前向过程中设计了一个额外的转移,并在反向过程中设计了一个转移,以支持对离散推荐数据的处理。我们还设计了一种不同的噪声添加策略,该策略仅对目标项目添加噪声,而不是对整个序列添加噪声,这更适合序列推荐。基于修改后的扩散过程,我们使用简化技术推导了框架的目标函数,并设计了一个去噪序列推荐器来实现该目标函数。由于扩散步骤的延长显著增加了时间复杂度,我们提出了一种高效的训练策略和一种高效的推理策略,以降低训练和推理成本并提高推荐多样性。在三个公开基准数据集上的大量实验结果验证了我们方法的有效性,并表明DiffRec优于最先进的序列推荐模型。

英文摘要:

Generative models, such as Variational Auto-Encoder (VAE) and Generative Adversarial Network (GAN), have been successfully applied in sequential recommendation. These methods require sampling from probability distributions and adopt auxiliary loss functions to optimize the model, which can capture the uncertainty of user behaviors and alleviate exposure bias. However, existing generative models still suffer from the posterior collapse problem or the model collapse problem, thus limiting their applications in sequential recommendation. To tackle the challenges mentioned above, we leverage a new paradigm of the generative models, i.e., diffusion models, and present sequential recommendation with diffusion models (DiffRec), which can avoid the issues of VAE- and GAN-based models and show better performance. While diffusion models are originally proposed to process continuous image data, we design an additional transition in the forward process together with a transition in the reverse process to enable the processing of the discrete recommendation data. We also design a different noising strategy that only noises the target item instead of the whole sequence, which is more suitable for sequential recommendation. Based on the modified diffusion process, we derive the objective function of our framework using a simplification technique and design a denoise sequential recommender to fulfill the objective function. As the lengthened diffusion steps substantially increase the time complexity, we propose an efficient training strategy and an efficient inference strategy to reduce training and inference cost and improve recommendation diversity. Extensive experiment results on three public benchmark datasets verify the effectiveness of our approach and show that DiffRec outperforms the state-of-the-art sequential recommendation models.

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

↑