A Thorough Performance Benchmarking on Lightweight Embedding-based Recommender Systems
Not on the map. Why?
Abstract (arXiv)
Since the creation of the Web, recommender systems (RSs) have been an indispensable mechanism in information filtering. State-of-the-art RSs primarily depend on categorical features, which ecoded by embedding vectors, resulting in excessively large embedding tables. To prevent over-parameterized embedding tables from harming scalability, both academia and industry have seen increasing efforts in compressing RS embeddings. However, despite the prosperity of lightweight embedding-based RSs (LERSs), a wide diversity is seen in evaluation protocols, resulting in obstacles when relating LERS performance to real-world usability. Moreover, despite the common goal of lightweight embeddings, LERSs are evaluated with a single choice between the two main recommendation tasks -- collaborative filtering and content-based recommendation. This lack of discussions on cross-task transferability hinders the development of unified, more scalable solutions. Motivated by these issues, this study investigates various LERSs' performance, efficiency, and cross-task transferability via a thorough benchmarking process. Additionally, we propose an efficient embedding compression method using magnitude pruning, which is an easy-to-deploy yet highly competitive baseline that outperforms various complex LERSs. Our study reveals the distinct performance of LERSs across the two tasks, shedding light on their effectiveness and generalizability. To support edge-based recommendations, we tested all LERSs on a Raspberry Pi 4, where the efficiency bottleneck is exposed. Finally, we conclude this paper with critical summaries of LERS performance, model selection suggestions, and underexplored challenges around LERSs for future research. To encourage future research, we publish source codes and artifacts at hrefthis linkhttps://github.com/chenxing1999/recsys-benchmark.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 55 of its 77 references that are in the map. First posted on arXiv 25 Jun 2024.
Builds on
55 of 77 references.
References · 55
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,134
- 2017Neural Collaborative Filteringcited by 1,108
- 2017DeepFM: A Factorization-Machine based Neural Network for CTR Predictioncited by 673
- 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 666
- 2019Neural Graph Collaborative Filteringcited by 660
- 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 521
38 more
- 2016Ups and Downs: Modeling the Visual Evolution of Fashion Trends with One-Class Collaborative Filteringcited by 517
- 2020DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systemscited by 326
- 2019Are We Really Making Much Progress? A Worrying Analysis of Recent Neural Recommendation Approachescited by 181
- 2021Are Graph Augmentations Necessary? Simple Graph Contrastive Learning for Recommendationcited by 173
- 2020Soft Threshold Weight Reparameterization for Learnable Sparsitycited by 150
- 2018Billion-scale Commodity Embedding for E-commerce Recommendation in Alibabacited by 105
- 2019Secure Federated Matrix Factorizationcited by 102
- 2019Federated Collaborative Filtering for Privacy-Preserving Personalized Recommendation Systemcited by 79
- 2021Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Modelscited by 78
- 2015Modeling User Exposure in Recommendationcited by 69
- 2019Compositional Embeddings Using Complementary Partitions for Memory-Efficient Recommendation Systemscited by 68
- 2019On-Device Machine Learning: An Algorithms and Learning Theory Perspectivecited by 63
- 2021TT-Rec: Tensor Train Compression for Deep Learning Recommendation Modelscited by 58
- 2022BARS: Towards Open Benchmarking for Recommender Systemscited by 54
- 2020Open Benchmarking for Click-Through Rate Predictioncited by 53
- 2020AutoEmb: Automated Embedding Dimensionality Search in Streaming Recommendationscited by 47
- 2019On the Difficulty of Evaluating Baselines: A Study on Recommender Systemscited by 46
- 2021Learnable Embedding Sizes for Recommender Systemscited by 42
- 2019Neural Input Search for Large Scale Recommendation Modelscited by 39
- 2020Learning to Embed Categorical Features without Embedding Tables for Recommendationcited by 38
- 2022On-Device Next-Item Recommendation with Self-Supervised Knowledge Distillationcited by 21
- 2024On-Device Recommender Systems: A Comprehensive Surveycited by 20
- 2021Learning Elastic Embeddings for Customizing On-Device Recommenderscited by 19
- 2022DaisyRec 2.0: Benchmarking Recommendation for Rigorous Evaluationcited by 19
- 2022OptEmbed: Learning Optimal Embedding Table for Click-through Rate Predictioncited by 14
- 2023Unified Embedding: Battle-Tested Feature Representations for Web-Scale ML Systemscited by 14
- 2022Learning Compressed Embeddings for On-Device Inferencecited by 13
- 2022Single-shot Embedding Dimension Search in Recommender Systemcited by 13
- 2022Effective and Efficient Training for Sequential Recommendation using Recency Samplingcited by 13
- 2023Experimental Analysis of Large-scale Learnable Vector Storage Compressioncited by 12
- 2022Adaptive Low-Precision Training for Embeddings in Click-Through Rate Predictioncited by 11
- 2023Towards Mitigating Dimensional Collapse of Representations in Collaborative Filteringcited by 10
- 2023MP-Rec: Hardware-Software Co-Design to Enable Multi-Path Recommendationcited by 9
- 2023Model-Agnostic Decentralized Collaborative Learning for On-Device POI Recommendationcited by 8
- 2023Continuous Input Embedding Size Search For Recommender Systemscited by 7
- 2024Dynamic Sparse Learning: A Novel Paradigm for Efficient Recommendationcited by 7
- 2022MultiBiSage: A Web-Scale Recommendation System Using Multiple Bipartite Graphs at Pinterestcited by 6
- 2022Nimble GNN Embedding with Tensor-Train Decompositioncited by 6
50 most cited of 55 All 55 (JSON)
Built on by
Citing papers per year: 2024: 2; 2025: 1; 2026: 1.
4 papers.
- 2024On-device Content-based Recommendation with Single-shot Embedding Pruning: A Cooperative Game Perspectivecited by 2
- 2024Beyond KAN: Introducing KarSein for Adaptive High-Order Feature Interaction Modeling in CTR Predictioncited by 1
- 2025Relational Database Distillation: From Structured Tables to Condensed Graph Datacited by 1
- 2026Efficient Content-based Recommendation Model Training via Noise-aware Coreset Selectioncited by 0
- 2026Efficient Content-based Recommendation Model Training via Noise-aware Coreset Selection15 Jan 2026
- 2025Relational Database Distillation: From Structured Tables to Condensed Graph Data8 Oct 2025
- 2024On-device Content-based Recommendation with Single-shot Embedding Pruning: A Cooperative Game Perspective20 Nov 2024
- 2024Beyond KAN: Introducing KarSein for Adaptive High-Order Feature Interaction Modeling in CTR Prediction16 Aug 2024