4 个月前

用于协同过滤的序列变分自编码器

用于协同过滤的序列变分自编码器

摘要

变分自编码器在计算机视觉和语音处理等领域已经证明了其成功。尽管如此,将其应用于建模用户偏好仍然是一个未充分探索的领域,尽管最近这一方向开始受到当前文献的关注。在这项工作中,我们提出了一种扩展变分自编码器的模型,该模型充分利用了过去偏好历史中的丰富信息。我们引入了一种递归版本的变分自编码器(VAE),与传统方法不同的是,我们不是忽略时间依赖性而传递整个历史的一个子集,而是通过递归神经网络传递消费序列的一个子集。在递归神经网络(RNN)的每个时间步长中,序列被输入到一系列全连接层中,这些层的输出用于建模最可能的未来偏好的概率分布。我们展示了处理时间信息对于提高变分自编码器的准确性至关重要:事实上,我们的模型由于能够在递归编码器中捕捉用户消费序列的时间依赖性,同时保留了变分自编码器的基本原理,因此在性能上显著优于当前最先进的方法。

代码仓库

noveens/svae_cf
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
collaborative-filtering-on-movielens-1mSVAE
nDCG@100: 0.2993

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于协同过滤的序列变分自编码器 | 论文 | HyperAI超神经