A Recommendation System Framework to Generalize AutoRec and Neural Collaborative Filtering

Abstract

AutoRec and neural collaborative filtering (NCF) are two widely used neural network-based frameworks in the recommendation system literature. In this paper, we show that these two apparently very different frameworks have a lot in common. We propose a general neural network-based frame-work, which gives us flexibility in choosing elements in the input sources, prediction functions, etc. Then, we show that AutoRec and NCF are special forms of our generalized framework. In our experimental results, first, we compare different variants of NCF and Autorec. Then, we indicate that it is necessary to use our general framework since there is no specific structure that performs well in all datasets. Finally, we show that by choosing the right elements, our framework outperforms the state-of-the-art methods with complicated structures.