English

Song Recommendation with Non-Negative Matrix Factorization and Graph Total Variation

Machine Learning 2016-01-14 v2 Information Retrieval Machine Learning Data Analysis, Statistics and Probability

Abstract

This work formulates a novel song recommender system as a matrix completion problem that benefits from collaborative filtering through Non-negative Matrix Factorization (NMF) and content-based filtering via total variation (TV) on graphs. The graphs encode both playlist proximity information and song similarity, using a rich combination of audio, meta-data and social features. As we demonstrate, our hybrid recommendation system is very versatile and incorporates several well-known methods while outperforming them. Particularly, we show on real-world data that our model overcomes w.r.t. two evaluation metrics the recommendation of models solely based on low-rank information, graph-based information or a combination of both.

Keywords

Cite

@article{arxiv.1601.01892,
  title  = {Song Recommendation with Non-Negative Matrix Factorization and Graph Total Variation},
  author = {Kirell Benzi and Vassilis Kalofolias and Xavier Bresson and Pierre Vandergheynst},
  journal= {arXiv preprint arXiv:1601.01892},
  year   = {2016}
}

Comments

Code available at: https://github.com/kikohs/recog

R2 v1 2026-06-22T12:25:34.713Z