.png){kind=icon}
Return
Spatial Re-Parameterization for N:M Sparsity
.png){kind=icon}
OA
AI
DOI：10.1109/TPAMI.2025.3572240.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
This paper presents a Spatial Re-parameterization (SpRe) method for the N:M sparsity. SpRe stems from an observation regarding the restricted variety in spatial sparsity of convolution kernels presented in N:M sparsity compared with unstructured sparsity. Particularly, N:M sparsity exhibits a fixed sparsity rate within the spatial domains due to its distinctive pattern that mandates N non-zero components among M successive weights in the input channel dimension of convolution filters. On the contrary, we observe that conventional unstructured sparsity displays a substantial divergence in sparsity across the spatial domains, which we experimentally verify to be very crucial for its robust performance retention compared with N:M sparsity. Therefore, SpRe employs the spatial-sparsity distribution of unstructured sparsity by assigning an extra branch in conjunction with the original N:M branch at training time, which allows the N:M sparse network to sustain a similar distribution of spatial sparsity with unstructured sparsity. During inference, the extra branch can be further re-parameterized into the main N:M branch, without exerting any distortion on the sparse pattern or additional computation costs. SpRe has achieved a commendable feat by matching the performance of N:M sparsity methods with state-of-the-art unstructured sparsity methods across various benchmarks. Our project is available at https://github.com/zyxxmu/SpRE.
Keywords:
Convolutional neural networks
network sparsity
N:M sparsity
Journal
IF:
18.6
Papers:
1.1K
Citations:
9.8W
Organization
View more organizations
Cited Papers
ShareSaveSparseRT.png)
Proceedings of the ACM International Conference on Parallel Architectures and Compilation Techniques
OAAI
Proceedings of the ACM International Conference on Parallel Architectures and Compilation TechniquesIF0
OAAIShareSaveShareSave View more