A Nonlinear Feature Fusion-Based Rating Prediction Algorithm in Heterogeneous Network
Yi Lei, Shujuan Ji, Lingmei Ren, Rui Su, Yongquan Liang · IEEE Transactions on Computational Social Systems · 2021
Due to the flexibility of heterogeneous information networks (HINs) in modeling heterogeneous data, researchers begin to use it to integrate the objects and relationships in recommender systems. However, how to abstract and exploit effective information and apply the information to recommender systems is a challenge. To fully mine nodes' structural features and better integrate these features simultaneously, we present a nonlinear feature fusion-based rating prediction algorithm. This algorithm first uses a meta-path-based HIN embedding model to extract the nodes' structural features. Then, the structural features are converted by a nonlinear fusion method. Finally, the fused features are input into the multilayer perceptron to achieve rating prediction. Experiments on real-life data sets, such as Movielens-100k, Yelp, Douban Book, and Douban Movie, are designed to prove the performance of the proposed model. Experimental results on the four data sets reveal that our algorithm is superior to the baselines.