HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Collaborative Similarity Embedding for Recommender Systems

Chih-Ming Chen; Chuan-Ju Wang; Ming-Feng Tsai; Yi-Hsuan Yang

Collaborative Similarity Embedding for Recommender Systems

Abstract

We present collaborative similarity embedding (CSE), a unified framework that exploits comprehensive collaborative relations available in a user-item bipartite graph for representation learning and recommendation. In the proposed framework, we differentiate two types of proximity relations: direct proximity and k-th order neighborhood proximity. While learning from the former exploits direct user-item associations observable from the graph, learning from the latter makes use of implicit associations such as user-user similarities and item-item similarities, which can provide valuable information especially when the graph is sparse. Moreover, for improving scalability and flexibility, we propose a sampling technique that is specifically designed to capture the two types of proximity relations. Extensive experiments on eight benchmark datasets show that CSE yields significantly better performance than state-of-the-art recommendation methods.

Code Repositories

bdnf/SBX-Recommendation-Engine
pytorch
Mentioned in GitHub
cnclabs/smore
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-netflixRATE-CSE
Recall@10: 0.2014
mAP@10: 0.1039
recommendation-systems-on-citeulikeRATE-CSE
Recall@10: 0.2362
mAP@10: 0.1452
recommendation-systems-on-echonestRANK-CSE
Recall@10: 0.1358
mAP@10: 0.0679
recommendation-systems-on-epinions-extendRANK-CSE
Recall@10: 0.1767
mAP@10: 0.0921
recommendation-systems-on-frappeRATE-CSE
Recall@10: 33.47
mAP@10: 0.2047
recommendation-systems-on-lastfm-360kRANK-CSE
Recall@10: 0.1762
mAP@10: 0.097
recommendation-systems-on-movielens-latestRATE-CSE
Recall@10: 0.3225
mAP@10: 0.199

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
Collaborative Similarity Embedding for Recommender Systems | Papers | HyperAI