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

01/08/2016 ∙ by Kirell Benzi, et al. ∙ 0

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.



There are no comments yet.


page 1

page 2

page 3

page 4

Code Repositories

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

Recommending movies on Netflix, friends on Facebook, or jobs on LinkedIn are tasks gaining an increasing interest over the last years. Low-rank matrix factorization techniques [1] where amongst the winners of the famous Netflix prize, involving explicit user ratings as input. Similar techniques were soon used in order to solve implicit feedback problems, where item preferences were implied for example by the actions of a user [2, 3]. Specifically regarding songs and playlists recommendation, various techniques have been proposed, ranging from pure content-based methods [4] to hybrid models [5]. A comprehensive review of related algorithms can be found in [6, 7]. Recently, graph regularization was proposed in order to enhance the quality of matrix completion problems [8, 9, 10].

The contributions of this paper are as follows:

  • A mathematically sound hybrid system that benefits from collaborative and content-based filtering.

  • The introduction of a new graph regularization term (TV) [11] in the context of recommendation that outperforms the widely used Tikhonov regularization. [10, 9],

  • A well-defined iterative optimization scheme based on proximal splitting methods [12].

Numerical experiments demonstrate the performance of our proposed recommender system.

Figure 1: The architecture of our playlist recommender system.

2 Our Recommendation Algorithm

Suppose we are given playlists, each containing some of songs. We define matrix as in [13, 3], that has a value if playlist contains song , otherwise. We also define a weight mask that has a ”confidence” value one if the entry is , and a small value , otherwise (we use ). This follows the example of implicit feedback problems [2], since a zero in matrix

does not mean that the corresponding song is irrelevant to the playlist, but that it is less probably relevant.

The goal of the training step is to find an approximate low-rank representation , where , non-negative and with small . This problem is known as Non-Negative Matrix Factorization (NMF) and has drawn a lot of attention after the seminal work [14]. The advantage of NMF over other factorization techniques is that the approximation is only based on adding factors, a property explained as learning the parts of objects [14], in this case the playlists. NMF comes to the cost of being NP-hard [15], so sophisticated regularization is important for finding a good local minimum. In our problem we use outside information given by the songs and playlists graphs to give structure to the factors and . Our model is formulated as


where is the pointwise multiplication operator and . We use a weighted Kullback-Leibler (KL) divergence as a distance measure between and , that has been shown to be more accurate than the Frobenius norm for various NMF settings [16]. The second term is the TV of the rows of on the playlists graph, so penalizing it promotes piecewise constant signals [11]. Similarly with the third term for columns of . Eventually, the proposed model leverages the works of [9, 16], and extends them to graphs using the TV semi-norm.

Graph Regularization with Total Variation. In our NMF-based recommender, each playlist is represented in a low-dimensional space by a row of the matrix . In order to learn better low-rank representations of the playlists, we also impose the pairwise similarities of the playlists on their corresponding low-rank representations. We can see this from the definition of the TV regularization term, . Hence, when two playlists are similar then they are also well-connected on the graph and the weight of the edge connecting these two playlists is large (here

). Moreover, any large distance between the corresponding low-dimensional representation vectors

is penalized, forcing to stay close in the low-dimensional space. In a similar way, each song is represented in a low-dimensional space by a column of the matrix . If two songs are close (), so will be with the graph regularization .

A similar idea has been used in [10] by incorporating the graph information through Tikhonov regularization, i.e. with the Dirichlet energy term . However, the latter promotes smooth changes between the columns of , while the graph TV term penalization promotes piecewise constant signals with potentially sharp transitions between columns and . This is advantageous in applications where well separated classes are sought, for example in clustering [17], or in our recommendation system where similar playlists might belong to different categories.

As we demonstrate in Sec. 4, the use of the graphs of songs and playlists improve significantly the recommendations, while the results are better when the more forgiving TV term is used instead of Tikhonov regularization.

Primal-dual optimization. Optimization problem (1) is globally non-convex, but separately convex w.r.t. and . A standard strategy is thus to optimize for fixed , then optimize for fixed , and repeat until convergence. We describe here the proposed optimization algorithm w.r.t. for fixed based on [18, 12, 16]. The same algorithm can be applied to for fixed . Let us rewrite problem (1) as:




where is the graph gradient operator [17], with being the number of edges in the graph of . Using the conjugate functions and of and , (2) is equivalent to the saddle-point problem:


where , . Let us now introduce the proximal terms and the time steps , :


The iterative scheme is thus for :


