Fine-grained Embedding Dimension Optimization During Training For Recommender Systems

Abstract

Huge embedding tables in modern deep learning recommender models (DLRM) require prohibitively large memory during training and inference. This paper proposes FIITED, a system to automatically reduce the memory footprint via FIne-grained In-Training Embedding Dimension pruning. By leveraging the key insight that embedding vectors are not equally important, FIITED adaptively adjusts the dimension of each individual embedding vector during model training, assigning larger dimensions to more important embeddings while adapting to dynamic changes in data. We prioritize embedding dimensions with higher frequencies and gradients as more important. To enable efficient pruning of embeddings and their dimensions during model training, we propose an embedding storage system based on virtually-hashed physically-indexed hash tables. Experiments on two industry models and months of realistic datasets show that FIITED can reduce DLRM embedding size by more than 65% while preserving model quality, out

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keyluo2024fine

Related papers

• CAFE: Towards Compact, Adaptive, And Fast Embedding For Large-scale Recommendation Models (2023)8.09
• Mem-rec: Memory Efficient Recommendation System Using Alternative Representation (2023)0.00
• Mixed-precision Embeddings For Large-scale Recommendation Models (2024)0.00
• Autoemb: Automated Embedding Dimensionality Search In Streaming Recommendations (2020)12.61
• Plug-and-play Parameter-efficient Tuning Of Embeddings For Federated Recommendation (2025)0.95
• Learning Compact Compositional Embeddings Via Regularized Pruning For Recommendation (2023)8.36
• Learning To Collide: Recommendation System Model Compression With Learned Hash Functions (2022)0.00
• Learning Compressed Embeddings For On-device Inference (2022)0.00