Neural variational matrix factorization with side information for collaborative filtering

Date

2019

Authors

Xiao, T.
Shen, H.

Editors

Yang, Q.
Zhou, Z.H.
Gong, Z.
Zhang, M.L.
Huang, S.J.

Advisors

Journal Title

Journal ISSN

Volume Title

Type:

Conference paper

Citation

Lecture Notes in Artificial Intelligence, 2019 / Yang, Q., Zhou, Z.H., Gong, Z., Zhang, M.L., Huang, S.J. (ed./s), vol.11439, pp.414-425

Statement of Responsibility

Teng Xiao and Hong Shen

Conference Name

Pacific-Asia Conference on Knowledge Discovery and Data Mining (PAKDD) (14 Apr 2019 - 17 Apr 2019 : Macau, China)

Abstract

Probabilistic Matrix Factorization (PMF) is a popular technique for collaborative filtering (CF) in recommendation systems. The purpose of PMF is to find the latent factors for users and items by decomposing a user-item rating matrix. Most methods based on PMF suffer from data sparsity and result in poor latent representations of users and items. To alleviate this problem, we propose the neural variational matrix factorization (NVMF) model, a novel deep generative model that incorporates side information (features) of both users and items, to capture better latent representations of users and items for the task of CF recommendation. Our NVMF consists of two end-to-end variational autoencoder neural networks, namely user neural network and item neural network respectively, which are capable of learning complex nonlinear distributed representations of users and items through our proposed variational inference. We derive a Stochastic Gradient Variational Bayes (SGVB) algorithm to approximate the intractable posterior distributions. Experiments conducted on three publicly available datasets show that our NVMF significantly outperforms the state-of-the-art methods.

School/Discipline

Dissertation Note

Provenance

Description

Access Status

Rights

© Springer Nature Switzerland AG 2019

License

Call number

Persistent link to this record