论文标题
推荐系统的基于GAN的基质分解
GAN-based Matrix Factorization for Recommender Systems
论文作者
论文摘要
提议在2014年提出的生成对抗网络(GAN)对生成建模产生了新的兴趣。他们立即在图像综合,图像到图像翻译,文本对图像生成,图像介绍,并已用于从药物到高能粒子物理学等等的科学中。尽管它们的知名度和学习任意分布的能力,但GAN并未被广泛应用于推荐系统(RS)。此外,只有少数在RS中引入GAN的技术直接将它们用作协作过滤(CF)模型。在这项工作中,我们提出了一种新的基于GAN的方法,该方法在矩阵分解设置中以通用TOP-N推荐问题学习用户和项目潜在因素。遵循CFGAN引入的RS的Vector GAN训练方法之后,我们在使用GAN进行CF时确定了2个独特的问题。我们通过使用自动编码器作为鉴别器并为发电机提出了额外的损失函数,为它们都提出了解决方案。我们通过RS社区中知名的数据集评估了我们的模型GANMF,并显示了对传统CF方法和基于GAN的模型的改进。通过对GANMF组成部分的消融研究,我们旨在了解我们的建筑选择的影响。最后,我们对GANMF的矩阵分解性能提供了定性评估。
Proposed in 2014, Generative Adversarial Networks (GAN) initiated a fresh interest in generative modelling. They immediately achieved state-of-the-art in image synthesis, image-to-image translation, text-to-image generation, image inpainting and have been used in sciences ranging from medicine to high-energy particle physics. Despite their popularity and ability to learn arbitrary distributions, GAN have not been widely applied in recommender systems (RS). Moreover, only few of the techniques that have introduced GAN in RS have employed them directly as a collaborative filtering (CF) model. In this work we propose a new GAN-based approach that learns user and item latent factors in a matrix factorization setting for the generic top-N recommendation problem. Following the vector-wise GAN training approach for RS introduced by CFGAN, we identify 2 unique issues when utilizing GAN for CF. We propose solutions for both of them by using an autoencoder as discriminator and incorporating an additional loss function for the generator. We evaluate our model, GANMF, through well-known datasets in the RS community and show improvements over traditional CF approaches and GAN-based models. Through an ablation study on the components of GANMF we aim to understand the effects of our architectural choices. Finally, we provide a qualitative evaluation of the matrix factorization performance of GANMF.