Command Palette
Search for a command to run...
Chanyoung Park; Donghyun Kim; Xing Xie; Hwanjo Yu

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| collaborative-filtering-on-flixster | TransCF | Hits@10: 0.7309 Hits@20: 0.8374 nDCG@10: 0.4986 nDCG@20: 0.5257 |
| recommendation-systems-on-amazon-ca | TransCF | Hits@10: 0.3436 Hits@20: 0.4658 nDCG@10: 0.2019 nDCG@20: 0.2323 |
| recommendation-systems-on-book-crossing-1 | TransCF | Hits@10: 0.3329 Hits@20: 0.4744 nDCG@10: 0.1865 nDCG@20: 0.2221 |
| recommendation-systems-on-ciao | TransCF | Hits@10: 0.2292 Hits@20: 0.374 nDCG@10: 0.1167 nDCG@20: 0.1525 |
| recommendation-systems-on-declicious | TransCF | Hits@10: 0.2586 Hits@20: 0.3786 nDCG@10: 0.1475 nDCG@20: 0.1781 |
| recommendation-systems-on-pinterest | TransCF | Hits@10: 0.5504 Hits@20: 0.8108 nDCG@10: 0.258 nDCG@20: 0.3242 |
| recommendation-systems-on-tradesy | TransCF | 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.