Learning the Number of Neurons in Deep Networks
Not on the map. Why?
Abstract (arXiv)
Nowadays, the number of layers and of neurons in each layer of a deep network are typically set manually. While very deep and wide networks have proven effective in general, they come at a high memory and computation cost, thus making them impractical for constrained platforms. These networks, however, are known to have many redundant parameters, and could thus, in principle, be replaced by more compact architectures. In this paper, we introduce an approach to automatically determining the number of neurons in each layer of a deep network during learning. To this end, we propose to make use of structured sparsity during learning. More precisely, we use a group sparsity regularizer on the parameters of the network, where each group is defined to act on a single neuron. Starting from an overcomplete network, we show that our approach can reduce the number of parameters by up to 80% while retaining or even improving the network accuracy.
Key facts. Cited by 69 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Nov 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 27; 2022: 17; 2023: 12; 2024: 7; 2025: 6.
69 papers.
Sort:
By area:
- 2021Distilling Object Detectors via Decoupled Featurescited by 105
- 2023Learning to Unlearn: Instance-wise Unlearning for Pre-trained Classifierscited by 66
- 2022Online Continual Learning for Embedded Devicescited by 48
- 2022Structural Pruning via Latency-Saliency Knapsackcited by 40
- 2021When to Prune? A Policy towards Early Structural Pruningcited by 38
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2022DropNet: Reducing Neural Network Complexity via Iterative Pruningcited by 29
- 2021Consistent Sparse Deep Learning: Theory and Computationcited by 27
- 2021ChipNet: Budget-Aware Pruning with Heaviside Continuous Approximationscited by 21
- 2021Layer-wise Model Pruning based on Mutual Informationcited by 19
- 2021Stealthy Backdoors as Compression Artifactscited by 17
- 2021Neural Network Training Using ℓ1-Regularization and Bi-fidelity Datacited by 16
38 more
- 2022Compression-aware Training of Neural Networks using Frank-Wolfecited by 16
- 2021Layer Adaptive Node Selection in Bayesian Neural Networks: Statistical Guarantees and Implementation Detailscited by 14
- 2021Convolutional Neural Network Compression through Generalized Kronecker Product Decompositioncited by 14
- 2022Provable and Efficient Continual Representation Learningcited by 14
- 2022Soft Masking for Cost-Constrained Channel Pruningcited by 12
- 2021Machine Learning for the Detection and Identification of Internet of Things (IoT) Devices: A Surveycited by 9
- 2021Modeling Ideological Salience and Framing in Polarized Online Groups with Graph Neural Networks and Structured Sparsitycited by 9
- 2021Manipulating Identical Filter Redundancy for Efficient Pruning on Deep and Complicated CNNcited by 9
- 2022Sequential Attention for Feature Selectioncited by 9
- 2022Statistical guarantees for sparse deep learningcited by 9
- 2023Efficient Neural Networks for Tiny Machine Learning: A Comprehensive Reviewcited by 9
- 2021HALP: Hardware-Aware Latency Pruningcited by 8
- 2021Deep Model Compression based on the Training Historycited by 7
- 2021Regularization and Reparameterization Avoid Vanishing Gradients in Sigmoid-Type Networkscited by 5
- 2021Distilling Image Classifiers in Object Detectorscited by 5
- 2022Pruning-aware Sparse Regularization for Network Pruningcited by 5
- 2023Low Rank Optimization for Efficient Deep Learning: Making A Balance between Compact Architecture and Fast Trainingcited by 5
- 2023Magnitude Attention-based Dynamic Pruningcited by 5
- 2021AACP: Model Compression by Accurate and Automatic Channel Pruningcited by 4
- 2023Spike-and-slab shrinkage priors for structurally sparse Bayesian neural networkscited by 4
- 2024Multi-Dimensional Pruning: Joint Channel, Layer and Block Pruning with Latency Constraintcited by 4
- 2025MDP: Multidimensional Vision Model Pruning with Latency Constraintcited by 4
- 2021Alternate Model Growth and Pruning for Efficient Training of Recommendation Systemscited by 3
- 2021Content-Aware Convolutional Neural Networkscited by 3
- 2022Entropy Induced Pruning Framework for Convolutional Neural Networkscited by 3
- 2022Learning sparse auto-encoders for green AI image codingcited by 3
- 2023Polynomially Over-Parameterized Convolutional Neural Networks Contain Structured Strong Winning Lottery Ticketscited by 3
- 2024DRIVE: Dual Gradient-Based Rapid Iterative Pruningcited by 3
- 2022Impact of Parameter Sparsity on Stochastic Gradient MCMC Methods for Bayesian Deep Learningcited by 2
- 2022Scaling Private Deep Learning with Low-Rank and Sparse Gradientscited by 2
- 2022MGiaD: Multigrid in all dimensions. Efficiency and robustness by coarsening in resolution and channel dimensionscited by 2
- 2023AICSD: Adaptive Inter-Class Similarity Distillation for Semantic Segmentationcited by 2
- 2024Enhanced Sparsification via Stimulative Trainingcited by 2
- 2025Regularization can make diffusion models more efficientcited by 2
- 2021Reframing Neural Networks: Deep Structure in Overcomplete Representationscited by 1
- 2021Schematic Memory Persistence and Transience for Efficient and Robust Continual Learningcited by 1
- 2021Lockout: Sparse Regularization of Neural Networkscited by 1
- 2021Searching for Minimal Optimal Neural Networkscited by 1
50 most cited of 69 All 69 (JSON)
- 2025Structured Sparsity and Weight-adaptive Pruning for Memory and Compute efficient Whisper models14 Oct 2025
- 2025Tin-Tin: Towards Tiny Learning on Tiny Devices with Integer-based Neural Network Training13 Apr 2025
- 2025MDP: Multidimensional Vision Model Pruning with Latency Constraint2 Apr 2025
- 2025Regularization can make diffusion models more efficient13 Feb 2025
- 2025Advancing Weight and Channel Sparsification with Enhanced Saliency5 Feb 2025
- 2025One-Cycle Structured Pruning via Stability-Driven Subnetwork Search23 Jan 2025
- 2024Sparse Deep Learning Models with the ℓ1 Regularization5 Aug 2024
- 2024A new Linear Time Bi-level ℓ1,∞ projection ; Application to the sparsification of auto-encoders neural networks23 Jul 2024
- 2024Multi-Dimensional Pruning: Joint Channel, Layer and Block Pruning with Latency Constraint17 Jun 2024
- 2024Subspace Node Pruning26 May 2024
- 2024Multi-level projection with exponential parallel speedup; Application to sparse auto-encoders neural networks3 May 2024
- 2024DRIVE: Dual Gradient-Based Rapid Iterative Pruning1 Apr 2024
38 more
- 2024Enhanced Sparsification via Stimulative Training11 Mar 2024
- 2023LayerCollapse: Adaptive compression of neural networks29 Nov 2023
- 2023Efficient Neural Networks for Tiny Machine Learning: A Comprehensive Review20 Nov 2023
- 2023Polynomially Over-Parameterized Convolutional Neural Networks Contain Structured Strong Winning Lottery Tickets16 Nov 2023
- 2023Spike-and-slab shrinkage priors for structurally sparse Bayesian neural networks17 Aug 2023
- 2023AICSD: Adaptive Inter-Class Similarity Distillation for Semantic Segmentation8 Aug 2023
- 2023Near-Linear Time Projection onto the ℓ1,∞ Ball; Application to Sparse Autoencoders19 Jul 2023
- 2023Performance of ℓ1 Regularization for Sparse Convex Optimization14 Jul 2023
- 2023A Proximal Algorithm for Network Slimming2 Jul 2023
- 2023Explainable Lifelong Stream Learning Based on "Glocal" Pairwise Fusion23 Jun 2023
- 2023Magnitude Attention-based Dynamic Pruning8 Jun 2023
- 2023Low Rank Optimization for Efficient Deep Learning: Making A Balance between Compact Architecture and Fast Training22 Mar 2023
- 2023Learning to Unlearn: Instance-wise Unlearning for Pre-trained Classifiers27 Jan 2023
- 2022Statistical guarantees for sparse deep learning11 Dec 2022
- 2022Pruning Very Deep Neural Network Channels for Efficient Inference14 Nov 2022
- 2022MGiaD: Multigrid in all dimensions. Efficiency and robustness by coarsening in resolution and channel dimensions10 Nov 2022
- 2022Soft Masking for Cost-Constrained Channel Pruning4 Nov 2022
- 2022Structural Pruning via Latency-Saliency Knapsack13 Oct 2022
- 2022Double Attention-based Lightweight Network for Plant Pest Recognition4 Oct 2022
- 2022Sequential Attention for Feature Selection29 Sep 2022
- 2022Learning sparse auto-encoders for green AI image coding9 Sep 2022
- 2022Entropy Induced Pruning Framework for Convolutional Neural Networks13 Aug 2022
- 2022DropNet: Reducing Neural Network Complexity via Iterative Pruning14 Jul 2022
- 2022Scaling Private Deep Learning with Low-Rank and Sparse Gradients6 Jul 2022
- 2022Compression-aware Training of Neural Networks using Frank-Wolfe24 May 2022
- 2022Online Continual Learning for Embedded Devices21 Mar 2022
- 2022Provable and Efficient Continual Representation Learning3 Mar 2022
- 2022Optimal channel selection with discrete QCQP24 Feb 2022
- 2022Impact of Parameter Sparsity on Stochastic Gradient MCMC Methods for Bayesian Deep Learning8 Feb 2022
- 2022Pruning-aware Sparse Regularization for Network Pruning18 Jan 2022
- 2021Challenges and Opportunities in Approximate Bayesian Deep Learning for Intelligent IoT Systems3 Dec 2021
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AI11 Nov 2021
- 2021When to Prune? A Policy towards Early Structural Pruning22 Oct 2021
- 2021HALP: Hardware-Aware Latency Pruning20 Oct 2021
- 2021Mining the Weights Knowledge for Optimizing Neural Network Structures11 Oct 2021
- 2021Convolutional Neural Network Compression through Generalized Kronecker Product Decomposition29 Sep 2021
- 2021Searching for Minimal Optimal Neural Networks27 Sep 2021
- 2021Layer-wise Model Pruning based on Mutual Information28 Aug 2021
50 newest of 69 All 69 (JSON)
- 2025Regularization can make diffusion models more efficientcited by 2
- 2025Regularization can make diffusion models more efficient13 Feb 2025
- 2021Distilling Object Detectors via Decoupled Featurescited by 105
- 2022Online Continual Learning for Embedded Devicescited by 48
- 2022Structural Pruning via Latency-Saliency Knapsackcited by 40
- 2021When to Prune? A Policy towards Early Structural Pruningcited by 38
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2022DropNet: Reducing Neural Network Complexity via Iterative Pruningcited by 29
- 2021Consistent Sparse Deep Learning: Theory and Computationcited by 27
- 2021ChipNet: Budget-Aware Pruning with Heaviside Continuous Approximationscited by 21
- 2021Layer-wise Model Pruning based on Mutual Informationcited by 19
- 2021Stealthy Backdoors as Compression Artifactscited by 17
- 2021Neural Network Training Using ℓ1-Regularization and Bi-fidelity Datacited by 16
- 2022Compression-aware Training of Neural Networks using Frank-Wolfecited by 16
12 most cited of 67 All 67 (JSON)
- 2025Structured Sparsity and Weight-adaptive Pruning for Memory and Compute efficient Whisper models14 Oct 2025
- 2025Tin-Tin: Towards Tiny Learning on Tiny Devices with Integer-based Neural Network Training13 Apr 2025
- 2025MDP: Multidimensional Vision Model Pruning with Latency Constraint2 Apr 2025
- 2025Advancing Weight and Channel Sparsification with Enhanced Saliency5 Feb 2025
- 2025One-Cycle Structured Pruning via Stability-Driven Subnetwork Search23 Jan 2025
- 2024Sparse Deep Learning Models with the ℓ1 Regularization5 Aug 2024
- 2024A new Linear Time Bi-level ℓ1,∞ projection ; Application to the sparsification of auto-encoders neural networks23 Jul 2024
- 2024Multi-Dimensional Pruning: Joint Channel, Layer and Block Pruning with Latency Constraint17 Jun 2024
- 2024Subspace Node Pruning26 May 2024
- 2024Multi-level projection with exponential parallel speedup; Application to sparse auto-encoders neural networks3 May 2024
- 2024DRIVE: Dual Gradient-Based Rapid Iterative Pruning1 Apr 2024
- 2024Enhanced Sparsification via Stimulative Training11 Mar 2024
12 newest of 67 All 67 (JSON)