cs.LGAug 26, 2026

Pruning Binarized Neural Networks: A Dedicated Framework and Globally Weighted Algorithms

Authors: Roan Rubiales, Jean Pierre David

Organizations: Polytechnique Montreal

Abstract

Extreme compression of deep neural networks, up to full binarization, dramatically reduces memory footprint and arithmetic complexity, facilitating deployment on constrained edge hardware with field-programmable gate arrays (FPGAs) and microcontrollers. Although combining binarization with pruning promises additional efficiency gains, existing pruning strategies are ill-suited to binarized representations and rarely translate into meaningful hardware savings. We introduce a PyTorch-based, research-oriented framework that incorporates freezing and pruning mechanisms for designing and optimizing binarized neural networks. The framework enables rapid and reproducible evaluation of state-of-the-art approaches and the fast prototyping of new ones. Leveraging this framework, we propose a novel pruning method that accounts for the relative importance of learned parameters across abstraction levels. Such a global weighting mechanism consistently achieves a superior trade-off between model accuracy and pruning rate, achieving a 70% pruning rate on VGG11 with constant accuracy, while state-of-the-art results reach only 41% in the binarized setting.

Figures & tables

Explore similar work

May 23, 2026

Pruning Deep Neural Networks via the Marchenko--Pastur Distribution

We study a Marchenko--Pastur (MP) random-matrix approach to pruning deep neural networks with very small post-pruning fine-tuning budgets. The main practical contribution is accuracy retention under short calibration and fine-tuning schedules, rather than a long post-pruning reoptimization pipeline. The theory gives deterministic data-path certificates: if the removed component RR has small propagated logit effect Ls∥Rψ1(s)∥∞L_s \| R ψ_1(s) \|_\infty, pruning decreases an elastic-net objective and preserves samples whose dense margin exceeds twice the perturbation. The zero-budget case gives perfect pruning; a prune--restore extension models weight restoration inside a fixed sparse-execution pattern; and an additive L2L_2-regularized model shows admissible random-like components vanish at the training limit, with persistent spikes stabilizing as the MP bulk collapses. Under iid-Gaussian sufficient conditions, the fitted MP edge σ+σ_+ gives a high-probability layerwise budget signal. On ImageNet-1k, after only three distillation epochs, ViT-B/16 2:4+2{:}4{+}ToMe reaches 83.41%83.41\% top-1 (−1.70-1.70 pp from dense) at 59.81%59.81\% sparse-execution MAC reduction, with 1.388×1.388\times best-observed A40 native-2:42{:}4 backend speedup for the same checkpoint and ToMe graph; a separate no-ToMe A100 endpoint gives 2.705×2.705\times. At structured sparsity, ViT-B/16 6:126{:}12 reaches 83.74%83.74\%, ViT-L/16 8:168{:}16 dense+permutation reaches 85.33%85.33\% (−0.51-0.51 pp), and ConvNeXtV2-Base 12:1612{:}16 reaches 86.35%86.35\% (−0.37-0.37 pp). For CNNs, ResNet50 8:168{:}16 dense+permutation reaches 75.87%75.87\% (−0.26-0.26 pp), and ResNet152d CAST-conv+permutation reaches 81.33%81.33\% (−1.53-1.53 pp) at ∼50%{\sim}50\% MAC accounting with a 1.62×1.62\times A40 im2col+2:4+2{:}4 sparse-GEMM audit.
Mar 16, 2026

Federated Learning of Binary Neural Networks: Enabling Low-Cost Inference

Federated Learning (FL) preserves privacy by distributing training across devices. However, using DNNs is computationally intensive at the low-powered edge during inference. Edge deployment demands models that simultaneously optimize memory footprint and computational efficiency, a dilemma where conventional DNNs fail by exceeding resource limits. Traditional post-training binarization reduces model size but suffers from severe accuracy loss due to quantization errors. To address these challenges, we propose FedBNN, a rotation-aware binary neural network framework that learns binary representations directly during local training. By encoding each weight as a single bit {+1,−1}\{+1, -1\} instead of a 3232-bit float, FedBNN shrinks the model footprint, significantly reducing runtime (during inference) FLOPs and memory requirements in comparison to federated methods using real models. Evaluations across multiple benchmark datasets demonstrate that FedBNN significantly reduces resource consumption while performing similarly to existing federated methods using real-valued models.
Feb 19, 2026

Bonsai: A Framework for Convolutional Neural Network Acceleration Using Criterion-Based Pruning

As the need for more accurate and powerful Convolutional Neural Networks (CNNs) increases, so too does the size, execution time, memory footprint, and power consumption. To overcome this, solutions such as pruning have been proposed with their own metrics and methodologies, or criteria, for how weights should be removed. These solutions do not share a common implementation and are difficult to implement and compare. In this work, we introduce Combine, a criterion- based pruning solution and demonstrate that it is fast and effective framework for iterative pruning, demonstrate that criterion have differing effects on different models, create a standard language for comparing criterion functions, and propose a few novel criterion functions. We show the capacity of these criterion functions and the framework on VGG inspired models, pruning up to 79% of filters while retaining or improving accuracy, and reducing the computations needed by the network by up to 68%.