HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SVD-AE: Simple Autoencoders for Collaborative Filtering

Seoyoung Hong Jeongwhan Choi Yeon-Chang Lee Srijan Kumar Noseong Park

SVD-AE: Simple Autoencoders for Collaborative Filtering

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

seoyoungh/svd-ae
jax
Mentioned in GitHub
jeongwhanchoi/svd-ae
Official
jax
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-movielens-10mSVD-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-1mSVD-AE
HR@10: 0.3179
HR@100: 0.5933
PSP@10: 0.0322
nDCG@10: 0.3355
nDCG@100: 0.4257
recommendation-systems-on-gowallaSVD-AE
HR@10: 0.144
HR@100: 0.3734
PSP@10: 0.248
nDCG@10: 0.1394
nDCG@100: 0.2115
recommendation-systems-on-yelp2018SVD-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.

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
SVD-AE: Simple Autoencoders for Collaborative Filtering | Papers | HyperAI