Adaptive Regularization for Large-Scale Sparse Feature Embedding Models
Mang Li, Wei Lyu
摘要
The one-epoch overfitting problem has drawn widespread attention, especially in CTR and CVR estimation models in search, advertising, and recommendation domains. These models which rely heavily on large-scale sparse categorical features, often suffer a significant decline in performance when trained for multiple epochs. Although recent studies have proposed heuristic solutions, the fundamental cause of this phenomenon remains unclear. In this work, we present a theoretical explanation grounded in Rademacher complexity, supported by empirical experiments, to explain why overfitting occurs in models with large-scale sparse categorical features. Based on this analysis, we propose a regularization method that constrains the norm budget of embedding layers adaptively. Our approach not only prevents the severe performance degradation observed during multi-epoch training, but also improves model performance within a single epoch. This method has already been deployed in online production systems.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper5
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida 等NeurIPS 2022 · 被引用 24,707 次
- Sharpness-aware Minimization for Efficiently Improving GeneralizationPierre Foret, Ariel Kleiner, Hossein Mobahi, Behnam NeyshaburICLR 2021 · 被引用 1,861 次
- On the Embedding Collapse when Scaling up Recommendation ModelsXingzhuo Guo, Junwei Pan, Ximei Wang, Baixu Chen 等ICML 2024 · 被引用 55 次
- Kraken: memory-efficient continual learning for large-scale real-time recommendationsMinhui Xie, Kai Ren, Youyou Lu, Guangxu Yang 等SC 2020 · 被引用 38 次
- Scaling Transformers for Discriminative Recommendation via Generative PretrainingChunqi Wang, Bingchao Wu, Zheng Chen, Lei Shen 等KDD 2025 · 被引用 1 次
相关 Paper
- CowClip: Reducing CTR Prediction Model Training Time from 12 Hours to 10 Minutes on 1 GPUZangwei Zheng, Pengtai Xu, Xuan Zou, Da Tang 等AAAI 2023 · 被引用 9 次
- Generalizable Multi-Pass Training of Ads Recommendation Models with Foundation Model GuidanceYunzhe Qi, Qinghai Zhou, Boyang Liu, Can Cui 等KDD 2026
- Unified Embedding: Battle-Tested Feature Representations for Web-Scale ML SystemsBenjamin Coleman, Wang-Cheng Kang, Matthew Fahrbach, Ruoxi Wang 等NeurIPS 2023 · 被引用 30 次
- FM2: Field-matrixed Factorization Machines for Recommender SystemsYang Sun, Junwei Pan, Alex Zhang, Aaron FloresWWW 2021 · 被引用 98 次
- AdaEmbed: Adaptive Embedding for Large-Scale Recommendation ModelsFan Lai, Wei Zhang, Rui Liu, William Tsai 等OSDI 2023 · 被引用 23 次
