Jun, 2023

N:M 稀疏性的空间重参数化

TL;DR本文提出了 Spatial Re-parameterization (SpRe) 方法,用于 N:M 稀疏性在 CNN 中的应用。该方法可以在训练期间利用非结构化稀疏性的空间稀疏性分布来为 N:M 稀疏性模型分配一个额外的支路,使其可以维持类似于非结构化稀疏性的空间稀疏性分布。在推理期间,该额外的支路可以进一步重新参数化为主要的 N:M 支路,而不会对稀疏模式或额外的计算成本产生扭曲。SpRe 方法可以在各种基准测试中匹配 N:M 稀疏性方法的表现与最先进的非结构化稀疏性方法。