LGIRMLDec 25, 2017

Neural Collaborative Autoencoder

arXiv:1712.09043v33 citations
Originality Incremental advance
AI Analysis

This work addresses generic collaborative filtering for recommendation systems, offering a solution that works for both explicit and implicit feedback, though it appears incremental in its approach.

The paper tackled the problem of existing deep learning models for recommendation being limited to either explicit or implicit feedback, not fully exploiting deep learning's potential, and overfitting on implicit data, by proposing the Neural Collaborative Autoencoder (NCAE) framework, which achieved significant state-of-the-art improvements in experiments on three real-world datasets.

In recent years, deep neural networks have yielded state-of-the-art performance on several tasks. Although some recent works have focused on combining deep learning with recommendation, we highlight three issues of existing models. First, these models cannot work on both explicit and implicit feedback, since the network structures are specially designed for one particular case. Second, due to the difficulty on training deep neural networks, existing explicit models do not fully exploit the expressive potential of deep learning. Third, neural network models are easier to overfit on the implicit setting than shallow models. To tackle these issues, we present a generic recommender framework called Neural Collaborative Autoencoder (NCAE) to perform collaborative filtering, which works well for both explicit feedback and implicit feedback. NCAE can effectively capture the subtle hidden relationships between interactions via a non-linear matrix factorization process. To optimize the deep architecture of NCAE, we develop a three-stage pre-training mechanism that combines supervised and unsupervised feature learning. Moreover, to prevent overfitting on the implicit setting, we propose an error reweighting module and a sparsity-aware data-augmentation strategy. Extensive experiments on three real-world datasets demonstrate that NCAE can significantly advance the state-of-the-art.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes