A Neural Autoregressive Approach to Collaborative Filtering

By Yin Zheng, Bangsheng Tang, Wenkui Ding, Hanning Zhou. 2016

227 citations recorded by the source corpus.

This paper proposes CF-NADE, a neural autoregressive architecture for collaborative filtering (CF) tasks, which is inspired by the Restricted Boltzmann Machine (RBM) based CF model and the Neural Autoregressive Distribution Estimator (NADE). We first describe the basic CF-NADE model for CF tasks. Then we propose to improve the model by sharing parameters between different ratings. A factored version of CF-NADE is also proposed for better scalability. Furthermore, we take the ordinal nature of the preferences into consideration and propose an ordinal cost to optimize CF-NADE, which shows superior performance. Finally, CF-NADE can be extended to a deep model, with only moderately increased computational complexity. Experimental results show that CF-NADE with a single hidden layer beats all previous state-of-the-art methods on MovieLens 1M, MovieLens 10M, and Netflix datasets, and adding more hidden layers can further improve the performance.

Subfields: Computer Vision and Pattern Recognition, Representation & Self-Supervised Learning, Signal Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Vision and Pattern Recognition

  1. Xiangnan He
  2. Tat-seng Chua
  3. Yongfeng Zhang
  4. Fuli Feng
  5. Xu Chen
  6. Liqiang Nie
  7. Mingkui Tan
  8. Hanwang Zhang

In Representation & Self-Supervised Learning

  1. Xiangnan He
  2. Tat-seng Chua
  3. Yongfeng Zhang
  4. Fuli Feng
  5. Xu Chen
  6. Liqiang Nie
  7. Yizhou Sun
  8. Ting-Li Chen

In Signal Processing

  1. Harald Steck
  2. Xiangnan He
  3. Fuli Feng
  4. Tat-seng Chua
  5. Federico Monti
  6. Yizhou Sun
  7. Xavier Bresson
  8. Michael M. Bronstein

Citations in the corpus