Incorporating Field-aware Deep Embedding Networks and Gradient Boosting Decision Trees for Music Recommendation

Bing Bai,Yushun Fan
2018-01-01
Abstract:On-line music streaming services, like KKBOX, is bringing great convenience for users to get access to all kinds of music, while the hierarchical data describing the songs and the random interests of users still make it a challenge for algorithms to make accurate recommendations, especially without enough historical data. In the WSDM KKBOX’s Music Recommendation Challenge, WSDM and KKBOX were challenging people to predict the chances of a user listening to a song repetitively, under the condition that many users/songs are cold-started. In this paper, we describe our solution to the task. Our solution comprised an ensemble of different models, including Field-aware Deep Embedding Networks and Gradient Boosting Decision Trees. We achieved an AUC score of 0.74787 on the Private Leaderboard, and finished the first place in the competition.
What problem does this paper attempt to address?