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

Weifeng Sun, Shumiao Yu, Jin Yang, Boxiang Dong

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-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
Title of host publication2020 IEEE Global Communications Conference, GLOBECOM 2020 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728182988
DOIs
StatePublished - Dec 2020
Event2020 IEEE Global Communications Conference, GLOBECOM 2020 - Virtual, Taipei, Taiwan, Province of China
Duration: 7 Dec 202011 Dec 2020

Publication series

Name2020 IEEE Global Communications Conference, GLOBECOM 2020 - Proceedings

Conference

Conference2020 IEEE Global Communications Conference, GLOBECOM 2020
Country/TerritoryTaiwan, Province of China
CityVirtual, Taipei
Period7/12/2011/12/20

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