where prox is the proximal operator [12] and . For our problem we have chosen the standard Arrow-Hurwicz time steps and , where is here the operator norm.

The proximal solutions (7) and (8) are given by:


where shrink is the soft shrinkage operator [19]. Note that the same algorithm could be used for Tikhonov regularization, i.e. replacing by by just changing the first proximal (10) to . In [10] this regularization is used along with a symmetric version of the KL divergence, however the latter has no analytic solution unlike the one we use in this work. As a result their objective function does not fit an efficient primal dual optimization scheme like the one we propose. We thus choose to keep the non symmetric KL model, denoted as GNMF in this paper, in order to compare the TV versus Tikhonov regularization.

Recommending songs. Once we have learned matrices and by solving (1), we wish to recommend a new playlist given a few songs (see Fig. 1). We also want to make real-time recommendations, so we design here a fast recommender function as follows:

Given the songs , we first find a good representation of the query on the learned low-rank space of playlists by solving a regularized least squares problem:
. The latter enjoys an analytic solution that is cheap to compute as is small (we use ).

The recommended playlist can benefit from the playlists that have similar representations as the one of the query, thus we use the weighted sum as the representation of the recommended playlist in the low dimensional space. Here the weights are defined as and depend on the distance of from other playlists representations, while . The final recommended playlist uses the low-rank representation :


Note finally that the recommended playlist is not binary, but with continued values that serve as song rankings.

3 Graphs of Playlists and Songs

Playlists Graph. The playlists graph naturally encodes pairwise similarities between playlists. The set of nodes of this graph is the set of playlists and the edge weight provides the proximity between two playlists. A large weight (here ) implies a strong proximity between the playlists. In this work, the edge weight of the playlists graph uses both “outside” information, i.e. the meta-data, and “inside” information, i.e. the songs that form the playlists. As meta-data, we use the predefined Art of the Mix playlist categories [20] onto which users label their mixes. The edge weight of the playlists graph is thus defined as follows:

where stands for playlist category, is the row of matrix and

is the cosine similarity distance between the vectors of the songs of the two playlists. In our case, the cosine similarity is the ratio between the songs in common and the square root of the product of the lengths of the two playlists. The two positive parameters

with allow to weight the importance of the playlist labels against their element-wise similarity. To control the edge density in each category and to give more flexibility to our recommendation model, we keep a random subset of of the edges between nodes of the same category. As we find experimentally, constitutes a good compromise, see Sec. 4.

The quality of the playlist graph is measured by partitioning the graph using the standard Louvain’s method [21]. The number of partitions is automatically given by the modularity dendrogram which is cut where the modularity is maximal. The graph used in Sec. 4 has a modularity of when using the cosine similarity () only. If we add the meta-data information by connecting of all playlist pairs within each category with , the modularity increases to .

Songs Graph.

The second graph used in our model is the graph of song similarity. It is created from a mixture of Echonest features extracted from the audio signal which we combine with meta-data information and social features for the track. Table 

1 gives a view of the features used to create the song graph.

High Level Features
acousticness Acoustic or electric?
valence Is the song positive or negative?
energy How energetic is the song?
liveness Is it a “live” recording?
speechiness How many spoken words?
danceability Is the song danceable?
tempo Normalized BPM.
instrumentalness Is the song instrumental?
Social Features
artist discovery How unexpectedly popular is the artist?
artist familiarity How familiar is the artist?
artist hotttnesss Is the artist currently popular?
song hotttnesss Is the song currently popular?
song currency How recently has it become popular?
Temporal Echonest Features
statistics on echonest segments Described in [22]
Metadata Features
genre ID3 genre extracted from tags given by LastFM api
Table 1: The features used to create graph of songs.

In order to improve the quality of our audio features, we trained a Large Margin Nearest Neighbors model [23] on the song genres extracted from the LastFm associated terms (tags). To extract real music genres we use the Levenshtein distance between those terms weighted by their popularity (according to LastFm) and the music genres defined in the ID3 tags.

Eventually, the songs graph is created using the nearest neighbors (here = 5) where the edge weight between two songs is given by for in the nearest neighbors of . The parameter acts as the scale parameter of the graph and is set to be the average distance of the neighbors. The obtained graph has a high modularity () and is quite pure with respect to song genres with around 65% of accuracy using an unsupervised

-NN classifier.

4 Experimental Results

In this section we validate our approach by comparing our model against three different recommender systems on a real world dataset. Our test dataset is extracted from the Art-of-the-Mix corpus created by McFee and al. in [20] onto which we extract the previously described features.

