VDGAN: A Collaborative Filtering Framework Based on Variational Denoising with GANs

Weifeng Sun, Shumiao Yu, Boxiang Dong

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Generative Adversarial Networks (GANs) effectively capture the true posterior distribution. When applied to Collaborative Filtering (CF), GANs can generate a recommendation list through implicit feedback. However, the discriminators in the existing GANs-based CF methods are not utilized fully, and the generators perform poorly on sparse data mining. In this paper, we propose an improved collaborative filtering framework based on variational denoising for GANs (VDGAN). Specifically, VDGAN integrates the variational encoder and the self-attention mechanism into the GANs. By using the positive-negative sampling mechanism to add specific noise to the input data, the variational encoder obtains a robust feature matrix and improves the sparse data processing capability of the generator. In VDGAN, the denoising generator reconstructs the user-items interaction matrix through the feature matrix. And the discriminator is composed of the self-attention mechanism to obtain the explicit features of user preferences, which extends the ability of the discriminator. Furthermore, reinforcement learning replaces the traditional objective function of GANs, which better optimizes the generator and further improves the recommendation accuracy of the model. From our comprehensive experiments on three real-world datasets, we demonstrate that the performance of VDGAN significantly outperforms the state-of-the-art methods based on GANs and Auto-Encoders.

Original languageEnglish
Title of host publicationIJCNN 2021 - International Joint Conference on Neural Networks, Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9780738133669
DOIs
StatePublished - 18 Jul 2021
Event2021 International Joint Conference on Neural Networks, IJCNN 2021 - Virtual, Shenzhen, China
Duration: 18 Jul 202122 Jul 2021

Publication series

NameProceedings of the International Joint Conference on Neural Networks
Volume2021-July

Conference

Conference2021 International Joint Conference on Neural Networks, IJCNN 2021
Country/TerritoryChina
CityVirtual, Shenzhen
Period18/07/2122/07/21

Keywords

  • collaborative filtering
  • generative adversarial networks
  • reinforcement learning
  • self-attention
  • variational autoencoders

Fingerprint

Dive into the research topics of 'VDGAN: A Collaborative Filtering Framework Based on Variational Denoising with GANs'. Together they form a unique fingerprint.

Cite this