Matrix factorization with a sigmoid-like loss control

Yuan -Yuan Xu, Hui Xiao,Heng-Ru Zhang,Wei-Zhi Wu,Fan Min

NEUROCOMPUTING(2024)

引用 0|浏览4
暂无评分
摘要
Matrix factorization is one of the fundamental approaches of recommender systems. With the popular L2 loss, learning models tend to overfit significantly deviated predictions. However, predicting the actual rating of 5 as 1 or 2 makes no essential difference in the application. In this paper, we design a sigmoid-like function to control the loss of each individual prediction, which has two advantages. First, it reduces the loss corresponding to significantly deviated predictions. Therefore, the impact of these predictions, some of which may be caused by outliers, is also reduced. Second, it is independent of two classical over -fitting control techniques using regular terms and validation data, respectively. Hence, it can be combined with them to form a more powerful method. Experiments are undertaken on six benchmark datasets in comparison with different losses. Results show that the proposed loss function has good performance in terms of MAE, RMSE, and NDCG, however not so good in terms of HR and MAP.
更多
查看译文
关键词
Loss function,Matrix factorization,Over-fitting,Recommender system
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要