Drs.NAS: Ultra-Efficient Neural Architecture Search for Recommendation Systems
Ruixuan Wang, Xun Jiao
Abstract
Deep learning-based recommendation systems (DRS) have become a dominant workload in hyperscale data centers. However, designing DRS architectures that balance high predictive performance with computational efficiency remains a major challenge due to ever-increasing model complexity and scale. Neural architecture search (NAS) has recently emerged as a promising automated design approach and is now adopted in production by major hyperscalers. Yet, existing NAS methods face two critical limitations: ( i ) prohibitive search costs—often requiring several GPU hours to days—which hinder rapid iteration, and ( ii ) the resulting architectures are typically computation- and memory-intensive, limiting practical deployment. In this paper, we propose Drs.NAS, an ultra-efficient NAS framework for DRS. ( i ) Ultra-efficient search: We propose a novel metric, superproxy , which enables NAS without the costly training and validation required by existing NAS methods. Compared to SOTA NAS search times of 5!∼!18 GPU-hours, Drs.NAS completes the search within two minutes on a commodity CPU . ( ii ) Ultra-efficient results: The models discovered by Drs.NAS drastically reduce resource demands—achieving on average 108.3× and 34.9× smaller model sizes, and 88.8× and 14.7× fewer FLOPs, compared to handcrafted and SOTA NAS results, respectively. Crucially, these gains come without sacrificing predictive quality : Drs.NAS delivers on par or even superior predictive performance, surpassing handcrafted and NAS baselines by 0.0123 and 0.0056 in average AUC across three representative benchmarks, respectively.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on14
- Once-for-All: Train One Network and Specialize it for Efficient DeploymentHan Cai, Chuang Gan, Tianzhe Wang, Zhekai Zhang et al.ICLR 2020 · 1,522 citations
- Pruning neural networks without any data by iteratively conserving synaptic flowHidenori Tanaka, Daniel Kunin, Daniel L. K. Yamins, Surya GanguliNeurIPS 2020 · 884 citations
- MCUNet: Tiny Deep Learning on IoT DevicesJi Lin, Wei-Ming Chen, Yujun Lin, John Cohn et al.NeurIPS 2020 · 827 citations
- DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank SystemsRuoxi Wang, Rakesh Shivanna, Derek Zhiyuan Cheng, Sagar Jain et al.WWW 2021 · 793 citations
- Neural Architecture Search without TrainingJoe Mellor, Jack Turner, Amos Storkey, Elliot J. CrowleyICML 2021 · 477 citations
Related papers
- ReNAS: Relativistic Evaluation of Neural Architecture SearchYixing Xu, Yunhe Wang, Kai Han, Yehui Tang et al.CVPR 2021
- Zero-Cost Proxies for Lightweight NASMohamed S. Abdelfattah, Abhinav Mehrotra, Lukasz Dudziak, Nicholas Donald LaneICLR 2021 · 65 citations
- Hyperscale Hardware Optimized Neural Architecture SearchSheng Li, Garrett Andersen, Tao Chen, Liqun Cheng et al.ASPLOS 2023 · 10 citations
- Fast and Practical Neural Architecture SearchJiequan Cui, Pengguang Chen, Ruiyu Li, Shu Liu et al.ICCV 2019 · 69 citations
- Extensible and Efficient Proxy for Neural Architecture SearchYuhong Li, Jiajie Li, Cong Hao, Pan Li et al.ICCV 2023 · 8 citations
