Pruning-as-Search: Efficient Neural Architecture Search via Channel Pruning and Structural Reparameterization
Not on the map. Why?
Abstract (arXiv)
Neural architecture search (NAS) and network pruning are widely studied efficient AI techniques, but not yet perfect. NAS performs exhaustive candidate architecture search, incurring tremendous search cost. Though (structured) pruning can simply shrink model dimension, it remains unclear how to decide the per-layer sparsity automatically and optimally. In this work, we revisit the problem of layer-width optimization and propose Pruning-as-Search (PaS), an end-to-end channel pruning method to search out desired sub-network automatically and efficiently. Specifically, we add a depth-wise binary convolution to learn pruning policies directly through gradient descent. By combining the structural reparameterization and PaS, we successfully searched out a new family of VGG-like and lightweight networks, which enable the flexibility of arbitrary width with respect to each layer instead of each stage. Experimental results show that our proposed architecture outperforms prior arts by around 1.0% top-1 accuracy under similar inference speed on ImageNet-1000 classification task. Furthermore, we demonstrate the effectiveness of our width search on complex tasks including instance segmentation and image translation. Code and models are released.
Key facts. Cited by 32 papers in the map (data through 14 Oct 2026). Builds on 27 of its 33 references that are in the map. First posted on arXiv 2 Jun 2022.
Builds on
27 of 33 references.
References · 27
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
- 2017Squeeze-and-Excitation Networkscited by 4,789
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,079
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
- 2018DARTS: Differentiable Architecture Searchcited by 1,798
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,499
- 2018MnasNet: Platform-Aware Neural Architecture Search for Mobilecited by 943
- 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 702
- 2017Learning Efficient Convolutional Networks through Network Slimmingcited by 676
15 more
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 522
- 2018FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable Neural Architecture Searchcited by 485
- 2019YOLACT: Real-time Instance Segmentationcited by 443
- 2017ThiNet: A Filter Level Pruning Method for Deep Neural Network Compressioncited by 442
- 2018AMC: AutoML for Model Compression and Acceleration on Mobile Devicescited by 378
- 2019MetaPruning: Meta Learning for Automatic Neural Network Channel Pruningcited by 227
- 2016Network Trimming: A Data-Driven Neuron Pruning Approach towards Efficient Deep Architecturescited by 221
- 2018ExpandNet: A Deep Convolutional Neural Network for High Dynamic Range Expansion from Low Dynamic Range Contentcited by 89
- 2020DMCP: Differentiable Markov Channel Pruning for Neural Networkscited by 69
- 2021Teachers Do More Than Teach: Compressing Image-to-Image Modelscited by 37
- 2021GDP: Stabilized Neural Network Pruning via Gates with Differentiable Polarizationcited by 29
- 2021BCNet: Searching for Network Width with Bilaterally Coupled Networkcited by 23
- 2020DAIS: Automatic Channel Pruning via Differentiable Annealing Indicator Searchcited by 21
- 2018StructADMM: A Systematic, High-Efficiency Framework of Structured Weight Pruning for DNNscited by 8
- 2018Systematic Weight Pruning of DNNs using Alternating Direction Method of Multiplierscited by 4
Built on by
Citing papers per year: 2022: 1; 2023: 6; 2024: 14; 2025: 8; 2026: 3.
32 papers.
Sort:
By area:
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
- 2024Finding Transformer Circuits with Edge Pruningcited by 60
- 2024LazyDiT: Lazy Learning for the Acceleration of Diffusion Transformerscited by 37
- 2024Numerical Pruning for Efficient Autoregressive Modelscited by 25
- 2024Pruning Foundation Models for High Accuracy without Retrainingcited by 20
- 2023Text-Visual Prompting for Efficient 2D Temporal Video Groundingcited by 18
- 2024Squat: Quant Small Language Models on the Edgecited by 16
- 2024Search for Efficient Large Language Modelscited by 14
- 2024Fast and Memory-Efficient Video Diffusion Using Streamlined Inferencecited by 14
- 2024Toward Adaptive Large Language Models Structured Pruning via Hybrid-grained Weight Importance Assessmentcited by 13
- 2025Structured Agent Distillation for Large Language Modelcited by 13
- 2023Performance-aware Approximation of Global Channel Pruning for Multitask CNNscited by 9
20 more
- 2025RoRA: Efficient Fine-Tuning of LLM with Reliability Optimization for Rank Adaptationcited by 9
- 2022Equivariance-aware Architectural Optimization of Neural Networkscited by 8
- 2025QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edgecited by 8
- 20247B Fully Open Source Moxin-LLM/VLM -- From Pretraining to GRPO-based Reinforcement Learning Enhancementcited by 5
- 2025Open-Source Multimodal Moxin Models with Moxin-VLM and Moxin-VLAcited by 5
- 2023UPSCALE: Unconstrained Channel Pruningcited by 4
- 2025On the creation of narrow AI: hierarchy and nonlocality of neural network skillscited by 4
- 2024Automatic Channel Pruning for Multi-Head Attentioncited by 3
- 2025Collaborative Compression for Large-Scale MoE Deployment on Edgecited by 3
- 2024Differentiable Model Scaling using Differentiable Topkcited by 2
- 2023End-to-End Neural Network Compression via fracℓ1ℓ2 Regularized Latency Surrogatescited by 1
- 2024Separate, Dynamic and Differentiable (SMART) Pruner for Block/Output Channel Pruning on Computer Vision Taskscited by 1
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platformcited by 1
- 2023A Survey on Multi-Objective Neural Architecture Searchcited by 0
- 2024Designing Semi-Structured Pruning of Graph Convolutional Networks for Skeleton-based Recognitioncited by 0
- 2024Learning Coarse-to-Fine Pruning of Graph Convolutional Networks for Skeleton-based Recognitioncited by 0
- 2025Pruning Increases Orderedness in Recurrent Computationcited by 0
- 2026Automated Model Design using Gated Neuron Selection in Telecomcited by 0
- 2026Mechanistic Interpretability of Structure-Aware Numerical Reasoning in LLaMA 3.1 8Bcited by 0
- 2026From Patching to Pruning Visual Computation in Vision Language Modelscited by 0
- 2026From Patching to Pruning Visual Computation in Vision Language Models2 Oct 2026
- 2026Mechanistic Interpretability of Structure-Aware Numerical Reasoning in LLaMA 3.1 8B19 Aug 2026
- 2026Automated Model Design using Gated Neuron Selection in Telecom11 Feb 2026
- 2025Open-Source Multimodal Moxin Models with Moxin-VLM and Moxin-VLA22 Dec 2025
- 2025Collaborative Compression for Large-Scale MoE Deployment on Edge30 Sep 2025
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platform17 Aug 2025
- 2025Pruning Increases Orderedness in Recurrent Computation19 Jul 2025
- 2025On the creation of narrow AI: hierarchy and nonlocality of neural network skills21 May 2025
- 2025Structured Agent Distillation for Large Language Model20 May 2025
- 2025QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edge20 Mar 2025
- 2025RoRA: Efficient Fine-Tuning of LLM with Reliability Optimization for Rank Adaptation8 Jan 2025
- 2024Learning Coarse-to-Fine Pruning of Graph Convolutional Networks for Skeleton-based Recognition17 Dec 2024
20 more
- 2024LazyDiT: Lazy Learning for the Acceleration of Diffusion Transformers17 Dec 2024
- 2024Numerical Pruning for Efficient Autoregressive Models17 Dec 2024
- 2024Designing Semi-Structured Pruning of Graph Convolutional Networks for Skeleton-based Recognition16 Dec 2024
- 20247B Fully Open Source Moxin-LLM/VLM -- From Pretraining to GRPO-based Reinforcement Learning Enhancement8 Dec 2024
- 2024Fast and Memory-Efficient Video Diffusion Using Streamlined Inference2 Nov 2024
- 2024Pruning Foundation Models for High Accuracy without Retraining21 Oct 2024
- 2024Search for Efficient Large Language Models25 Sep 2024
- 2024Finding Transformer Circuits with Edge Pruning24 Jun 2024
- 2024Automatic Channel Pruning for Multi-Head Attention31 May 2024
- 2024Differentiable Model Scaling using Differentiable Topk12 May 2024
- 2024Separate, Dynamic and Differentiable (SMART) Pruner for Block/Output Channel Pruning on Computer Vision Tasks29 Mar 2024
- 2024Toward Adaptive Large Language Models Structured Pruning via Hybrid-grained Weight Importance Assessment16 Mar 2024
- 2024Squat: Quant Small Language Models on the Edge16 Feb 2024
- 2023A Survey on Multi-Objective Neural Architecture Search18 Jul 2023
- 2023UPSCALE: Unconstrained Channel Pruning17 Jul 2023
- 2023End-to-End Neural Network Compression via fracℓ1ℓ2 Regularized Latency Surrogates9 Jun 2023
- 2023Performance-aware Approximation of Global Channel Pruning for Multitask CNNs21 Mar 2023
- 2023Text-Visual Prompting for Efficient 2D Temporal Video Grounding9 Mar 2023
- 2023Structured Pruning for Deep Convolutional Neural Networks: A survey1 Mar 2023
- 2022Equivariance-aware Architectural Optimization of Neural Networks11 Oct 2022
- 2025Structured Agent Distillation for Large Language Modelcited by 13
- 2025Structured Agent Distillation for Large Language Model20 May 2025
- 2024Search for Efficient Large Language Modelscited by 14
- 2024Search for Efficient Large Language Models25 Sep 2024
- 2023Structured Pruning for Deep Convolutional Neural Networks: A surveycited by 122
- 2024Finding Transformer Circuits with Edge Pruningcited by 60
- 2024Numerical Pruning for Efficient Autoregressive Modelscited by 25
- 2024Pruning Foundation Models for High Accuracy without Retrainingcited by 20
- 2023Text-Visual Prompting for Efficient 2D Temporal Video Groundingcited by 18
- 2024Squat: Quant Small Language Models on the Edgecited by 16
- 2024Toward Adaptive Large Language Models Structured Pruning via Hybrid-grained Weight Importance Assessmentcited by 13
- 2023Performance-aware Approximation of Global Channel Pruning for Multitask CNNscited by 9
- 2025RoRA: Efficient Fine-Tuning of LLM with Reliability Optimization for Rank Adaptationcited by 9
- 2022Equivariance-aware Architectural Optimization of Neural Networkscited by 8
- 2025QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edgecited by 8
- 2025Open-Source Multimodal Moxin Models with Moxin-VLM and Moxin-VLAcited by 5
12 most cited of 24 All 24 (JSON)
- 2026Automated Model Design using Gated Neuron Selection in Telecom11 Feb 2026
- 2025Open-Source Multimodal Moxin Models with Moxin-VLM and Moxin-VLA22 Dec 2025
- 2025TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platform17 Aug 2025
- 2025Pruning Increases Orderedness in Recurrent Computation19 Jul 2025
- 2025On the creation of narrow AI: hierarchy and nonlocality of neural network skills21 May 2025
- 2025QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edge20 Mar 2025
- 2025RoRA: Efficient Fine-Tuning of LLM with Reliability Optimization for Rank Adaptation8 Jan 2025
- 2024Learning Coarse-to-Fine Pruning of Graph Convolutional Networks for Skeleton-based Recognition17 Dec 2024
- 2024Numerical Pruning for Efficient Autoregressive Models17 Dec 2024
- 2024Designing Semi-Structured Pruning of Graph Convolutional Networks for Skeleton-based Recognition16 Dec 2024
- 2024Pruning Foundation Models for High Accuracy without Retraining21 Oct 2024
- 2024Finding Transformer Circuits with Edge Pruning24 Jun 2024
12 newest of 24 All 24 (JSON)