1School of Management, Hefei University of Technology, Hefei 230009, China 2Key Laboratory of Process Optimization and Intelligent Decision-making of Ministry of Education, Hefei 230009, China
[Objective] This paper proposes a model to predict online ratings with the help of network representation learning and XGBoost—N2V_XGB. [Methods] First, we retrieved metadata and existing online rating data. Then, we extracted and merged the similarity weights of collected data to construct a homogenous relationship network. Third, we used network representation learning to automatically extract user and item features. Finally, we input these data to XGBoost, and obtained the best model with iteratively training. [Results] The MAE and RMSE of the proposed N2V_XGB model were 0.686 7 and 0.873 7, which were lower than the four classic models. [Limitations] We did not make good use of time features and the prediction results did not reflect time-series changes. [Conclusions] The proposed N2V_XGB model effectively address the data sparseness issues and improve the prediction accuracy of user ratings.
( Li Xiaoju. Research on Data Sparsity and Cold-Start Problem in Collaborative Filtering Recommender System[D]. Shanghai: East China Normal University, 2018.)
( Leng Yajun, Lu Qing, Liang Changyong. Survey of Recommendation Based on Collaborative Filtering[J]. Pattern Recognition and Artificial Intelligence, 2014,27(8):720-734.)
( Gong An, Gao Yun, Gao Hongfu. A Collaborative Filtering Recommendation Algorithm Based on Ratings of Item Attributes[J]. Computer Engineering and Science, 2015,37(12):2366-2371.)
( Ding Shaoheng, Ji Donghong, Wang Lulu. Collaborative Filtering Recommendation Algorithm Based on User Attributes and Scores[J]. Computer Engineering and Design, 2015,36(2):487-491, 497.)
[5]
Davoudi A, Chatterjee M. Product Rating Prediction Using Trust Relationships in Social Networks[C]// Proceedings of the 13th IEEE Annual Consumer Communications & Networking Conference, Las Vegas, NV, USA. IEEE, 2016.
( Xiao Zhiyu, Zhai Yuqing. Improved Rating Prediction Model Basing on Trust Network and Random Walk Strategy[J]. Journal of Nanjing University of Science and Technology, 2015,39(5):602-608.)
[7]
Davoudi A, Chatterjee M. Social Trust Model for Rating Prediction in Recommender Systems: Effects of Similarity, Centrality, and Social Ties[J]. Online Social Networks and Media, 2018,7:1-11.
doi: 10.1016/j.osnem.2018.05.001
( Xue Fuliang, Liu Junling. Improving Collaborative Filtering Recommendation Based on Trust Relationship Among Users[J]. Data Analysis and Knowledge Discovery, 2017,1(7):90-99.)
[9]
Ren Y, Li G, Zhang J, et al. Lazy Collaborative Filtering for Data Sets with Missing Values[J]. IEEE Transactions on Cybernetics, 2013,43(6):1822-1834.
pmid: 23757575
( Li Zheng, Duan Lei. Improved Hybrid Recommendation Approach Based on User Interest Ratings Filling[J]. Advanced Engineering Sciences, 2019,51(1):189-196.)
( Peng Shi, Zhou Zhibin, Wang Guojun. Collaborative Filtering Algorithm Based on Rating Matrix Pre-filling[J]. Computer Engineering, 2013,39(1):175-178.)
doi: 10.3969/j.issn.1000-3428.2013.01.037
[12]
Koren Y, Bell R, Volinsky C. Matrix Factorization Techniques for Recommender Systems[J]. Computer, 2009,42(8):30-37.
[13]
Li L, Zhang Y J. FastNMF: Highly Efficient Monotonic Fixed-Point Nonnegative Matrix Factorization Algorithm with Good Applicability[J]. Journal of Electronic Imaging, 2009,18(3):033004.
doi: 10.1117/1.3184771
( Bi Hualing, Zhou Wei, Lu Fuqiang. Bias Based Matrix Factorization Recommender Techniques[J]. Application Research of Computers, 2018,35(10):2928-2931, 2964.)
( Chen Ye, Liu Zhiqiang. Research on Improved Recommendation Algorithm Based on LFM Matrix Factorization[J]. Computer Engineering and Applications, 2019,55(2):116-120.)
( He Jinlin, Liu Xuejun, Xu Xinyan, et al. Implicit Feedback Recommendation Model Combining Node2Vec and Deep Neural Networks[J]. Computer Science, 2019,46(6):41-48.)
( Ma Chunping, Chen Wenliang. Personalized Model for Rating Prediction Based on Review Analysis[J]. Acta Scientiarum Naturalium Universitatis Pekinensis, 2016,52(1):165-170.)
doi: 10.13209/j.0479-8023.2016.011
( Zhang Hongli, Liu Jiying, Yang Sinan, et al. Predicting Online Users’ Ratings with Comments[J]. Data Analysis and Knowledge Discovery, 2017,1(8):48-58.)
[20]
Grover A, Leskovec J. Node2Vec: Scalable Feature Learning for Networks[C]// Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. ACM, 2016: 855-864.
[21]
Chen T, Guesintr C. XGBoost: A Scalable Tree Boosting System[C]// Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. ACM, 2016: 785-794.
[22]
Deshpande M, Karypis G. Item-Based Top-N Recommendation Algorithms[J]. ACM Transactions on Information Systems, 2004,22(1):143-177.
doi: 10.1145/963770.963776
[23]
Liang L, Tang R. An Improved Collaborative Filtering Algorithm Based on Node2Vec[C]// Proceedings of the 2nd International Conference on Computer Science and Artificial Intelligence. 2018: 218-222.
( Cui Yan, Qi Wei, Pang Hailong, et al. Extreme Gradient Boosting Recommendation Algorithm with Collaborative Filtering[J]. Application Research of Computer, 2020,37(1):62-65.)