HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

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

Lucas Vinh Tran; Yi Tay; Shuai Zhang; Gao Cong; Xiaoli Li

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

Abstract

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.

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-movielens-1mHyperML
HR@10: 0.7563
nDCG@10: 0.5620
collaborative-filtering-on-movielens-20mHyperML
HR@10: 0.8736
nDCG@10: 0.6404

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
HyperML: A Boosting Metric Learning Approach in Hyperbolic Space for Recommender Systems | Papers | HyperAI