Jun, 2023

剪枝以实现更好的领域通用性

TL;DR本文研究:剪枝是否可靠地提升模型的泛化能力。我们发现现有剪枝方法像L2已经可以在目标域性能上提供小的改进。我们进一步提出了一种被称为DSS的新的剪枝评分方法,旨在直接提高模型的鲁棒性。通过实证实验,我们验证了我们的方法,并表明它可以与最先进的泛化工作(如MIRO)相结合,进一步提高性能。在MNIST到MNIST-M上,我们可以通过将60%通道稀疏引入模型来提高基线性能5个点以上。在DomainBed基准和最先进的MIRO上,我们只需将10%稀疏引入模型即可进一步提高其性能。