mlsanityData to 7 Oct 2026

A Thorough Performance Benchmarking on Lightweight Embedding-based Recommender Systems

arXiv 2406.17335 ↗First posted 25 Jun 2024cs.IR cs.LGcited by 4

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

  1. 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
  2. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
  3. 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
  4. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  5. 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
  6. 2018DARTS: Differentiable Architecture Searchcited by 1,782
  7. 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,134
  8. 2017Neural Collaborative Filteringcited by 1,108
  9. 2017DeepFM: A Factorization-Machine based Neural Network for CTR Predictioncited by 673
  10. 2018ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardwarecited by 666
  11. 2019Neural Graph Collaborative Filteringcited by 660
  12. 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 521
38 more
  1. 2016Ups and Downs: Modeling the Visual Evolution of Fashion Trends with One-Class Collaborative Filteringcited by 517
  2. 2020DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systemscited by 326
  3. 2019Are We Really Making Much Progress? A Worrying Analysis of Recent Neural Recommendation Approachescited by 181
  4. 2021Are Graph Augmentations Necessary? Simple Graph Contrastive Learning for Recommendationcited by 173
  5. 2020Soft Threshold Weight Reparameterization for Learnable Sparsitycited by 150
  6. 2018Billion-scale Commodity Embedding for E-commerce Recommendation in Alibabacited by 105
  7. 2019Secure Federated Matrix Factorizationcited by 102
  8. 2019Federated Collaborative Filtering for Privacy-Preserving Personalized Recommendation Systemcited by 79
  9. 2021Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Modelscited by 78
  10. 2015Modeling User Exposure in Recommendationcited by 69
  11. 2019Compositional Embeddings Using Complementary Partitions for Memory-Efficient Recommendation Systemscited by 68
  12. 2019On-Device Machine Learning: An Algorithms and Learning Theory Perspectivecited by 63
  13. 2021TT-Rec: Tensor Train Compression for Deep Learning Recommendation Modelscited by 58
  14. 2022BARS: Towards Open Benchmarking for Recommender Systemscited by 54
  15. 2020Open Benchmarking for Click-Through Rate Predictioncited by 53
  16. 2020AutoEmb: Automated Embedding Dimensionality Search in Streaming Recommendationscited by 47
  17. 2019On the Difficulty of Evaluating Baselines: A Study on Recommender Systemscited by 46
  18. 2021Learnable Embedding Sizes for Recommender Systemscited by 42
  19. 2019Neural Input Search for Large Scale Recommendation Modelscited by 39
  20. 2020Learning to Embed Categorical Features without Embedding Tables for Recommendationcited by 38
  21. 2022On-Device Next-Item Recommendation with Self-Supervised Knowledge Distillationcited by 21
  22. 2024On-Device Recommender Systems: A Comprehensive Surveycited by 20
  23. 2021Learning Elastic Embeddings for Customizing On-Device Recommenderscited by 19
  24. 2022DaisyRec 2.0: Benchmarking Recommendation for Rigorous Evaluationcited by 19
  25. 2022OptEmbed: Learning Optimal Embedding Table for Click-through Rate Predictioncited by 14
  26. 2023Unified Embedding: Battle-Tested Feature Representations for Web-Scale ML Systemscited by 14
  27. 2022Learning Compressed Embeddings for On-Device Inferencecited by 13
  28. 2022Single-shot Embedding Dimension Search in Recommender Systemcited by 13
  29. 2022Effective and Efficient Training for Sequential Recommendation using Recency Samplingcited by 13
  30. 2023Experimental Analysis of Large-scale Learnable Vector Storage Compressioncited by 12
  31. 2022Adaptive Low-Precision Training for Embeddings in Click-Through Rate Predictioncited by 11
  32. 2023Towards Mitigating Dimensional Collapse of Representations in Collaborative Filteringcited by 10
  33. 2023MP-Rec: Hardware-Software Co-Design to Enable Multi-Path Recommendationcited by 9
  34. 2023Model-Agnostic Decentralized Collaborative Learning for On-Device POI Recommendationcited by 8
  35. 2023Continuous Input Embedding Size Search For Recommender Systemscited by 7
  36. 2024Dynamic Sparse Learning: A Novel Paradigm for Efficient Recommendationcited by 7
  37. 2022MultiBiSage: A Web-Scale Recommendation System Using Multiple Bipartite Graphs at Pinterestcited by 6
  38. 2022Nimble GNN Embedding with Tensor-Train Decompositioncited by 6

50 most cited of 55 All 55 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 2; 2025: 1; 2026: 1.

4 papers.