Once-for-All: Train One Network and Specialize it for Efficient Deployment
Not on the map. Why?
Abstract (arXiv)
We address the challenging problem of efficient inference across many devices and resource constraints, especially on edge devices. Conventional approaches either manually design or use neural architecture search (NAS) to find a specialized neural network and train it from scratch for each case, which is computationally prohibitive (causing CO2 emission as much as 5 cars' lifetime) thus unscalable. In this work, we propose to train a once-for-all (OFA) network that supports diverse architectural settings by decoupling training and search, to reduce the cost. We can quickly get a specialized sub-network by selecting from the OFA network without additional training. To efficiently train OFA networks, we also propose a novel progressive shrinking algorithm, a generalized pruning method that reduces the model size across many more dimensions than pruning (depth, width, kernel size, and resolution). It can obtain a surprisingly large number of sub-networks (> 1019) that can fit different hardware platforms and latency constraints while maintaining the same level of accuracy as training independently. On diverse edge devices, OFA consistently outperforms state-of-the-art (SOTA) NAS methods (up to 4.0% ImageNet top1 accuracy improvement over MobileNetV3, or same accuracy but 1.5x faster than MobileNetV3, 2.6x faster than EfficientNet w.r.t measured latency) while reducing many orders of magnitude GPU hours and CO2 emission. In particular, OFA achieves a new SOTA 80.0% ImageNet top-1 accuracy under the mobile setting (<600M MACs). OFA is the winning solution for the 3rd Low Power Computer Vision Challenge (LPCVC), DSP classification track and the 4th LPCVC, both classification track and detection track. Code and 50 pre-trained models (for many devices & many latency constraints) are released at https://github.com/mit-han-lab/once-for-all.
Key facts. Cited by 778 papers in the map (data through 7 Oct 2026). First posted on arXiv 26 Aug 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 184; 2022: 146; 2023: 145; 2024: 112; 2025: 104; 2026: 87.
778 papers.
Sort:
By area:
- 2022Inner Monologue: Embodied Reasoning through Planning with Language Modelscited by 1,057
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2021ImageNet-21K Pretraining for the Massescited by 607
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2022Matryoshka Representation Learningcited by 268
- 2023Q-Diffusion: Quantizing Diffusion Modelscited by 266
- 2021Pruning and Quantization for Deep Neural Network Acceleration: A Surveycited by 259
- 2021Sustainable AI: Environmental Implications, Challenges and Opportunitiescited by 242
- 2021Parametric Contrastive Learningcited by 217
- 2021AutoFormer: Searching Transformers for Visual Recognitioncited by 186
- 2025SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformercited by 165
- 2022Compute Trends Across Three Eras of Machine Learningcited by 161
38 more
- 2022FlexiViT: One Model for All Patch Sizescited by 151
- 2023Neural Architecture Search: Insights from 1000 Paperscited by 136
- 2023Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Modelscited by 131
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
- 2023MobileVLM : A Fast, Strong and Open Vision Language Assistant for Mobile Devicescited by 120
- 2021Learning Task-Oriented Communication for Edge Inference: An Information Bottleneck Approachcited by 105
- 2023Skeleton-of-Thought: Prompting LLMs for Efficient Parallel Generationcited by 103
- 2024MobileNetV4 -- Universal Models for the Mobile Ecosystemcited by 98
- 2021A Survey of Modern Deep Learning based Object Detection Modelscited by 97
- 2021QuantumNAS: Noise-Adaptive Search for Robust Quantum Circuitscited by 97
- 2021ResLT: Residual Learning for Long-tailed Recognitioncited by 96
- 2021Parameter Prediction for Unseen Deep Architecturescited by 89
- 2021MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learningcited by 87
- 2021Dynamic Slimmable Networkcited by 83
- 2021LightTrack: Finding Lightweight Neural Networks for Object Tracking via One-Shot Architecture Searchcited by 82
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2023MatFormer: Nested Transformer for Elastic Inferencecited by 79
- 2021Nonuniform-to-Uniform Quantization: Towards Accurate Quantization via Generalized Straight-Through Estimationcited by 78
- 2023Sensitivity-Aware Visual Parameter-Efficient Fine-Tuningcited by 74
- 2025RF-DETR: Neural Architecture Search for Real-Time Detection Transformerscited by 70
- 2024Scaling Diffusion Transformers to 16 Billion Parameterscited by 65
- 2021A Comprehensive Survey on Hardware-Aware Neural Architecture Searchcited by 60
- 2023SparseViT: Revisiting Activation Sparsity for Efficient High-Resolution Vision Transformercited by 60
- 2021EPMF: Efficient Perception-aware Multi-sensor Fusion for 3D Semantic Segmentationcited by 59
- 2023ZiCo: Zero-shot NAS via Inverse Coefficient of Variation on Gradientscited by 57
- 2021NAS-BERT: Task-Agnostic and Adaptive-Size BERT Compression with Neural Architecture Searchcited by 55
- 2021Anycost GANs for Interactive Image Synthesis and Editingcited by 54
- 2021A Survey on Green Deep Learningcited by 52
- 2023ZipLM: Inference-Aware Structured Pruning of Language Modelscited by 46
- 2022LightHuBERT: Lightweight and Configurable Speech Representation Learning with Once-for-All Hidden-Unit BERTcited by 45
- 2022Generalized Parametric Contrastive Learningcited by 45
- 2022GhostNetV2: Enhance Cheap Operation with Long-Range Attentioncited by 44
- 2025Empowering Edge Intelligence: A Comprehensive Survey on On-Device AI Modelscited by 44
- 2021Hessian-Aware Pruning and Optimal Neural Implantcited by 43
- 2022Membership Inference Attacks and Defenses in Neural Network Pruningcited by 42
- 2022Exploring Patch-wise Semantic Relation for Contrastive Learning in Image-to-Image Translation Taskscited by 42
- 2022EfficientViT: Multi-Scale Linear Attention for High-Resolution Dense Predictioncited by 42
- 2024Flextron: Many-in-One Flexible Large Language Modelcited by 42
50 most cited of 778 All 778 (JSON)
- 2026Evolve on the Host, Predict on the Edge: Deploying Online Neuroevolutionary Architecture Search for Cross-sectional Stock Return Prediction7 Oct 2026
- 2026LESS: Lightweight Evolutionary Supernet Search in Minutes1 Oct 2026
- 2026Right In-Place (RiP) Convolution: A Simple, General, and Near-Optimal Strategy for Memory-Efficient CNN Inference30 Sep 2026
- 2026Telescopic Language Models28 Sep 2026
- 2026EdgeCraft: Automated Model Crafting for Edge IoT28 Sep 2026
- 2026Does per-frame early exit pay? A compute-matched study of dynamic depth for on-device speech enhancement24 Sep 2026
- 2026Neural Spectral Capacity: Measuring and Designing Architectures from Network Specification Alone19 Sep 2026
- 2026Scout: Open-World Species Recognition on the Edge19 Sep 2026
- 2026CoRA-NAS: Coarse Ranking and Anchor-Residual Refinement for Neural Architecture Search10 Sep 2026
- 2026FANS: Federated Adaptive Network Search Learning for Heterogeneous Devices5 Sep 2026
- 2026Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference4 Sep 2026
- 2026Measurement-Driven Sub-Network Selection for On-Premise Retrieval-Augmented Factory Agents2 Sep 2026
38 more
- 2026H3DNAS: Hardware-Aware ONNX-Native 3D Point Cloud Model Compression2 Sep 2026
- 2026Network-Aware Forecasting on Wireless Access Points2 Sep 2026
- 2026Node-wise Feature Encoding for Neural Performance Prediction28 Aug 2026
- 2026Rank Portability Does Not Imply Feasibility Portability: Target-Specific Evaluation of Joint Hardware Constraints20 Aug 2026
- 2026A Generalized Optimization Engine (GOE) for Edge AI Inference Acceleration19 Aug 2026
- 2026LP-NAS: Linear Programming-based Neural Architecture Search14 Aug 2026
- 2026ENAS: An Efficient Hardware-Aware Neural Architecture Search Framework for TinyML on Resource-Constrained Microcontrollers10 Aug 2026
- 2026Matryoshka Language Model Suites10 Aug 2026
- 2026FEAST: Federated Shared-Space Training for Resource-Heterogeneous Clients10 Aug 2026
- 2026How Much Does Message Passing Matter? A Drop-In Study of GNN Layers for Neural Network Graph Regression29 Jul 2026
- 2026Mitigating Compiler Fusion-Induced Power Bursts in Mobile NPU Inference as the Battery Depletes17 Jul 2026
- 2026Adaptive Routing for Efficient Diffusion Transformer-Based PNI Prediction13 Jul 2026
- 2026MOSAIC: Adaptive Inter-layer Composition for Efficient Heterogeneous Vision-Language Models10 Jul 2026
- 2026Agentic Neural Architecture Search8 Jul 2026
- 2026Bilevel Optimization for Neural Architecture Search28 Jun 2026
- 2026The Inference-Compute Frontier and a Latency-Efficient Architecture for Limit Order Book Prediction24 Jun 2026
- 2026Efficient Network Inference via Hardware-Aware Architecture Search, Model Pruning & Quantization22 Jun 2026
- 2026Neural Architecture Distributions: A New Paradigm for Stochastic Segmentation19 Jun 2026
- 2026InTrain: Intrinsic Trainability for Zero-Cost Neural Architecture Search17 Jun 2026
- 2026PreLort: Prefix-Nested LoRA for Federated Fine-Tuning under Rank Heterogeneity14 Jun 2026
- 2026Running hardware-aware neural architecture search on embedded devices under 512MB of RAM12 Jun 2026
- 2026Edge-Constrained UAV Small-Object Detection with P2 Enhancement and Quantum-Inspired Lightweight Structure Search8 Jun 2026
- 2026Structure-Conditioned Actor-Critic Branches for Quality-Diversity Reinforcement Learning7 Jun 2026
- 2026HASA: Subnet Allocation for Compute-Constrained Model-Heterogeneous Federated Learning30 May 2026
- 2026DREAM-S: Speculative Decoding with Searchable Drafting and Target-Aware Refinement for Multimodal Generation30 May 2026
- 2026KLAS: Using Similarity to Stitch Neural Networks for Improved Accuracy-Efficiency Tradeoffs28 May 2026
- 2026JetViT: Efficient High-Resolution Vision Transformer with Post-Training Attention Search26 May 2026
- 2026Slimmable ConvNeXt: Width-Adaptive Inference for Efficient Multi-Device Deployment21 May 2026
- 2026AutoMCU: Feasibility-First MCU Neural Network Customization via LLM-based Multi-Agent Systems20 May 2026
- 2026Structuring Open-Ended NAS: Semi-Automated Design Knowledge Structuring with LLMs for Efficient Neural Architecture Search19 May 2026
- 2026SNAC-Pack 2.0: Scaled-Out Surrogate Neural Architecture Codesign15 May 2026
- 2026RAMS: Resource-Adaptive and Detection-Conditioned Model Switching for Embedded Edge Perception14 May 2026
- 2026Do Vision Transformers Need All-to-All Attention? Global Communication Through Elastic Learned Cores12 May 2026
- 2026TCP-SSM: Efficient Vision State Space Models with Token-Conditioned Poles12 May 2026
- 2026MatryoshkaLoRA: Learning Accurate Hierarchical Low-Rank Representations for LLM Fine-Tuning8 May 2026
- 2026TAS-LoRA: Transformer Architecture Search with Mixture-of-LoRA Experts8 May 2026
- 2026XiYOLO: Energy-Aware Object Detection via Iterative Architecture Search and Scaling7 May 2026
- 2026Budgeted Attention Allocation: Cost-Conditioned Compute Control for Efficient Transformers7 May 2026
50 newest of 778 All 778 (JSON)
- 2023Q-Diffusion: Quantizing Diffusion Modelscited by 266
- 2023Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Modelscited by 131
- 2024Scaling Diffusion Transformers to 16 Billion Parameterscited by 65
- 2023DiffusionNAG: Predictor-guided Neural Architecture Generation with Diffusion Modelscited by 15
- 2024Not All Prompts Are Made Equal: Prompt-based Pruning of Text-to-Image Diffusion Modelscited by 13
- 2024Mixture of Efficient Diffusion Experts Through Automatic Interval and Sub-Network Selectioncited by 11
- 2024Efficient Training of Large Vision Models via Advanced Automated Progressive Learningcited by 4
- 2026SnapGen++: Unleashing Diffusion Transformers for Efficient High-Fidelity Image Generation on Edge Devicescited by 3
- 2026One Model, Many Budgets: Elastic Latent Interfaces for Diffusion Transformerscited by 3
- 2025Your Image Generator Is Your New Private Datasetcited by 2
- 2026Adaptive Routing for Efficient Diffusion Transformer-Based PNI Predictioncited by 1
- 2026OFA-Diffusion Compression: Compressing Diffusion Model in One-Shot Mannercited by 0
- 2026Adaptive Routing for Efficient Diffusion Transformer-Based PNI Prediction13 Jul 2026
- 2026OFA-Diffusion Compression: Compressing Diffusion Model in One-Shot Manner14 Apr 2026
- 2026One Model, Many Budgets: Elastic Latent Interfaces for Diffusion Transformers12 Mar 2026
- 2026SnapGen++: Unleashing Diffusion Transformers for Efficient High-Fidelity Image Generation on Edge Devices13 Jan 2026
- 2025Your Image Generator Is Your New Private Dataset6 Apr 2025
- 2024Mixture of Efficient Diffusion Experts Through Automatic Interval and Sub-Network Selection23 Sep 2024
- 2024Efficient Training of Large Vision Models via Advanced Automated Progressive Learning6 Sep 2024
- 2024Scaling Diffusion Transformers to 16 Billion Parameters16 Jul 2024
- 2024Not All Prompts Are Made Equal: Prompt-based Pruning of Text-to-Image Diffusion Models17 Jun 2024
- 2023DiffusionNAG: Predictor-guided Neural Architecture Generation with Diffusion Models26 May 2023
- 2023Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Models25 Apr 2023
- 2023Q-Diffusion: Quantizing Diffusion Models8 Feb 2023
- 2021Elbert: Fast Albert with Confidence-Window Based Early Exitcited by 14
- 2023Accelerating Transducers through Adjacent Token Mergingcited by 14
- 2024QCQA: Quality and Capacity-aware grouped Query Attentioncited by 5
- 2024Tiny Models are the Computational Saver for Large Modelscited by 4
- 2024DAISY: Data Adaptive Self-Supervised Early Exit for Speech Representation Modelscited by 4
- 2023FIANCEE: Faster Inference of Adversarial Networks via Conditional Early Exitscited by 3
- 2025You Only Look Once at Anytime (AnytimeYOLO): Analysis and Optimization of Early-Exits for Object-Detectioncited by 3
- 2026Matryoshka Language Model Suitescited by 0
- 2026A Generalized Optimization Engine (GOE) for Edge AI Inference Accelerationcited by 0
- 2026Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inferencecited by 0
- 2026Does per-frame early exit pay? A compute-matched study of dynamic depth for on-device speech enhancementcited by 0
- 2026Does per-frame early exit pay? A compute-matched study of dynamic depth for on-device speech enhancement24 Sep 2026
- 2026Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference4 Sep 2026
- 2026A Generalized Optimization Engine (GOE) for Edge AI Inference Acceleration19 Aug 2026
- 2026Matryoshka Language Model Suites10 Aug 2026
- 2025You Only Look Once at Anytime (AnytimeYOLO): Analysis and Optimization of Early-Exits for Object-Detection21 Mar 2025
- 2024QCQA: Quality and Capacity-aware grouped Query Attention8 Jun 2024
- 2024DAISY: Data Adaptive Self-Supervised Early Exit for Speech Representation Models8 Jun 2024
- 2024Tiny Models are the Computational Saver for Large Models26 Mar 2024
- 2023Accelerating Transducers through Adjacent Token Merging28 Jun 2023
- 2023FIANCEE: Faster Inference of Adversarial Networks via Conditional Early Exits20 Apr 2023
- 2021Elbert: Fast Albert with Confidence-Window Based Early Exit1 Jul 2021
- 2023MobileVLM : A Fast, Strong and Open Vision Language Assistant for Mobile Devicescited by 120
- 2024Trust but Verify: Programmatic VLM Evaluation in the Wildcited by 5
- 2026DREAM-S: Speculative Decoding with Searchable Drafting and Target-Aware Refinement for Multimodal Generationcited by 1
- 2026VISion On Request: Enhanced VLLM efficiency with sparse, dynamically selected, vision-language interactionscited by 0
- 2026MOSAIC: Adaptive Inter-layer Composition for Efficient Heterogeneous Vision-Language Modelscited by 0
- 2026Scout: Open-World Species Recognition on the Edgecited by 0
- 2026Scout: Open-World Species Recognition on the Edge19 Sep 2026
- 2026MOSAIC: Adaptive Inter-layer Composition for Efficient Heterogeneous Vision-Language Models10 Jul 2026
- 2026DREAM-S: Speculative Decoding with Searchable Drafting and Target-Aware Refinement for Multimodal Generation30 May 2026
- 2026VISion On Request: Enhanced VLLM efficiency with sparse, dynamically selected, vision-language interactions24 Mar 2026
- 2024Trust but Verify: Programmatic VLM Evaluation in the Wild17 Oct 2024
- 2023MobileVLM : A Fast, Strong and Open Vision Language Assistant for Mobile Devices28 Dec 2023
- 2022Mod-Squad: Designing Mixture of Experts As Modular Multi-Task Learnerscited by 15
- 2022AutoMoE: Heterogeneous Mixture-of-Experts with Adaptive Computation for Efficient Neural Machine Translationcited by 9
- 2023Mixture-of-Supernets: Improving Weight-Sharing Supernet Training with Architecture-Routed Mixture-of-Expertscited by 7
- 2024Duo-LLM: A Framework for Studying Adaptive Computation in Large Language Modelscited by 6
- 2026MoSE: Mixture of Slimmable Experts for Efficient and Adaptive Language Modelscited by 3
- 2026DynaMoE: Dynamic Token-Level Expert Activation with Layer-Wise Adaptive Capacity for Mixture-of-Experts Neural Networkscited by 1
- 2026DynaMoE: Dynamic Token-Level Expert Activation with Layer-Wise Adaptive Capacity for Mixture-of-Experts Neural Networks2 Mar 2026
- 2026MoSE: Mixture of Slimmable Experts for Efficient and Adaptive Language Models5 Feb 2026
- 2024Duo-LLM: A Framework for Studying Adaptive Computation in Large Language Models1 Oct 2024
- 2023Mixture-of-Supernets: Improving Weight-Sharing Supernet Training with Architecture-Routed Mixture-of-Experts8 Jun 2023
- 2022Mod-Squad: Designing Mixture of Experts As Modular Multi-Task Learners15 Dec 2022
- 2022AutoMoE: Heterogeneous Mixture-of-Experts with Adaptive Computation for Efficient Neural Machine Translation14 Oct 2022
- 2025SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformercited by 165
- 2023Skeleton-of-Thought: Prompting LLMs for Efficient Parallel Generationcited by 103
- 2024Dense Training, Sparse Inference: Rethinking Training of Mixture-of-Experts Language Modelscited by 41
- 2025SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformer30 Jan 2025
- 2024Dense Training, Sparse Inference: Rethinking Training of Mixture-of-Experts Language Models8 Apr 2024
- 2023Skeleton-of-Thought: Prompting LLMs for Efficient Parallel Generation28 Jul 2023
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2021ImageNet-21K Pretraining for the Massescited by 607
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2022Matryoshka Representation Learningcited by 268
- 2021Pruning and Quantization for Deep Neural Network Acceleration: A Surveycited by 259
- 2021Sustainable AI: Environmental Implications, Challenges and Opportunitiescited by 242
- 2021Parametric Contrastive Learningcited by 217
- 2021AutoFormer: Searching Transformers for Visual Recognitioncited by 186
- 2022Compute Trends Across Three Eras of Machine Learningcited by 161
- 2022FlexiViT: One Model for All Patch Sizescited by 151
- 2023Neural Architecture Search: Insights from 1000 Paperscited by 136
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
12 most cited of 733 All 733 (JSON)
- 2026Evolve on the Host, Predict on the Edge: Deploying Online Neuroevolutionary Architecture Search for Cross-sectional Stock Return Prediction7 Oct 2026
- 2026LESS: Lightweight Evolutionary Supernet Search in Minutes1 Oct 2026
- 2026Right In-Place (RiP) Convolution: A Simple, General, and Near-Optimal Strategy for Memory-Efficient CNN Inference30 Sep 2026
- 2026Telescopic Language Models28 Sep 2026
- 2026EdgeCraft: Automated Model Crafting for Edge IoT28 Sep 2026
- 2026Neural Spectral Capacity: Measuring and Designing Architectures from Network Specification Alone19 Sep 2026
- 2026CoRA-NAS: Coarse Ranking and Anchor-Residual Refinement for Neural Architecture Search10 Sep 2026
- 2026FANS: Federated Adaptive Network Search Learning for Heterogeneous Devices5 Sep 2026
- 2026H3DNAS: Hardware-Aware ONNX-Native 3D Point Cloud Model Compression2 Sep 2026
- 2026Network-Aware Forecasting on Wireless Access Points2 Sep 2026
- 2026Node-wise Feature Encoding for Neural Performance Prediction28 Aug 2026
- 2026Rank Portability Does Not Imply Feasibility Portability: Target-Specific Evaluation of Joint Hardware Constraints20 Aug 2026
12 newest of 733 All 733 (JSON)