Hybrid music recommendation with graph neural networks

被引:2
作者
Bevec, Matej [1 ]
Tkalcic, Marko [2 ]
Pesek, Matevz [1 ]
机构
[1] Univ Ljubljana, Fac Comp & Informat Sci, Vecna pot 113, Ljubljana 1000, Slovenia
[2] Univ Primorska, Fac Math, Nat Sci & Informat Technol, Glagoljaska 8, Koper 6000, Slovenia
关键词
Embeddings; Music recommendation systems; Graph neural networks; PinSage; Beyond-accuracy evaluation; SYSTEMS; LISTEN;
D O I
10.1007/s11257-024-09410-4
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Modern music streaming services rely on recommender systems to help users navigate within their large collections. Collaborative filtering (CF) methods, that leverage past user-item interactions, have been most successful, but have various limitations, like performing poorly among sparsely connected items. Conversely, content-based models circumvent the data-sparsity issue by recommending based on item content alone, but have seen limited success. Recently, graph-based machine learning approaches have shown, in other domains, to be able to address the aforementioned issues. Graph neural networks (GNN) in particular promise to learn from both the complex relationships within a user interaction graph, as well as content to generate hybrid recommendations. Here, we propose a music recommender system using a state-of-the-art GNN, PinSage, and evaluate it on a novel Spotify dataset against traditional CF, graph-based CF and content-based methods on a related song prediction task, venturing beyond accuracy in our evaluation. Our experiments show that (i) our approach is among the top performers and stands out as the most well rounded compared to baselines, (ii) graph-based CF methods outperform matrix-based CF approaches, suggesting that user interaction data may be better represented as a graph and (iii) in our evaluation, CF methods do not exhibit a performance drop in the long tail, where the hybrid approach does not offer an advantage.
引用
收藏
页码:1891 / 1928
页数:38
相关论文
共 64 条
[1]   Toward the next generation of recommender systems: A survey of the state-of-the-art and possible extensions [J].
Adomavicius, G ;
Tuzhilin, A .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2005, 17 (06) :734-749
[2]   Look, Listen and Learn [J].
Arandjelovic, Relja ;
Zisserman, Andrew .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :609-617
[3]   A review on deep learning for recommender systems: challenges and remedies [J].
Batmaz, Zeynep ;
Yurekli, Ali ;
Bilge, Alper ;
Kaleli, Cihan .
ARTIFICIAL INTELLIGENCE REVIEW, 2019, 52 (01) :1-37
[4]  
Bell R., 2007, SIGKDD Explorations Newsletter, V9, P75, DOI [DOI 10.1145/1345448.1345465, 10.1145/1345448.1345465]
[5]  
Berg R.v.d., 2017, ARXIV
[6]   The Music Streaming Sessions Dataset [J].
Brost, Brian ;
Mehrotra, Rishabh ;
Jehan, Tristan .
WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, :2594-2600
[7]  
Celma Herrada O., 2006, CEUR WORKSHOP P
[8]  
Choi K., 2018, ARXIV
[9]  
Cramer J, 2019, INT CONF ACOUST SPEE, P3852, DOI 10.1109/ICASSP.2019.8682475
[10]  
Eksombatchai C, 2018, WEB CONFERENCE 2018: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW2018), P1775