Adaptive Low-Precision Training for Embeddings in Click-Through Rate Prediction
Not on the map. Why?
Abstract (arXiv)
Embedding tables are usually huge in click-through rate (CTR) prediction models. To train and deploy the CTR models efficiently and economically, it is necessary to compress their embedding tables at the training stage. To this end, we formulate a novel quantization training paradigm to compress the embeddings from the training stage, termed low-precision training (LPT). Also, we provide theoretical analysis on its convergence. The results show that stochastic weight quantization has a faster convergence rate and a smaller convergence error than deterministic weight quantization in LPT. Further, to reduce the accuracy degradation, we propose adaptive low-precision training (ALPT) that learns the step size (i.e., the quantization resolution) through gradient descent. Experiments on two real-world datasets confirm our analysis and show that ALPT can significantly improve the prediction accuracy, especially at extremely low bit widths. For the first time in CTR models, we successfully train 8-bit embeddings without sacrificing prediction accuracy. The code of ALPT is publicly available.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 30 of its 39 references that are in the map. First posted on arXiv 12 Dec 2022.
Builds on
30 of 39 references.
References · 30
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 2016XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networkscited by 829
- 2016Wide & Deep Learning for Recommender Systemscited by 745
- 2017DeepFM: A Factorization-Machine based Neural Network for CTR Predictioncited by 673
- 2019Learned Step Size Quantizationcited by 604
- 2020Up or Down? Adaptive Rounding for Post-Training Quantizationcited by 600
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 571
- 2017Deep Interest Network for Click-Through Rate Predictioncited by 541
- 2018PACT: Parameterized Clipping Activation for Quantized Neural Networkscited by 456
- 2017Deep & Cross Network for Ad Click Predictionscited by 392
- 2017Training Quantized Nets: A Deeper Understandingcited by 74
18 more
- 2019Compositional Embeddings Using Complementary Partitions for Memory-Efficient Recommendation Systemscited by 68
- 2021TT-Rec: Tensor Train Compression for Deep Learning Recommendation Modelscited by 58
- 2016Loss-aware Binarization of Deep Networkscited by 56
- 2020Open Benchmarking for Click-Through Rate Predictioncited by 53
- 2015Neural Networks with Few Multiplicationscited by 46
- 2021Learnable Embedding Sizes for Recommender Systemscited by 42
- 2019Neural Input Search for Large Scale Recommendation Modelscited by 39
- 2020An Embedding Learning Framework for Numerical Features in CTR Predictioncited by 37
- 2020Model Size Reduction Using Frequency Based Double Hashing for Recommender Systemscited by 33
- 2018Loss-aware Weight Quantization of Deep Networkscited by 31
- 2020Detecting Beneficial Feature Interactions for Recommender Systemscited by 22
- 2019Post-Training 4-bit Quantization on Embedding Tablescited by 21
- 2022On-Device Next-Item Recommendation with Self-Supervised Knowledge Distillationcited by 21
- 2021Learning Elastic Embeddings for Customizing On-Device Recommenderscited by 19
- 2020Mixed-Precision Embedding Using a Cachecited by 18
- 2020Memory-efficient Embedding for Recommendationscited by 15
- 2022OptEmbed: Learning Optimal Embedding Table for Click-through Rate Predictioncited by 14
- 2021ScaleFreeCTR: MixCache-based Distributed Training System for CTR Models with Huge Embedding Tablecited by 12
Built on by
Citing papers per year: 2023: 1; 2024: 2; 2025: 8.
11 papers.
- 2025Beyond Zero Initialization: Investigating the Impact of Non-Zero Initialization on LoRA Fine-Tuning Dynamicscited by 16
- 2023CAFE: Towards Compact, Adaptive, and Fast Embedding for Large-scale Recommendation Modelscited by 7
- 2025Semantic Retrieval Augmented Contrastive Learning for Sequential Recommendationcited by 6
- 2025The Panaceas for Improving Low-Rank Decomposition in Communication-Efficient Federated Learningcited by 5
- 2025Beyond Higher Rank: Token-wise Input-Output Projections for Efficient Low-Rank Adaptationcited by 5
- 2024A Thorough Performance Benchmarking on Lightweight Embedding-based Recommender Systemscited by 4
- 2025The Future is Sparse: Embedding Compression for Scalable Retrieval in Recommender Systemscited by 4
- 2025FedBiF: Communication-Efficient Federated Learning via Bits Freezingcited by 1
- 2024ProactivePIM: Accelerating Weight-Sharing Embedding Layer with PIM for Scalable Recommendation Systemcited by 0
- 2025ShuffleGate: Scalable Feature Optimization for Recommender Systems via Batch-wise Sensitivity Learningcited by 0
- 2025DimGrow: Memory-Efficient Field-level Embedding Dimension Searchcited by 0
- 2025Beyond Higher Rank: Token-wise Input-Output Projections for Efficient Low-Rank Adaptation27 Oct 2025
- 2025FedBiF: Communication-Efficient Federated Learning via Bits Freezing12 Sep 2025
- 2025Beyond Zero Initialization: Investigating the Impact of Non-Zero Initialization on LoRA Fine-Tuning Dynamics29 May 2025
- 2025The Panaceas for Improving Low-Rank Decomposition in Communication-Efficient Federated Learning29 May 2025
- 2025DimGrow: Memory-Efficient Field-level Embedding Dimension Search19 May 2025
- 2025The Future is Sparse: Embedding Compression for Scalable Retrieval in Recommender Systems16 May 2025
- 2025ShuffleGate: Scalable Feature Optimization for Recommender Systems via Batch-wise Sensitivity Learning12 Mar 2025
- 2025Semantic Retrieval Augmented Contrastive Learning for Sequential Recommendation6 Mar 2025
- 2024A Thorough Performance Benchmarking on Lightweight Embedding-based Recommender Systems25 Jun 2024
- 2024ProactivePIM: Accelerating Weight-Sharing Embedding Layer with PIM for Scalable Recommendation System6 Feb 2024
- 2023CAFE: Towards Compact, Adaptive, and Fast Embedding for Large-scale Recommendation Models6 Dec 2023