LilNetX: Lightweight Networks with EXtreme Model Compression and Structured Sparsification
Not on the map. Why?
Abstract (arXiv)
We introduce LilNetX, an end-to-end trainable technique for neural networks that enables learning models with specified accuracy-rate-computation trade-off. Prior works approach these problems one at a time and often require post-processing or multistage training which become less practical and do not scale very well for large datasets or architectures. Our method constructs a joint training objective that penalizes the self-information of network parameters in a reparameterized latent space to encourage small model size while also introducing priors to increase structured sparsity in the parameter space to reduce computation. We achieve up to 50% smaller model size and 98% model sparsity on ResNet-20 while retaining the same accuracy on the CIFAR-10 dataset as well as 35% smaller model size and 42% structured sparsity on ResNet-50 trained on ImageNet, when compared to existing state-of-the-art model compression methods. Code is available at https://github.com/Sharath-girish/LilNetX.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 75 of its 97 references that are in the map. First posted on arXiv 6 Apr 2022.
Builds on
75 of 97 references.
References · 75
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 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
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,593
- 2016Pruning Filters for Efficient ConvNetscited by 1,143
38 more
- 2018Variational image compression with a scale hyperpriorcited by 935
- 2016XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networkscited by 849
- 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 812
- 2017Channel Pruning for Accelerating Very Deep Neural Networkscited by 702
- 2017Learning Efficient Convolutional Networks through Network Slimmingcited by 676
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 592
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 583
- 2016Learning Structured Sparsity in Deep Neural Networkscited by 541
- 2019Rigging the Lottery: Making All Tickets Winnerscited by 495
- 2020Pruning neural networks without any data by iteratively conserving synaptic flowcited by 479
- 2020Picking Winning Tickets Before Training by Preserving Gradient Flowcited by 471
- 2018HAQ: Hardware-Aware Automated Quantization with Mixed Precisioncited by 443
- 2017ThiNet: A Filter Level Pruning Method for Deep Neural Network Compressioncited by 442
- 2016Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activationscited by 426
- 2018Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Accelerationcited by 367
- 2019Data-Free Quantization Through Weight Equalization and Bias Correctioncited by 336
- 2018Soft Filter Pruning for Accelerating Deep Convolutional Neural Networkscited by 315
- 2020The Lottery Ticket Hypothesis for Pre-trained BERT Networkscited by 275
- 2018LQ-Nets: Learned Quantization for Highly Accurate and Compact Deep Neural Networkscited by 274
- 2021NeRV: Neural Representations for Videoscited by 256
- 2020Comparing Rewinding and Fine-tuning in Neural Network Pruningcited by 240
- 2017Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weightscited by 217
- 2015Quantized Convolutional Neural Networks for Mobile Devicescited by 215
- 2016Trained Ternary Quantizationcited by 208
- 2016Ternary Weight Networkscited by 204
- 2020Proving the Lottery Ticket Hypothesis: Pruning is All You Needcited by 201
- 2014Compressing Deep Convolutional Networks using Vector Quantizationcited by 195
- 2020Pruning Neural Networks at Initialization: Why are We Missing the Mark?cited by 182
- 2017Data-Driven Sparse Structure Selection for Deep Neural Networkscited by 181
- 2021Learning N:M Fine-grained Structured Sparse Neural Networks From Scratchcited by 179
- 2018Discrimination-aware Channel Pruning for Deep Neural Networkscited by 171
- 2019Drawing Early-Bird Tickets: Towards More Efficient Training of Deep Networkscited by 160
- 2019Improving Neural Network Quantization without Retraining using Outlier Channel Splittingcited by 151
- 2015Compressing Neural Networks with the Hashing Trickcited by 150
- 2019Gate Decorator: Global Filter Pruning Method for Accelerating Deep Convolutional Neural Networkscited by 118
- 2017Bayesian Compression for Deep Learningcited by 116
- 2019Winning the Lottery with Continuous Sparsificationcited by 100
50 most cited of 75 All 75 (JSON)
Built on by
Citing papers per year: 2022: 1; 2023: 4; 2024: 2.
7 papers.
- 2023EAGLES: Efficient Accelerated 3D Gaussians with Lightweight EncodingScited by 196
- 2022NIRVANA: Neural Implicit Representations of Videos with Adaptive Networks and Autoregressive Patch-wise Modelingcited by 51
- 2023SHACIRA: Scalable HAsh-grid Compression for Implicit Neural Representationscited by 31
- 2023ECRF: Entropy-Constrained Neural Radiance Fields Compression with Frequency Domain Optimizationcited by 8
- 2024Neural NeRF Compressioncited by 4
- 2024MCNC: Manifold-Constrained Reparameterization for Neural Compressioncited by 4
- 2023Bandwidth-efficient Inference for Neural Image Compressioncited by 1
- 2024MCNC: Manifold-Constrained Reparameterization for Neural Compression27 Jun 2024
- 2024Neural NeRF Compression13 Jun 2024
- 2023EAGLES: Efficient Accelerated 3D Gaussians with Lightweight EncodingS7 Dec 2023
- 2023ECRF: Entropy-Constrained Neural Radiance Fields Compression with Frequency Domain Optimization23 Nov 2023
- 2023SHACIRA: Scalable HAsh-grid Compression for Implicit Neural Representations27 Sep 2023
- 2023Bandwidth-efficient Inference for Neural Image Compression6 Sep 2023
- 2022NIRVANA: Neural Implicit Representations of Videos with Adaptive Networks and Autoregressive Patch-wise Modeling30 Dec 2022