Jun, 2023

基于二项式扩散过程的一维数据推荐算法 RecFusion

TL;DR本文提出了 RecFusion,它包含一组扩散模型,用于推荐。我们在一个一维向量上制定了扩散并提出了二项式扩散,它用伯努利过程明确地建模了二元用户 - 项目交互。我们展示了 RecFusion 在核心推荐设置(二元非顺序反馈的 top-n 推荐)和最常见的数据集(MovieLens 和 Netflix)上接近于复杂 VAE 基线的性能。我们所提出的针对 1D 和 / 或二元设置的扩散模型具有医学领域 MRI 和 CT 扫描等超越推荐系统的含义。