Convolutional Neural Collaborative Filtering with Stacked Embeddings

Liu Han, Hailong Wu, Nan Hu, Binbin Qu
Proceedings of The Eleventh Asian Conference on Machine Learning, PMLR 101:726-741, 2019.

Abstract

Recommender System plays an important role in keeping people engaged with online services, and collaborative filtering is a main technique for recommendation. With the immense influence of deep learning, there is a growing interest in applying it to collaborative filtering. Existing methods have applied different ways to learn the user-item interaction function, however, most of these methods have limitation in modeling user-item correlations because they ignore the original user-item information and the large size of embeddings. In this work we propose Stacked Embedding Convolutional Neural Collaborative Filtering (SECNCF), a novel neural collaborative filtering architecture. The idea is to create a pedrail by stacking embeddings which are composed of user embedding, item embedding and latent factors. We apply convolutional neural network (CNN) above the pedrail layer to capture the local features of dimension correlations. This method is good at extracting rich local dimension correlations of embeddings and is scalable for modeling user-item interactions. Extensive experiments on three public accessible datasets show that our method makes significant improvement over the state-of-the-art methods.

Cite this Paper

BibTeX
@InProceedings{pmlr-v101-han19a, title = {Convolutional Neural Collaborative Filtering with Stacked Embeddings }, author = {Han, Liu and Wu, Hailong and Hu, Nan and Qu, Binbin}, booktitle = {Proceedings of The Eleventh Asian Conference on Machine Learning}, pages = {726--741}, year = {2019}, editor = {Lee, Wee Sun and Suzuki, Taiji}, volume = {101}, series = {Proceedings of Machine Learning Research}, month = {17--19 Nov}, publisher = {PMLR}, pdf = {http://proceedings.mlr.press/v101/han19a/han19a.pdf}, url = {https://proceedings.mlr.press/v101/han19a.html}, abstract = {Recommender System plays an important role in keeping people engaged with online services, and collaborative filtering is a main technique for recommendation. With the immense influence of deep learning, there is a growing interest in applying it to collaborative filtering. Existing methods have applied different ways to learn the user-item interaction function, however, most of these methods have limitation in modeling user-item correlations because they ignore the original user-item information and the large size of embeddings. In this work we propose Stacked Embedding Convolutional Neural Collaborative Filtering (SECNCF), a novel neural collaborative filtering architecture. The idea is to create a pedrail by stacking embeddings which are composed of user embedding, item embedding and latent factors. We apply convolutional neural network (CNN) above the pedrail layer to capture the local features of dimension correlations. This method is good at extracting rich local dimension correlations of embeddings and is scalable for modeling user-item interactions. Extensive experiments on three public accessible datasets show that our method makes significant improvement over the state-of-the-art methods.} }
Endnote
%0 Conference Paper %T Convolutional Neural Collaborative Filtering with Stacked Embeddings %A Liu Han %A Hailong Wu %A Nan Hu %A Binbin Qu %B Proceedings of The Eleventh Asian Conference on Machine Learning %C Proceedings of Machine Learning Research %D 2019 %E Wee Sun Lee %E Taiji Suzuki %F pmlr-v101-han19a %I PMLR %P 726--741 %U https://proceedings.mlr.press/v101/han19a.html %V 101 %X Recommender System plays an important role in keeping people engaged with online services, and collaborative filtering is a main technique for recommendation. With the immense influence of deep learning, there is a growing interest in applying it to collaborative filtering. Existing methods have applied different ways to learn the user-item interaction function, however, most of these methods have limitation in modeling user-item correlations because they ignore the original user-item information and the large size of embeddings. In this work we propose Stacked Embedding Convolutional Neural Collaborative Filtering (SECNCF), a novel neural collaborative filtering architecture. The idea is to create a pedrail by stacking embeddings which are composed of user embedding, item embedding and latent factors. We apply convolutional neural network (CNN) above the pedrail layer to capture the local features of dimension correlations. This method is good at extracting rich local dimension correlations of embeddings and is scalable for modeling user-item interactions. Extensive experiments on three public accessible datasets show that our method makes significant improvement over the state-of-the-art methods.
APA
Han, L., Wu, H., Hu, N. & Qu, B.. (2019). Convolutional Neural Collaborative Filtering with Stacked Embeddings . Proceedings of The Eleventh Asian Conference on Machine Learning, in Proceedings of Machine Learning Research 101:726-741 Available from https://proceedings.mlr.press/v101/han19a.html.

Related Material