Command Palette
Search for a command to run...
Seoyoung Hong Jeongwhan Choi Yeon-Chang Lee Srijan Kumar Noseong Park

Abstract
Collaborative filtering (CF) methods for recommendation systems have been extensively researched, ranging from matrix factorization and autoencoder-based to graph filtering-based methods. Recently, lightweight methods that require almost no training have been recently proposed to reduce overall computation. However, existing methods still have room to improve the trade-offs among accuracy, efficiency, and robustness. In particular, there are no well-designed closed-form studies for \emph{balanced} CF in terms of the aforementioned trade-offs. In this paper, we design SVD-AE, a simple yet effective singular vector decomposition (SVD)-based linear autoencoder, whose closed-form solution can be defined based on SVD for CF. SVD-AE does not require iterative training processes as its closed-form solution can be calculated at once. Furthermore, given the noisy nature of the rating matrix, we explore the robustness against such noisy interactions of existing CF methods and our SVD-AE. As a result, we demonstrate that our simple design choice based on truncated SVD can be used to strengthen the noise robustness of the recommendation while improving efficiency. Code is available at https://github.com/seoyoungh/svd-ae.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| collaborative-filtering-on-movielens-10m | SVD-AE | HR@10: 0.3676 HR@100: 0.648 PSP@10: 0.0493 nDCG@10: 0.3775 nDCG@100: 0.4697 |
| collaborative-filtering-on-movielens-1m | SVD-AE | HR@10: 0.3179 HR@100: 0.5933 PSP@10: 0.0322 nDCG@10: 0.3355 nDCG@100: 0.4257 |
| recommendation-systems-on-gowalla | SVD-AE | HR@10: 0.144 HR@100: 0.3734 PSP@10: 0.248 nDCG@10: 0.1394 nDCG@100: 0.2115 |
| recommendation-systems-on-yelp2018 | SVD-AE | HR@10: 0.049 HR@100: 0.1979 PSP@10: 45 nDCG@10: 0.0474 nDCG@100: 0.1022 |
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.