Command Palette
Search for a command to run...
Scalable Approximate NonSymmetric Autoencoder for Collaborative Filtering
{Miroslav Tůma Ladislav Peška Antonín Hoskovec Radek Bartyzal Martin Spišák}
Abstract
In the field of recommender systems, shallow autoencoders have recently gained significant attention. One of the most highly acclaimed shallow autoencoders is EASE, favored for its competitive recommendation accuracy and simultaneous simplicity. However, the poor scalability of EASE (both in time and especially in memory) severely restricts its use in production environments with vast item sets. In this paper, we propose a hyperefficient factorization technique for sparse approximate inversion of the data-Gram matrix used in EASE. The resulting autoencoder, SANSA, is an end-to-end sparse solution with prescribable density and almost arbitrarily low memory requirements — even for training. As such, SANSA allows us to effortlessly scale the concept of EASE to millions of items and beyond.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| collaborative-filtering-on-amazon-book | SANSA | NDCG@20: 0.0637 Recall@20: 0.0768 |
| collaborative-filtering-on-million-song | SANSA | Recall@20: 0.332 Recall@50: 0.427 nDCG@100: 0.388 |
| recommendation-systems-on-amazon-book | SANSA | Recall@20: 0.0768 nDCG@20: 0.0637 |
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.