Assessing the quality of any music recommender systems is well-known to be a challenging problem [7]. In this work, we use a typical metric for recommender system with implicit feedback, Mean Percentage Ranking (MPR) described in [2] and the playlist category accuracy, that is the percentage of the recommended songs that have already been used in playlists from the requested category in the past.

Models. We first compare our model against a graphs-only based approach, labeled as Cosine only. For a given input, this model computes the -closest playlists (here ) using cosine similarity. Songs are recommended by computing a histogram of all the songs contained in these playlists weighted by the cosine similarity weight, as defined by eq. (11). The second model is NMF using KL divergence, labeled NMF [16]. The last model, GNMF [10] described in Sec. 2, is based on the KL divergence with Tikhonov regularization using the graphs of our model.

Queries. We test our model with three different types of queries. In all cases, a query contains input songs, and the system returns the top output songs as a playlist using eq. (11). The first type of queries, Random, contains completely randomly chosen songs from all categories and is solely used as a comparison baseline. The second type of queries, Test, picks randomly songs from a playlist of the test set. Lastly, Sampled, contains randomly chosen songs from a given category. It simulates a recommender system based on chosen playlist categories input by a user.

Training. We train our model using a randomly selected subset of of the playlists. As our model is not jointly convex, initialization may change the performance of the system, so we use the nowadays standard technique of NNDSVD [24] to get a good approximate solution. In all our experiments a value of the rank performs well, which is expected as each row has between and non-zero values. The best set of parameters and is found using a grid search using queries on the validation set. In order to prevent overfitting, we perform early stopping as soon as the MPR on the validation set ceases to increase.

Validation set. We create the “playlists” of the validation set by creating artificial queries from the different playlist categories. That is, for each category we randomly pick songs that have been previously used in user-made playlists labeled by the given category.

Results. The performance in terms of playlist category accuracy and MPR of the different models are reported in Table 2 and Table 3 respectively. As expected, for random category queries all models fail to return playlists from the categories of the input songs. At the same time, the performance of NMF as collaborative filtering without the graphs information is poor. This can be explained by the sparsity of the dataset, that only contains to non-zero elements per row, i.e. only 0.11-0.46% sparsity. Collaborative filtering models are known to perform better as more observed ratings are available [9]. The cosine model performs better in terms of category accuracy, as it directly uses the cosine distance between the input query and playlists from pure categories. However, its high MPR value shows that our model, albeit more complex, achieves better song recommendations.

only [16] [10]
Random 0.135 0.150 0.167 0.210 0.183
Test 0.530 0.236 0.332 0.544 0.646
Sampled 0.822 0.237 0.366 0.598 0.846
Table 2: Category accuracy for all models for different types of 3-song queries (higher is better). Results are averaged over 10 train/validation runs with 300 queries each.
only [16] [10]
Test 0.208 0.248 0.181 0.153 0.146
Sampled 0.226 0.319 0.211 0.164 0.074
Table 3: Mean percentage ranking (MPR) for all models for different types of 3-song queries (lower is better). Results are averaged over 10 train/validation runs with 300 queries each.

Figure 2: MPR for each playlist category on the test set. Our models use the same parameters of Table 3. Ambiguous categories such as Rock, Punk have the highest MPR on the test set. Our model outperforms significantly the others methods on those specific categories.

5 Conclusion

