Oct, 2023

Feather: 一种高效 DNN 稀疏化的优雅解决方案

TL;DR利用强大的直通估计作为核心,结合新的阈值运算符和梯度缩放技术,Feather 是一种有效的稀疏训练模块,能够在标准训练过程中实现健壮的稀疏化性能,通过在 CIFAR 数据集上使用各种体系结构进行演示,而在 ImageNet 上,它使用 ResNet-50 体系结构实现了最新的 Top-1 验证准确性,超过了现有方法,并且具有显著的优势,包括更复杂和计算量较大的方法。