ShuffleGate: Scalable Feature Optimization for Recommender Systems via Batch-wise Sensitivity Learning
Not on the map. Why?
Abstract (arXiv)
Feature optimization -- specifically Feature Selection (FS) and Dimension Selection (DS) -- is critical for the efficiency and generalization of large-scale recommender systems. While conceptually related, these tasks are typically tackled with isolated solutions that often suffer from ambiguous importance scores or prohibitive computational costs. In this paper, we propose ShuffleGate, a unified and interpretable mechanism that estimates component importance by measuring the model's sensitivity to information loss. Unlike conventional gating that learns relative weights, ShuffleGate introduces a batch-wise shuffling strategy to effectively "erase" information in an end-to-end differentiable manner. This paradigm shift yields naturally polarized importance distributions, bridging the long-standing "search-retrain gap" and distinguishing essential signals from noise without complex threshold tuning. Extensive experiments across four benchmarks validate that ShuffleGate consistently outperforms state-of-the-art methods in both Feature and Dimension Selection tasks. It achieves a 15× speedup over permutation baselines and demonstrates extreme scalability by processing 270M parameters in just 700 seconds. Finally, in a top-tier industrial deployment, it compressed input dimensions by 10×, yielding a 91% increase in training throughput while serving billions of daily requests without performance degradation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 16 of its 35 references that are in the map. First posted on arXiv 12 Mar 2025.
Builds on
16 of 35 references.
References · 16
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2018DARTS: Differentiable Architecture Searchcited by 1,798
- 2016Wide & Deep Learning for Recommender Systemscited by 747
- 2017Deep & Cross Network for Ad Click Predictionscited by 393
- 2018All Models are Wrong, but Many are Useful: Learning a Variable's Importance by Studying an Entire Class of Prediction Models Simultaneouslycited by 373
- 2016Learning Hidden Unit Contributions for Unsupervised Acoustic Model Adaptationcited by 40
- 2020Learning to Embed Categorical Features without Embedding Tables for Recommendationcited by 38
- 2020Model Size Reduction Using Frequency Based Double Hashing for Recommender Systemscited by 33
- 2022AutoField: Automating Feature Selection in Deep Recommender Systemscited by 30
- 2019Post-Training 4-bit Quantization on Embedding Tablescited by 22
- 2021Deep Clustering based Fair Outlier Detectioncited by 19
4 more
- 2020Mixed-Precision Embedding Using a Cachecited by 18
- 2022Single-shot Embedding Dimension Search in Recommender Systemcited by 13
- 2022Adaptive Low-Precision Training for Embeddings in Click-Through Rate Predictioncited by 11
- 2023SHARK: A Lightweight Model Compression Approach for Large-scale Recommender Systemscited by 6
Built on by
Not cited yet.