In this work we introduce a novel flexible song recommender system that combines collaborative filtering with playlist and song proximity information encoded by graphs. We use a primal-dual based optimization scheme to achieve a highly parallelizable algorithm with the potential to scale up to very large datasets. We choose graph TV instead of Tikhonov regularization and demonstrate the model’s superiority by comparing our system against three other recommendation models on a real music playlists dataset.



  • [1] Yehuda Koren, Robert Bell, and Chris Volinsky, “Matrix factorization techniques for recommender systems,” Computer, vol. 8, pp. 30–37, 2009.
  • [2] Yifan Hu, Yehuda Koren, and Chris Volinsky, “Collaborative filtering for implicit feedback datasets,” in In IEEE International Conference on Data Mining (ICDM 2008, 2008, pp. 263–272.
  • [3] S. Rendle, C. Freudenthaler, Z. Gantner, and L. Schmidt-Thieme, “BPR: Bayesian Personalized Ranking from Implicit Feedback,” in

    Proceedings of Conference on Uncertainty in Artificial Intelligence

    , 2009, pp. 452–461.
  • [4] Aaron Van den Oord, Sander Dieleman, and Benjamin Schrauwen, “Deep content-based music recommendation,” in Advances in Neural Information Processing Systems, 2013, pp. 2643–2651.
  • [5] Bo Shao, Dingding Wang, Tao Li, and Mitsunori Ogihara, “Music recommendation based on acoustic features and user access patterns,” Audio, Speech, and Language Processing, IEEE Transactions on, vol. 17, no. 8, pp. 1602–1611, 2009.
  • [6] Oscar Celma, “Music recommendation,” in Music Recommendation and Discovery, pp. 43–85. Springer Berlin Heidelberg, 2010.
  • [7] Geoffray Bonnin and Dietmar Jannach, “Automated Generation of Music Playlists: Survey and Experiments,” ACM Computing Surveys (CSUR), vol. 47, no. 2, pp. 1–35, nov 2014.
  • [8] Hao Ma, Dengyong Zhou, Chao Liu, Michael R Lyu, and Irwin King, “Recommender systems with social regularization,” in Proceedings of the fourth ACM international conference on Web search and data mining. ACM, 2011, pp. 287–296.
  • [9] V. Kalofolias, X. Bresson, M. Bronstein, and P. Vandergheynst, “Matrix Completion on Graphs,” arXiv, vol. preprint arXiv:1408.1717, 2014.
  • [10] Deng Cai, Xiaofei He, Jiawei Han, and Thomas S Huang, “Graph regularized nonnegative matrix factorization for data representation,” Pattern Analysis and Machine Intelligence, IEEE Transactions on, vol. 33, no. 8, pp. 1548–1560, 2011.
  • [11] L. I. Rudin, S. Osher, and E. Fatemi, “Nonlinear Total Variation Based Noise Removal Algorithms,” Physica D, vol. 60(1-4), pp. 259 – 268, 1992.
  • [12] P.L. Combettes and J.C. Pesquet, “Proximal Splitting Methods in Signal Processing,” Fixed-Point Algorithms for Inverse Problems in Science and Engineering, pp. 185–212, 2011.
  • [13] N. Hariri, B. Mobasher, and R. Burke, “Context-Aware Music Recommendation based on Latent Topic Sequential Patterns,” in Proceedings of ACM conference on Recommender systems, 2012, pp. 131–138.
  • [14] Daniel D Lee and H Sebastian Seung, “Learning the parts of objects by non-negative matrix factorization,” Nature, vol. 401, no. 6755, pp. 788–791, 1999.
  • [15] Stephen A Vavasis, “On the complexity of nonnegative matrix factorization,” SIAM Journal on Optimization, vol. 20, no. 3, pp. 1364–1377, 2009.
  • [16] F. Yanez and F. Bach, “Primal-Dual Algorithms for Non-negative Matrix Factorization with the Kullback-Leibler Divergence,” arXiv:1412.1788.
  • [17] X. Bresson, T. Laurent, D. Uminsky, and J.H. von Brecht, “Multiclass Total Variation Clustering,” Annual Conference on Neural Information Processing Systems (NIPS), pp. 1421–1429, 2013.
  • [18] A. Chambolle and T. Pock, “A First-Order Primal-Dual Algorithm for Convex Problems with Applications to Imaging,” Journal of Mathematical Imaging and Vision, vol. 40(1), pp. 120–145, 2011.
  • [19] D. Donoho, “De-Noising by Soft-Thresholding,” IEEE Transactions on Information Theory, vol. 41(33), pp. 613–627, 1995.
  • [20] Brian McFee and Gert RG Lanckriet, “Hypergraph Models of Playlist Dialects.,” in ISMIR. Citeseer, 2012, pp. 343–348.
  • [21] Vincent D Blondel, Jean-Loup Guillaume, Renaud Lambiotte, and Etienne Lefebvre, “Fast unfolding of communities in large networks,” Journal of Statistical Mechanics: Theory and Experiment, vol. 2008, no. 10, pp. P10008, 2008.
  • [22] Alexander Schindler and Andreas Rauber, “Capturing the temporal domain in echonest features for improved classification effectiveness,” in Adaptive Multimedia Retrieval: Semantics, Context, and Adaptation, pp. 214–227. Springer, 2014.
  • [23] Kilian Q Weinberger, John Blitzer, and Lawrence K Saul, “Distance metric learning for large margin nearest neighbor classification,” in Advances in neural information processing systems, 2005, pp. 1473–1480.
  • [24] Christos Boutsidis and Efstratios Gallopoulos, “SVD Based Initialization: A Head Start For Nonnegative Matrix Factorization,” Pattern Recognition, vol. 41, no. 4, pp. 1350–1362, 2008.