• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2023, Vol. 45 ›› Issue (10): 1891-1900.

• Artificial Intelligence and Data Mining • Previous Articles    

Deep input-aware factorization machine based on Setwise ranking

LIU Tong,ZHOU Ning-ning   

  1. (School of Computer Science,Nanjing University of Posts and Telecommunications,Nanjing 210023,China) 
  • Received:2022-05-09 Revised:2022-07-19 Accepted:2023-10-25 Online:2023-10-25 Published:2023-10-17

Abstract: SetRank is a novel Setwise Bayesian collaborative ranking model that can model implicit feedback data more closely to real-world scenarios. However, SetRank only considers collaborative information and lacks effective use of content information. In order to solve the above problems, a factorization machine model based on Setwise ranking, named SRFMs, is proposed. Drawing on the Setwise ranking proposed in SetRank to solve the problem of missing negative samples in implicit feedback, a factorization machine is chosen as a predictor to model content information, and model user preferences from the perspective of optimal item ranking. Furthermore, to improve the fixed feature representation and the lack of higher-order feature interactions in standard FM, inspired by IFM, SR-DIFM model is constructed to incorporate the SRFMs by combining input-aware networks with neural networks. Experimental results on two real-world datasets demonstrate that the proposed  model outperforms the state-of-art model in terms of evaluation metrics including HR, NDCG and mAP, and can improve the accuracy of recommendations which make better use of user and item content information while solving the recommendation problem under implicit feedback. 

Key words: implicit feedback, content information, learning to rank, factorization machine, neural network