Authors: Osama Orabi, Artur Zagitov, Hadi Salloum, Viktor A. Lobachev, Kasymkhan Khubiev, Yaroslav Kholodov
Organizations: Department of Computer Science and Engineering, Innopolis University, 420500 Innopolis, Russia · Q Deep, Laboratory of Quantum Computing, 420502 Innopolis, Russia · Research Center of the Artificial Intelligence Institute, Innopolis University, 420500 Innopolis, Russia · Moscow Institute of Physics and Technology, 141701 Dolgoprudny, Russia · Sirius University of Science and Technology, 354340 Sochi, Russia
Neural network pruning can be formulated as a combinatorial optimization problem, yet most existing approaches rely on greedy heuristics that ignore complex interactions between filters. Formal optimization methods such as Quadratic Unconstrained Binary Optimization (QUBO) provide a principled alternative but have so far underperformed due to oversimplified objective formulations based on metrics like the L1-norm. In this work, we propose a unified Hybrid QUBO framework that bridges heuristic importance estimation with global combinatorial optimization. Our formulation integrates gradient-aware sensitivity metrics - specifically first-order Taylor and second-order Fisher information - into the linear term, while utilizing data-driven activation similarity in the quadratic term. This allows the QUBO objective to jointly capture individual filter relevance and inter-filter functional redundancy. We further introduce a dynamic capacity-driven search to strictly enforce target sparsity without distorting the optimization landscape. Finally, we employ a two-stage pipeline featuring a Tensor-Train (TT) Refinement stage - a gradient-free optimizer that fine-tunes the QUBO-derived solution directly against the true evaluation metric. Experiments on the SIDD image denoising dataset demonstrate that the proposed Hybrid QUBO significantly outperforms both greedy Taylor pruning and traditional L1-based QUBO, with TT Refinement providing further consistent gains at appropriate combinatorial scales. This highlights the potential of hybrid combinatorial formulations for robust, scalable, and interpretable neural network compression.
Figures & tables
Figure 1: Overview of the proposed hybrid pruning pipeline. The process begins by analyzing a dense baseline model to compute gradient sensitivities and activation redundancy (Stages 1-2). These heuristics populate a Hybrid QUBO matrix, which is solved using a dynamic capacity search (Stage 3) to navigate the discrete energy landscape (Stage 4). The highest-quality mask seeds a Tensor-Train (PROTES) refinement stage (Stage 5) to directly optimize the true non-differentiable validation metric, yielding the final optimized network (Stages 6-7).
Figure 2: Hyperparameter search landscape showing validation PSNR across different coefficient samplings (e.g., log10(αT) vs. log10(αF) ). Each point represents a sampled configuration evaluated at a fixed sparsity. Strong solutions form continuous manifolds, indicating that performance relies fundamentally on the balance between competing energy terms.
Method
PSNR (dB)
SSIM
Baseline (Unpruned)
37.2632
0.9327
Classic QUBO (L1)
24.7963
0.6346
Taylor (Greedy)
34.8082
0.8974
Hybrid QUBO (Ours)
35.0715
0.9140
TT Refinement (Ours)
35.0715
0.9140
Table 1: Results for Full Model (36% Pruning). The Hybrid QUBO significantly outperforms the Taylor baseline.
Method ( N=4 Layers)
PSNR (dB)
SSIM
Baseline (Unpruned)
37.2632
0.9316
Taylor (Greedy)
37.0546
0.9284
Hybrid QUBO
37.1742
0.9300
TT Refinement
37.1893
0.9300
Table 2: Results for pruning 4 layers (60 total filters pruned).
Method ( N=8 Layers)
PSNR (dB)
SSIM
Baseline (Unpruned)
36.6977
0.9258
Taylor (Greedy)
33.9939
0.8912
Hybrid QUBO
35.4934
0.9091
TT Refinement
35.5075
0.9093
Table 3: Results for pruning 8 layers (200 total filters pruned).
Method ( N=16 Layers)
PSNR (dB)
SSIM
Baseline (Unpruned)
36.0683
0.9163
Taylor (Greedy)
33.8815
0.8836
Hybrid QUBO
34.4419
0.8953
TT Refinement
34.5659
0.8973
Table 4: Results for pruning 16 layers (480 total filters pruned).
Figure 3: PSNR comparison across 7 independent optimization runs. The TT Refinement consistently maintains a measurable advantage over the single-stage QUBO and the baseline Taylor methods.
Figure 4: Training and validation history of the baseline Half-UNet model over 100 epochs. The plots display the stable convergence of the custom Loss (top), Peak Signal-to-Noise Ratio (middle), and Structural Similarity Index Measure (bottom).
Figure 5: Qualitative comparison of image denoising results. Our highly compressed model (Hybrid QUBO + TT Refinement) maintains the visual structure and fidelity of the unpruned baseline without introducing pruning artefacts.
The optimization of over-parameterized deep neural networks represents a large-scale, high-dimensional, and strongly non-convex decision problem that challenges existing optimization frameworks. Current evolutionary and gradient-based pruning methods often struggle to scale to such dimensionalities, as they rely on flat search spaces, scalarized objectives, or repeated retraining, leading to premature convergence and prohibitive computational cost. This paper introduces a hierarchical importance-guided evolutionary framework that reformulates convolutional network pruning as a tractable large-scale multi-objective optimization problem. In the first phase, a continuous evolutionary search performs coarse exploration of weight-wise pruning thresholds to shrink the search space and identify promising regions of the Pareto set. The second phase applies a fine-grained binary evolutionary optimization constrained to the surviving weights, where importance-aware sampling and adaptive variation operators refine local search in the sparse region of the Pareto set. This hierarchical design combines global exploration and localized exploitation to achieve a well-distributed Pareto set of networks balancing compactness and accuracy. Empirical results on CIFAR-10 and CIFAR-100 using ResNet-56 and ResNet-110 confirm the method's effectiveness compared to existing evolutionary approaches: pruning achieves up to 51.9% and 38.9% parameter reductions with almost no accuracy loss compared to state-of-the-art evolutionary DNN pruning methods. The proposed method contributes a scalable evolutionary approach for solving very-large-scale multi-objective optimization problems, offering a general paradigm extendable to other domains where the decision space is exponentially large, objective functions are conflicting, and efficient trade-off discovery is essential.
Zak Khan, Azam Asilian Bidgoli
Department of Computer Science and Physics, Wilfrid Laurier University, Waterloo, N2L 3C5, Ontario, Canada
Deploying deep neural networks (DNNs) on edge devices requires strong compression with minimal accuracy loss. This paper introduces Mix-and-Match Pruning, a globally guided, layer-wise sparsification framework that leverages sensitivity scores and simple architectural rules to generate diverse, high-quality pruning configurations. The framework addresses a key limitation that different layers and architectures respond differently to pruning, making single-strategy approaches suboptimal. Mix-and-Match derives architecture-aware sparsity ranges, e.g., preserving normalization layers while pruning classifiers more aggressively, and systematically samples these ranges to produce ten strategies per sensitivity signal (magnitude, gradient, or their combination). This eliminates repeated pruning runs while offering deployment-ready accuracy-sparsity trade-offs. Experiments on CNNs and Vision Transformers demonstrate Pareto-optimal results, with Mix-and-Match reducing accuracy degradation on Swin-Tiny by 40% relative to standard single-criterion pruning. These findings show that coordinating existing pruning signals enables more reliable and efficient compressed models than introducing new criteria.
Danial Monachan, Samira Nazari, Mahdi Taheri +4
Brandenburg Technical University, Cottbus, Germany · University of Zanjan, Iran · Tallinn University of Technology, Estonia +1
We present HiReLC, a hierarchical ensemble-reinforcement learning framework for automated joint quantization and structured pruning of deep neural networks. The framework decomposes the compression search across two levels of abstraction: low-level agents (LLAs) operate independently per block, selecting per-kernel configurations over a multi-discrete action space spanning bitwidth, pruning keep-ratio, quantization type, and granularity, while high-level agents (HLAs) coordinate global budget allocation via ensemble voting guided by Fisher Information-based sensitivity estimates. To mitigate the computational cost of policy evaluation, an iterative active learning loop interleaves surrogate-guided RL optimization with post-compression fine-tuning, using a lightweight MLP surrogate to amortize expensive evaluations and a logit-MSE proxy during cold-start. The surrogate is used for reward shaping rather than as a replacement for final post-compression evaluation. The controller is architecture-agnostic by design, with a modular layer abstraction decoupling the RL environment from the underlying network topology. Experiments across Vision Transformer and CNN benchmarks demonstrate effective parameter-storage compression ratios of 5.99 - 6.72× with a 3.83 % gain in one setting and 0.55 - 5.62 % accuracy drops elsewhere, supporting hierarchical policy decomposition and sensitivity-aware guidance as practical design choices for joint neural network compression.
Kamar Hibatallah Baghdadi, Kawther Guoual Belhamidi, Sara Belhadj +2
aNational School of Artificial Intelligence (ENSIA), Sidi Abdellah Campus, Algiers, Algeria · bCosys-Grettia, Univ Gustave Eiffel, F-77454 Marne-la-Vallee, France. · aNational School of Artificial Intelligence (ENSIA), Sidi Abdellah