Knowledge Graph Based Recommender for Automatic Playlist Continuation
Abstract
:1. Introduction
2. Related Work
3. Methodology
3.1. Augmentation
Algorithm 1: Data Augmentation. |
3.2. Representation
3.2.1. Node Classification Embeddings
3.2.2. Link Prediction Embeddings
3.3. Recommendation
User Behavior over Time
Algorithm 2: Learning the learnable parameter w. |
- For the first track , we will not retrieve any tracks from the embedding space, as .
- For the second track , we will not retrieve any tracks from the embedding space, as .
- For the third track , we will retrieve tracks from the embedding space.…
- For the last track , we will retrieve tracks from the embedding space.
4. Evaluation and Results
- Semi-Supervised—Representation: To evaluate the quality of our model’s representations, we leverage the graph structure by masking part of the graph during training and adding negative samples. The model is trained to reconstruct the missing elements of the graph and we evaluate how closely the reconstructed graph matches the actual graph.
- Gold Standard Data—Recommendation: To evaluate the recommendation performance of our model, we use a test set consisting of playlists, with the original tracks added by the user provided as ground truth. We compare the tracks recommended by our model for each playlist with the original tracks added by the user to evaluate the accuracy of our recommendations.
Discussion
5. Conclusions and Future Work
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Henning, V.; Reichelt, J. Mendeley-a last.fm for research? In Proceedings of the 2008 IEEE Fourth International Conference on eScience, Indianapolis, IN, USA, 7–12 December 2008; pp. 327–328. [Google Scholar]
- Bennett, J.; Lanning, S. The Netflix Prize. In Proceedings of the KDD Cup Workshop 2007, San Jose, CA, USA, 12 August 2007; pp. 3–6. [Google Scholar]
- Backstrom, L.; Leskovec, J. Supervised random walks: Predicting and recommending links in social networks. In Proceedings of the Fourth Association for Computing Machinery International Conference on Web Search and Data Mining, WSDM’11, Hong Kong, China, 9–12 February 2011; pp. 635–644. [Google Scholar]
- Amit, S. Introducing the Knowledge Graph: Things, Not Strings. 2012. Available online: https://sonic.northwestern.edu/introducing-the-knowledge-graph-things-not-strings-official-google-blog/ (accessed on 13 September 2023).
- Raimond, Y.; Abdallah, S.; Sandler, M.; Giasson, F. The Music Ontology. In Proceedings of the 8th International Society for Music Information Retrieval Conference, Graz, Austria, 23–27 September 2007; ISMIR: San Francisco, CA, USA, 2007; pp. 417–422. [Google Scholar]
- Saravanou, A.; Tomasi, F.; Mehrotra, R.; Lalmas, M. Multi-Task Learning of Graph-based Inductive Representations of Music Content. In Proceedings of the 22nd International Society for Music Information Retrieval Conference, Genève, Switzerland, 7–11 November 2021; ISMIR: San Francisco, CA, USA, 2021. [Google Scholar]
- Zhou, J.; Cui, G.; Hu, S.; Zhang, Z.; Yang, C.; Liu, Z.; Wang, L.; Li, C.; Sun, M. Graph neural networks: A review of methods and applications. AI Open 2020, 1, 57–81. [Google Scholar] [CrossRef]
- Jannach, D.; Zanker, M.; Felfernig, A.; Friedrich, G. Recommender Systems: An Introduction; Cambridge University Press: Cambridge, UK, 2010. [Google Scholar]
- Guo, Q.; Zhuang, F.; Qin, C.; Zhu, H.; Xie, X.; Xiong, H.; He, Q. A Survey on Knowledge Graph-Based Recommender Systems. IEEE Trans. Knowl. Data Eng. 2020, 34, 3549–3568. [Google Scholar] [CrossRef]
- Chen, C.W.; Lamere, P.; Schedl, M.; Zamani, H. Recsys challenge 2018. In Proceedings of the 12th Association for Computing Machinery Conference on Recommender Systems, New York, NY, USA, 2–7 September 2018. [Google Scholar]
- Ferraro, A.; Bogdanov, D.; Yoon, J.; Kim, K.; Serra, X. Automatic playlist continuation using a hybrid recommender system combining features from text and audio. In Proceedings of the 12th Association for Computing Machinery Conference on Recommender Systems, New York, NY, USA, 2–7 September 2018. [Google Scholar]
- Ludewig, M.; Kamehkhosh, I.; Landia, N.; Jannach, D. Effective Nearest-Neighbor Music Recommendations. In Proceedings of the 12th Association for Computing Machinery Conference on Recommender Systems, New York, NY, USA, 2–7 September 2018. [Google Scholar]
- Maksims, V.; Rai, H.; Cheng, Z.; Wu, G.; Lu, Y.; Sanner, S. Two-Stage Model for Automatic Playlist Continuation at Scale. In Proceedings of the 12th Association for Computing Machinery Conference on Recommender Systems, New York, NY, USA, 2–7 September 2018. [Google Scholar]
- Teinemaa, I.; Tax, N.; Bentes, C. Automatic Playlist Continuation through a Composition of Collaborative Filters. arXiv 2018, arXiv:1808.04288. [Google Scholar]
- Kelen, D.; Berecz, D.; Béres, F.; Benczúr, A.A. Efficient K-NN for Playlist Continuation. In Proceedings of the 12th Association for Computing Machinery Conference on Recommender Systems, New York, NY, USA, 2–7 September 2018. [Google Scholar]
- Maillet, F.; Eck, D.; Desjardins, G.; Lamere, P. Steerable Playlist Generation by Learning Song Similarity from Radio Station Playlists. In Proceedings of the 10th International Society for Music Information Retrieval Conference, Kobe, Japan, 26–30 October 2009; ISMIR: San Francisco, CA, USA, 2010; pp. 345–350. [Google Scholar]
- Boom, C.D.; Agrawal, R.; Hansen, S.; Kumar, E.; Yon, R.; Chen, C.W.; Demeester, T.; Dhoedt, B. Large-scale user modeling with recurrent neural networks for music discovery on multiple time scales. Multimed. Tools Appl. 2017, 77, 15385–15407. [Google Scholar] [CrossRef]
- Wang, X.; Wang, Y. Improving Content-Based and Hybrid Music Recommendation Using Deep Learning. In Proceedings of the 22nd Association for Computing Machinery International Conference on Multimedia, New York, NY, USA, 3–7 November 2014; pp. 627–636. [Google Scholar]
- Sun, Z.; Yang, J.; Feng, K.; Fang, H.; Qu, X.; Ong, Y.S. Revisiting Bundle Recommendation: Datasets, Tasks, Challenges and Opportunities for Intent-Aware Product Bundling. In Proceedings of the 45th International ACM SIGIR Conference on Research and Development in Information Retrieval, New York, NY, USA, 11–15 July 2022; pp. 2900–2911. [Google Scholar]
- Chang, J.; Gao, C.; He, X.; Jin, D.; Li, Y. Bundle Recommendation with Graph Convolutional Networks. In Proceedings of the 43rd International Association for Computing Machinery SIGIR Conference on Research and Development in Information Retrieval, New York, NY, USA, 20–30 July 2020; pp. 1673–1676. [Google Scholar]
- Xiao, Y.; Pei, Q.; Xiao, T.; Yao, L.; Liu, H. MutualRec: Joint friend and item recommendations with mutualistic attentional graph neural networks. J. Netw. Comput. Appl. 2021, 177, 102954. [Google Scholar] [CrossRef]
- Gao, C.; Zheng, Y.; Li, N.; Li, Y.; Qin, Y.; Piao, J.; Quan, Y.; Chang, J.; Jin, D.; He, X.; et al. A Survey of Graph Neural Networks for Recommender Systems: Challenges, Methods, and Directions. ACM Trans. Recomm. Syst. 2021, 1, 1–51. [Google Scholar]
- Xu, K.; Hu, W.; Leskovec, J.; Jegelka, S. How powerful are graph neural networks? In Proceedings of the 7th International Conference on Learning Representations ICLR, New Orleans, LA, USA, 6–9 May 2019. [Google Scholar]
- Wang, B.; Cai, W. Knowledge-Enhanced Graph Neural Networks for Sequential Recommendation. Information 2020, 11, 388. [Google Scholar] [CrossRef]
- Hamilton, W.L.; Ying, R.; Leskovec, J. Inductive Representation Learning on Large Graphs. In Proceedings of the 31st International Conference on Neural Information Processing Systems, Red Hook, NY, USA, 4–9 December 2017; pp. 1025–1035. [Google Scholar]
- Loshchilov, I.; Hutter, F. Decoupled Weight Decay Regularization. arXiv 2019, arXiv:1711.05101. [Google Scholar]
- Loshchilov, I.; Hutter, F. SGDR: Stochastic Gradient Descent with Warm Restarts. arXiv 2017, arXiv:1711.05101. [Google Scholar]
- Dodge, Y. Mean Squared Error. In The Concise Encyclopedia of Statistics; Springer: New York, NY, USA, 2008; Chapter M; pp. 337–339. [Google Scholar]
- Wang, Y.; Wang, L.; Li, Y.; He, D.; Liu, T.Y.; Chen, W. A Theoretical Analysis of NDCG Type Ranking Measures. In Proceedings of the 26th Annual Conference on Learning Theory, Princeton, NJ, USA, 12–14 June 2013. [Google Scholar]
- Fey, M.; Lenssen, J.E. Fast Graph Representation Learning with PyTorch Geometric. In Proceedings of the ICLR Workshop on Representation Learning on Graphs and Manifolds, New Orleans, LA, USA, 6 May 2019. [Google Scholar]
- Qdrant—Vector Database. Available online: https://qdrant.tech/ (accessed on 13 September 2023).
Evaluation | Metric | Representation Approach | |
---|---|---|---|
Node Classification | Link Prediction | ||
Semi-Supervised | Accuracy | 0.789 | 0.657 |
Recall | 0.712 | 0.587 | |
Gold Standard | R-Precision | 0.568 | 0.127 |
NDCG | 0.758 | 0.089 |
Rank | Team | R-Precision | NDCG |
---|---|---|---|
1 | vl6 | 0.223 | 0.394 |
2 | Creamy Fireflies | 0.220 | 0.385 |
3 | KAENEN | 0.209 | 0.375 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2023 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
Share and Cite
Ivanovski, A.; Jovanovik, M.; Stojanov, R.; Trajanov, D. Knowledge Graph Based Recommender for Automatic Playlist Continuation. Information 2023, 14, 510. https://doi.org/10.3390/info14090510
Ivanovski A, Jovanovik M, Stojanov R, Trajanov D. Knowledge Graph Based Recommender for Automatic Playlist Continuation. Information. 2023; 14(9):510. https://doi.org/10.3390/info14090510
Chicago/Turabian StyleIvanovski, Aleksandar, Milos Jovanovik, Riste Stojanov, and Dimitar Trajanov. 2023. "Knowledge Graph Based Recommender for Automatic Playlist Continuation" Information 14, no. 9: 510. https://doi.org/10.3390/info14090510
APA StyleIvanovski, A., Jovanovik, M., Stojanov, R., & Trajanov, D. (2023). Knowledge Graph Based Recommender for Automatic Playlist Continuation. Information, 14(9), 510. https://doi.org/10.3390/info14090510