摘要
结构化剪枝已成为构建更高效模型的一种很有前景的方法。然而,该领域缺乏标准化的基准和指标,导致这一领域的进展并未得到充分理解。为了填补这一空白,我们提出了第一个全面的结构化剪枝基准测试,名为PruningBench。PruningBench具有以下三个特点:1)PruningBench采用统一一致的框架来评估各种结构化剪枝技术的有效性;2)PruningBench系统地评估了16种现有的剪枝方法,涵盖了各种模型(例如,CNN和ViT)和任务(例如,分类和检测);3)PruningBench提供易于实现的接口,以方便未来剪枝方法的实现,并使后续研究人员能够将其工作纳入我们的排行榜。我们提供了一个在线剪枝平台http://pruning.vipazoo.cn,用于自定义剪枝任务和复现本文中的所有结果。排行榜结果可在https://github.com/HollyLee2000/PruningBench上获取。