Chih-Ming Chen, Chuan-Ju Wang, Ming-Feng Tsai, Yi-Hsuan Yang
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.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Recommendation Systems | CiteULike | Recall@10 | 0.2362 | RATE-CSE |
| Recommendation Systems | CiteULike | mAP@10 | 0.1452 | RATE-CSE |
| Recommendation Systems | MovieLens-Latest | Recall@10 | 0.3225 | RATE-CSE |
| Recommendation Systems | MovieLens-Latest | mAP@10 | 0.199 | RATE-CSE |
| Recommendation Systems | Netflix | Recall@10 | 0.2014 | RATE-CSE |
| Recommendation Systems | Netflix | mAP@10 | 0.1039 | RATE-CSE |
| Recommendation Systems | Frappe | Recall@10 | 33.47 | RATE-CSE |
| Recommendation Systems | Frappe | mAP@10 | 0.2047 | RATE-CSE |
| Recommendation Systems | Last.FM-360k | Recall@10 | 0.1762 | RANK-CSE |
| Recommendation Systems | Last.FM-360k | mAP@10 | 0.097 | RANK-CSE |
| Recommendation Systems | Echonest | Recall@10 | 0.1358 | RANK-CSE |
| Recommendation Systems | Echonest | mAP@10 | 0.0679 | RANK-CSE |
| Recommendation Systems | Epinions-Extend | Recall@10 | 0.1767 | RANK-CSE |
| Recommendation Systems | Epinions-Extend | mAP@10 | 0.0921 | RANK-CSE |