A Novel Collaborative Filtering Framework Based on Variational Self-Attention GAN

Weifeng Sun, Shumiao Yu, Jin Yang, Boxiang Dong

Research output: Contribution to journalConference articlepeer-review

1 Scopus citations

Abstract

It is difficult for users to find the required information promptly in the massive data. The collaborative filtering is an effective way to help users get the proper information. To achieve better performance, an improved framework based on variational Generative Adversarial Networks with self-attention (VGCF) is proposed. In VGCF, self-attention mechanism and Variational Autoencoders are combined to form self-attention variational encoder, which improves the ability of acquiring explicit and implicit features of sparse data and obtains the personal preference and the correlation between users. By utilizing Generative Adversarial Networks for prediction, the generative model uses the compression matrix obtained by self-attention variational encoder to generate the predicted user-items interaction matrix. The discriminative model provides a better approximation for the posterior and maximum-likelihood assignment, which makes the generated result closer to the real data distribution. Finally, we show that the performance of VGCF is significantly better than the state-of-the-art recommendation methods on several real-world datasets.

Original languageEnglish
Article number9322300
JournalProceedings - IEEE Global Communications Conference, GLOBECOM
DOIs
StatePublished - 2020
Event2020 IEEE Global Communications Conference, GLOBECOM 2020 - Virtual, Taipei, Taiwan, Province of China
Duration: 7 Dec 202011 Dec 2020

Keywords

  • collaborative filtering
  • generative adversarial networks
  • self-attentions
  • variational autoencoders

Fingerprint

Dive into the research topics of 'A Novel Collaborative Filtering Framework Based on Variational Self-Attention GAN'. Together they form a unique fingerprint.

Cite this