Encoding temporal and structural information in machine learning models for recommendation

Tiphaine Viard (1) , Raphaël Fournier-S'Niehotta (2)

1 RIKEN AIP - RIKEN Center for Advanced Intelligence Project [Tokyo]
2 CEDRIC - VERTIGO - CEDRIC. Données complexes, apprentissage et représentations

Recommender systems focus on the task of selecting relevant items for a user, according to their taste: movies to watch, books to read, etc. In this paper, we provide a proof-of-concept that graphs and link streams can be relevant for the recommendation task. We design simple features modelling the structure and its evolution with time of recommender systems in order to validate this claim. We learn these features alongside classical content-based features, using a gradient boosting machine model (XgBoost) and perform rating prediction on a movie rating dataset, MovieLens20M, and a book rating dataset, from GoodReads. We obtain comparable performance to the state-of-the-art, and some interesting leads in terms of explicability.

Domaines: Machine Learning [stat.ML]; Réseaux sociaux et d'information [cs.SI]; Recherche d'information [cs.IR]

Tiphaine Viard, Raphaël Fournier-S'Niehotta. Encoding temporal and structural information in machine learning models for recommendation. LEG @ ECML-PKDD 2019, Sep 2019, Würzburg, Germany. ⟨hal-02444211⟩