TY - GEN
T1 - VDGAN
T2 - 2021 International Joint Conference on Neural Networks, IJCNN 2021
AU - Sun, Weifeng
AU - Yu, Shumiao
AU - Dong, Boxiang
N1 - Publisher Copyright:
© 2021 IEEE.
PY - 2021/7/18
Y1 - 2021/7/18
N2 - 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.
AB - 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.
KW - collaborative filtering
KW - generative adversarial networks
KW - reinforcement learning
KW - self-attention
KW - variational autoencoders
UR - http://www.scopus.com/inward/record.url?scp=85116399012&partnerID=8YFLogxK
U2 - 10.1109/IJCNN52387.2021.9533585
DO - 10.1109/IJCNN52387.2021.9533585
M3 - Conference contribution
AN - SCOPUS:85116399012
T3 - Proceedings of the International Joint Conference on Neural Networks
BT - IJCNN 2021 - International Joint Conference on Neural Networks, Proceedings
PB - Institute of Electrical and Electronics Engineers Inc.
Y2 - 18 July 2021 through 22 July 2021
ER -