Learning Best Combination for Efficient N:M Sparsity
Not on the map. Why?
Abstract (arXiv)
By forcing at most N out of M consecutive weights to be non-zero, the recent N:M network sparsity has received increasing attention for its two attractive advantages: 1) Promising performance at a high sparsity. 2) Significant speedups on NVIDIA A100 GPUs. Recent studies require an expensive pre-training phase or a heavy dense-gradient computation. In this paper, we show that the N:M learning can be naturally characterized as a combinatorial problem which searches for the best combination candidate within a finite collection. Motivated by this characteristic, we solve N:M sparsity in an efficient divide-and-conquer manner. First, we divide the weight vector into CMN combination subsets of a fixed size N. Then, we conquer the combinatorial problem by assigning each combination a learnable score that is jointly optimized with its associate weights. We prove that the introduced scoring mechanism can well model the relative importance between combination subsets. And by gradually removing low-scored subsets, N:M fine-grained sparsity can be efficiently optimized during the normal training phase. Comprehensive experiments demonstrate that our learning best combination (LBC) performs consistently better than off-the-shelf N:M sparsity methods across various networks. Our project is released at urlhttps://github.com/zyxxmu/LBC.
Key facts. Cited by 32 papers in the map (data through 14 Oct 2026). Builds on 34 of its 56 references that are in the map. First posted on arXiv 14 Jun 2022.
Builds on
34 of 56 references.
References · 34
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2017Mask R-CNNcited by 7,277
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2015Fast R-CNNcited by 2,933
- 2013Rich feature hierarchies for accurate object detection and semantic segmentationcited by 2,162
- 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
- 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 812
- 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 603
22 more
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 583
- 2019Rigging the Lottery: Making All Tickets Winnerscited by 495
- 2017ThiNet: A Filter Level Pruning Method for Deep Neural Network Compressioncited by 442
- 2018Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Accelerationcited by 367
- 2018Soft Filter Pruning for Accelerating Deep Convolutional Neural Networkscited by 315
- 2020HRank: Filter Pruning using High-Rank Feature Mapcited by 292
- 2019MetaPruning: Meta Learning for Automatic Neural Network Channel Pruningcited by 227
- 2017Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weightscited by 217
- 2021Learning N:M Fine-grained Structured Sparse Neural Networks From Scratchcited by 179
- 2020Soft Threshold Weight Reparameterization for Learnable Sparsitycited by 151
- 2021Accelerated Sparse Neural Training: A Provable and Efficient Method to Find N:M Transposable Maskscited by 103
- 2020Channel Pruning via Automatic Structure Searchcited by 85
- 2021Sparse Training via Boosting Pruning Plasticity with Neuroregenerationcited by 84
- 2021Top-KAST: Top-K Always Sparse Trainingcited by 79
- 2018AutoPruner: An End-to-End Trainable Filter Pruning Method for Efficient Deep Model Inferencecited by 77
- 2019Global Sparse Momentum SGD for Pruning Very Deep Neural Networkscited by 70
- 2020DSA: More Efficient Budgeted Pruning via Differentiable Sparsity Allocationcited by 47
- 2019Approximated Oracle Filter Pruning for Destructive CNN Width Optimizationcited by 35
- 2020SparseRT: Accelerating Unstructured Sparsity on GPUs for Deep Learning Inferencecited by 30
- 2021Carrying out CNN Channel Pruning in a White Boxcited by 22
- 2018Unsupervised learning of foreground object detectioncited by 15
- 2022Minimum Variance Unbiased N:M Sparsity for the Neural Gradientscited by 9
Built on by
Citing papers per year: 2023: 8; 2024: 13; 2025: 10; 2026: 1.
32 papers.
Sort:
By area:
- 2023Dynamic Sparse No Training: Training-Free Fine-tuning for Sparse LLMscited by 61
- 2024MaskLLM: Learnable Semi-Structured Sparsity for Large Language Modelscited by 50
- 2024BESA: Pruning Large Language Models with Blockwise Parameter-Efficient Sparsity Allocationcited by 45
- 2024MaskGaussian: Adaptive 3D Gaussian Representation from Probabilistic Maskscited by 41
- 2024STBLLM: Breaking the 1-Bit Barrier with Structured Binary LLMscited by 31
- 2024Progressive Gradient Flow for Robust N:M Sparsity Training in Transformerscited by 18
- 2024SparseDM: Toward Sparse Efficient Diffusion Modelscited by 18
- 2023Bi-directional Masks for Efficient N:M Sparse Trainingcited by 13
- 2025A Survey on Inference Engines for Large Language Models: Perspectives on Optimization and Efficiencycited by 12
- 2024Preserving Deep Representations In One-Shot Pruning: A Hessian-Free Second-Order Optimization Frameworkcited by 11
- 2025Optimizing LLMs for Resource-Constrained Environments: A Survey of Model Compression Techniquescited by 9
- 2024Enabling Unstructured Sparse Acceleration on Structured Sparse Acceleratorscited by 8
20 more
- 2024Data-independent Module-aware Pruning for Hierarchical Vision Transformerscited by 6
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuningcited by 6
- 2024Efficient Deep Learning Infrastructures for Embedded Computing Systems: A Comprehensive Survey and Future Envisioncited by 5
- 2025Neural Architecture Codesign for Fast Physics Applicationscited by 5
- 2025Symmetric Pruning of Large Language Modelscited by 3
- 2024ELSA: Exploiting Layer-wise N:M Sparsity for Vision Transformer Accelerationcited by 2
- 2025EvoP: Robust LLM Inference via Evolutionary Pruningcited by 2
- 2023Spatial Re-parameterization for N:M Sparsitycited by 1
- 2023SUBP: Soft Uniform Block Pruning for 1xN Sparse CNNs Multithreading Accelerationcited by 1
- 2024Torch2Chip: An End-to-end Customizable Deep Neural Network Compression and Deployment Toolkit for Prototype Hardware Accelerator Designcited by 1
- 2025One-Cycle Structured Pruning via Stability-Driven Subnetwork Searchcited by 1
- 2025MaskPro: Linear-Space Probabilistic Learning for Strict (N:M)-Sparsity on LLMscited by 1
- 2025From 2:4 to 8:16 sparsity patterns in LLMs for Outliers and Weights with Variance Correctioncited by 1
- 2025PermLLM: Learnable Channel Permutation for N:M Sparse Large Language Modelscited by 1
- 2023Revisiting Implicit Models: Sparsity Trade-offs Capability in Weight-tied Model for Vision Taskscited by 0
- 2023Neural Architecture Codesign for Fast Bragg Peak Analysiscited by 0
- 2023ELSA: Partial Weight Freezing for Overhead-Free Sparse Network Deploymentcited by 0
- 2023MaxQ: Multi-Axis Query for N:M Sparsity Networkcited by 0
- 2025Extreme Model Compression with Structured Sparsity at Low Precisioncited by 0
- 2026Defense Against LLM Backdoors using Critical Neuron Isolation Pruningcited by 0
- 2026Defense Against LLM Backdoors using Critical Neuron Isolation Pruning22 Jul 2026
- 2025Extreme Model Compression with Structured Sparsity at Low Precision11 Nov 2025
- 2025PermLLM: Learnable Channel Permutation for N:M Sparse Large Language Models11 Oct 2025
- 2025From 2:4 to 8:16 sparsity patterns in LLMs for Outliers and Weights with Variance Correction3 Jul 2025
- 2025MaskPro: Linear-Space Probabilistic Learning for Strict (N:M)-Sparsity on LLMs15 Jun 2025
- 2025Optimizing LLMs for Resource-Constrained Environments: A Survey of Model Compression Techniques5 May 2025
- 2025A Survey on Inference Engines for Large Language Models: Perspectives on Optimization and Efficiency3 May 2025
- 2025EvoP: Robust LLM Inference via Evolutionary Pruning19 Feb 2025
- 2025Symmetric Pruning of Large Language Models31 Jan 2025
- 2025One-Cycle Structured Pruning via Stability-Driven Subnetwork Search23 Jan 2025
- 2025Neural Architecture Codesign for Fast Physics Applications9 Jan 2025
- 2024MaskGaussian: Adaptive 3D Gaussian Representation from Probabilistic Masks29 Dec 2024
20 more
- 2024Preserving Deep Representations In One-Shot Pruning: A Hessian-Free Second-Order Optimization Framework27 Nov 2024
- 2024Efficient Deep Learning Infrastructures for Embedded Computing Systems: A Comprehensive Survey and Future Envision3 Nov 2024
- 2024MaskLLM: Learnable Semi-Structured Sparsity for Large Language Models26 Sep 2024
- 2024ELSA: Exploiting Layer-wise N:M Sparsity for Vision Transformer Acceleration15 Sep 2024
- 2024STBLLM: Breaking the 1-Bit Barrier with Structured Binary LLMs3 Aug 2024
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuning1 Jul 2024
- 2024Torch2Chip: An End-to-end Customizable Deep Neural Network Compression and Deployment Toolkit for Prototype Hardware Accelerator Design2 May 2024
- 2024Data-independent Module-aware Pruning for Hierarchical Vision Transformers21 Apr 2024
- 2024SparseDM: Toward Sparse Efficient Diffusion Models16 Apr 2024
- 2024Enabling Unstructured Sparse Acceleration on Structured Sparse Accelerators12 Mar 2024
- 2024BESA: Pruning Large Language Models with Blockwise Parameter-Efficient Sparsity Allocation18 Feb 2024
- 2024Progressive Gradient Flow for Robust N:M Sparsity Training in Transformers7 Feb 2024
- 2023MaxQ: Multi-Axis Query for N:M Sparsity Network12 Dec 2023
- 2023ELSA: Partial Weight Freezing for Overhead-Free Sparse Network Deployment11 Dec 2023
- 2023Neural Architecture Codesign for Fast Bragg Peak Analysis10 Dec 2023
- 2023Dynamic Sparse No Training: Training-Free Fine-tuning for Sparse LLMs13 Oct 2023
- 2023SUBP: Soft Uniform Block Pruning for 1xN Sparse CNNs Multithreading Acceleration10 Oct 2023
- 2023Revisiting Implicit Models: Sparsity Trade-offs Capability in Weight-tied Model for Vision Tasks16 Jul 2023
- 2023Spatial Re-parameterization for N:M Sparsity9 Jun 2023
- 2023Bi-directional Masks for Efficient N:M Sparse Training13 Feb 2023
- 2024SparseDM: Toward Sparse Efficient Diffusion Modelscited by 18
- 2024SparseDM: Toward Sparse Efficient Diffusion Models16 Apr 2024
- 2023Dynamic Sparse No Training: Training-Free Fine-tuning for Sparse LLMscited by 61
- 2024MaskLLM: Learnable Semi-Structured Sparsity for Large Language Modelscited by 50
- 2024BESA: Pruning Large Language Models with Blockwise Parameter-Efficient Sparsity Allocationcited by 45
- 2024MaskGaussian: Adaptive 3D Gaussian Representation from Probabilistic Maskscited by 41
- 2024STBLLM: Breaking the 1-Bit Barrier with Structured Binary LLMscited by 31
- 2024Progressive Gradient Flow for Robust N:M Sparsity Training in Transformerscited by 18
- 2023Bi-directional Masks for Efficient N:M Sparse Trainingcited by 13
- 2025A Survey on Inference Engines for Large Language Models: Perspectives on Optimization and Efficiencycited by 12
- 2024Preserving Deep Representations In One-Shot Pruning: A Hessian-Free Second-Order Optimization Frameworkcited by 11
- 2025Optimizing LLMs for Resource-Constrained Environments: A Survey of Model Compression Techniquescited by 9
- 2024Enabling Unstructured Sparse Acceleration on Structured Sparse Acceleratorscited by 8
- 2024Data-independent Module-aware Pruning for Hierarchical Vision Transformerscited by 6
12 most cited of 30 All 30 (JSON)
- 2025Extreme Model Compression with Structured Sparsity at Low Precision11 Nov 2025
- 2025PermLLM: Learnable Channel Permutation for N:M Sparse Large Language Models11 Oct 2025
- 2025From 2:4 to 8:16 sparsity patterns in LLMs for Outliers and Weights with Variance Correction3 Jul 2025
- 2025MaskPro: Linear-Space Probabilistic Learning for Strict (N:M)-Sparsity on LLMs15 Jun 2025
- 2025Optimizing LLMs for Resource-Constrained Environments: A Survey of Model Compression Techniques5 May 2025
- 2025A Survey on Inference Engines for Large Language Models: Perspectives on Optimization and Efficiency3 May 2025
- 2025EvoP: Robust LLM Inference via Evolutionary Pruning19 Feb 2025
- 2025Symmetric Pruning of Large Language Models31 Jan 2025
- 2025One-Cycle Structured Pruning via Stability-Driven Subnetwork Search23 Jan 2025
- 2025Neural Architecture Codesign for Fast Physics Applications9 Jan 2025
- 2024MaskGaussian: Adaptive 3D Gaussian Representation from Probabilistic Masks29 Dec 2024
- 2024Preserving Deep Representations In One-Shot Pruning: A Hessian-Free Second-Order Optimization Framework27 Nov 2024
12 newest of 30 All 30 (JSON)