HyperML: A Boosting Metric Learning Approach in Hyperbolic Space for Recommender Systems

5 Sep 2018  ·  Lucas Vinh Tran, Yi Tay, Shuai Zhang, Gao Cong, Xiao-Li Li ·

This paper investigates the notion of learning user and item representations in non-Euclidean space. Specifically, we study the connection between metric learning in hyperbolic space and collaborative filtering by exploring Mobius gyrovector spaces where the formalism of the spaces could be utilized to generalize the most common Euclidean vector operations. Overall, this work aims to bridge the gap between Euclidean and hyperbolic geometry in recommender systems through metric learning approach. We propose HyperML (Hyperbolic Metric Learning), a conceptually simple but highly effective model for boosting the performance. Via a series of extensive experiments, we show that our proposed HyperML not only outperforms their Euclidean counterparts, but also achieves state-of-the-art performance on multiple benchmark datasets, demonstrating the effectiveness of personalized recommendation in hyperbolic geometry.

PDF Abstract

Datasets


Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Recommendation Systems MovieLens 1M HyperML HR@10 0.7563 # 4
nDCG@10 0.5620 # 3
Recommendation Systems MovieLens 20M HyperML HR@10 0.8736 # 1
nDCG@10 0.6404 # 1

Methods


No methods listed for this paper. Add relevant methods here