AICF: Attention-based item collaborative filtering. (April 2020)

• Record Type: Journal Article
• Authors: Lv, Yanxia; Zheng, Ying; Wei, Fangna; Wang, Cong; Wang, Cuirong
• Is Part Of: Advanced engineering informatics. Volume 44(2020)
• Publication Date: 2020-04
• Subjects: Collaborative filtering -- Item-based CF -- Recommender systems -- Attention mechanisms; Computer-aided engineering -- Periodicals; Engineering -- Data processing -- Periodicals; 620.00285
• DOI: 10.1016/j.aei.2020.101090
• Languages: English
• ISSNs: 1474-0346

Abstract

Abstract: Item-to-item collaborative filtering (short for ICF) has been widely used in ecommerce websites due to his interpretability and simplicity in real-time personalized recommendation. The focus of ICF is to calculate the similarity between items. With the rapid development of machine learning in recent years, it takes similarity model instead of cosine similarity and Pearson coefficient to calculate the similarity between items in recommendation. However, the existing similarity models can not sufficient to express the preferences of users for different items. In this work, we propose a novel attention-based item collaborative filtering model(AICF) which adopts three different attention mechanisms to estimate the weights of historical items that users have interacted with. Compared with the state-of-the-art recommendation models, the AICF model with simple attention mechanism Self-Attention can better estimate the weight of historical items on non-sparse data sets. Due to depth models can model complex connection between items, our model with the more complex Transformer achieves superior recommendation performance on sparse data. Extensive experiments on ML-1M and Pinterest-20 show that the proposed model greatly outperforms other novel models in recommendation accuracy and provides users with personalized recommendation list more in line with their interests.