Adversarial Collaborative Auto-encoder for Top-N Recommendation

TL;DR

This paper introduces an adversarial training framework for neural recommendation models to improve robustness against noisy user feedback while maintaining high recommendation performance.

Contribution

It proposes a general adversarial training approach applied to collaborative auto-encoders, enhancing robustness and performance in top-N recommendation tasks.

Findings

Outperforms state-of-the-art recommendation methods on benchmark datasets.

Effectively mitigates the impact of noisy user feedback.

Provides insights into the tradeoff between robustness and accuracy.

Abstract

During the past decade, model-based recommendation methods have evolved from latent factor models to neural network-based models. Most of these techniques mainly focus on improving the overall performance, such as the root mean square error for rating predictions and hit ratio for top-N recommendation, where the users' feedback is considered as the ground-truth. However, in real-world applications, the users' feedback is possibly contaminated by imperfect user behaviours, namely, careless preference selection. Such data contamination poses challenges on the design of robust recommendation methods. In this work, to address the above issue, we propose a general adversial training framework for neural network-based recommendation models, which improves both the model robustness and the overall performance. We point out the tradeoffs between performance and robustness enhancement with detailed instructions on how to strike a balance. Specifically, we implement our approach on the collaborative auto-encoder, followed by experiments on three public available datasets: MovieLens-1M, Ciao, and FilmTrust. We show that our approach outperforms highly competitive state-of-the-art recommendation methods. In addition, we carry out a thorough analysis on the noise impacts, as well as the complex interactions between model nonlinearity and noise levels. Through simple modifications, our adversarial training framework can be applied to a host of neural network-based models whose robustness and performance are expected to be both enhanced.