Command Palette
Search for a command to run...
Chih-Ming Chen; Chuan-Ju Wang; Ming-Feng Tsai; Yi-Hsuan Yang

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
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| collaborative-filtering-on-netflix | RATE-CSE | Recall@10: 0.2014 mAP@10: 0.1039 |
| recommendation-systems-on-citeulike | RATE-CSE | Recall@10: 0.2362 mAP@10: 0.1452 |
| recommendation-systems-on-echonest | RANK-CSE | Recall@10: 0.1358 mAP@10: 0.0679 |
| recommendation-systems-on-epinions-extend | RANK-CSE | Recall@10: 0.1767 mAP@10: 0.0921 |
| recommendation-systems-on-frappe | RATE-CSE | Recall@10: 33.47 mAP@10: 0.2047 |
| recommendation-systems-on-lastfm-360k | RANK-CSE | Recall@10: 0.1762 mAP@10: 0.097 |
| recommendation-systems-on-movielens-latest | RATE-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.