HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Collaborative Translational Metric Learning

Chanyoung Park; Donghyun Kim; Xing Xie; Hwanjo Yu

Collaborative Translational Metric Learning

Abstract

Recently, matrix factorization-based recommendation methods have been criticized for the problem raised by the triangle inequality violation. Although several metric learning-based approaches have been proposed to overcome this issue, existing approaches typically project each user to a single point in the metric space, and thus do not suffice for properly modeling the intensity and the heterogeneity of user-item relationships in implicit feedback. In this paper, we propose TransCF to discover such latent user-item relationships embodied in implicit user-item interactions. Inspired by the translation mechanism popularized by knowledge graph embedding, we construct user-item specific translation vectors by employing the neighborhood information of users and items, and translate each user toward items according to the user's relationships with the items. Our proposed method outperforms several state-of-the-art methods for top-N recommendation on seven real-world data by up to 17% in terms of hit ratio. We also conduct extensive qualitative evaluations on the translation vectors learned by our proposed method to ascertain the benefit of adopting the translation mechanism for implicit feedback-based recommendations.

Code Repositories

hkuds/recdiff
pytorch
Mentioned in GitHub
pcy1302/TransCF
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-flixsterTransCF
Hits@10: 0.7309
Hits@20: 0.8374
nDCG@10: 0.4986
nDCG@20: 0.5257
recommendation-systems-on-amazon-caTransCF
Hits@10: 0.3436
Hits@20: 0.4658
nDCG@10: 0.2019
nDCG@20: 0.2323
recommendation-systems-on-book-crossing-1TransCF
Hits@10: 0.3329
Hits@20: 0.4744
nDCG@10: 0.1865
nDCG@20: 0.2221
recommendation-systems-on-ciaoTransCF
Hits@10: 0.2292
Hits@20: 0.374
nDCG@10: 0.1167
nDCG@20: 0.1525
recommendation-systems-on-decliciousTransCF
Hits@10: 0.2586
Hits@20: 0.3786
nDCG@10: 0.1475
nDCG@20: 0.1781
recommendation-systems-on-pinterestTransCF
Hits@10: 0.5504
Hits@20: 0.8108
nDCG@10: 0.258
nDCG@20: 0.3242
recommendation-systems-on-tradesyTransCF
Hits@10: 0.3198
Hits@20: 0.4505
nDCG@10: 0.1767
nDCG@20: 0.2095

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