Interpretable Recommender System With Heterogeneous Information: A Geometric Deep Learning Perspective

20 Sep 2020  Â·  Yan Leng, Rodrigo Ruiz, Xiaowen Dong, Alex Pentland ·

Recommender systems (RS) are ubiquitous in the digital space. This paper develops a deep learning-based approach to address three practical challenges in RS: complex structures of high-dimensional data, noise in relational information, and the black-box nature of machine learning algorithms. Our method—Multi-Graph Graph Attention Network (MG-GAT)—learns latent user and business representations by aggregating a diverse set of information from neighbors of each user (business) on a neighbor importance graph. MG-GAT out-performs state-of-the-art deep learning models in the recommendation task using two large-scale datasets collected from Yelp and four other standard datasets in RS. The improved performance highlights MG-GAT’s advantage in incorporating multi-modal features in a principled manner. The features importance, neighbor importance graph and latent representations reveal business insights on predictive features and explainable characteristics of business and users. Moreover, the learned neighbor importance graph can be used in a variety of management applications, such as targeting customers, promoting new businesses, and designing information acquisition strategies. Our paper presents a quintessential big data application of deep learning models in management while providing interpretability essential for real-world decision-making.

PDF Abstract

Datasets


Results from the Paper


 Ranked #1 on Recommendation Systems on YahooMusic Monti (using extra training data)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Benchmark
Recommendation Systems Douban Monti MG-GAT RMSE 0.727 # 3
Recommendation Systems Flixster Monti MG-GAT RMSE 0.876 # 2
Recommendation Systems MovieLens 100K MG-GAT RMSE (u1 Splits) 0.890 # 3
Recommendation Systems YahooMusic Monti MG-GAT RMSE 18.9 # 1

Methods