CA-NCF: A Category Assisted Neural Collaborative Filtering Approach for Personalized Recommendation

Abstract

In the big data environment, the sparsity problem of collaborative filtering recommendation algorithm becomes increasingly serious, which has a great impact on the accuracy of recommendation. In some recent researches, item categories were input into neural networks to enrich the embedded information in the process of training. However, these methods generally simultaneously use item categories and items as embedded information, which may weaken the importance of item categories. Therefore, this paper proposes a neural collaborative filtering method based on category assistance. In this method, the interaction between item category and user is first modeled by Neural Matrix Factorization ((Neu-MF)), which raises the impact of item category in the relationship extraction between items and users. Then, only the items in the trained results of categories are used in an optimized Neural Collaborative Filtering (NCF) framework for item recommendation. Based on the real ecommerce data set from Alibaba, experimental results show that this method obtains better result in the Hit Rate (HR) and the Normalized Discounted Cumulative Gain (NDCG) compared with other baseline methods.