中文

神经网络初始化时剪枝的近期进展

机器学习 2022-05-25 v3 人工智能 计算机视觉与模式识别 神经与进化计算

摘要

神经网络剪枝通常从预训练收敛的模型中移除连接或神经元;而一种新的剪枝范式——初始化时剪枝(PaI)——试图对随机初始化的网络进行剪枝。本文提供了首个专注于这一新兴剪枝方式的综述。我们首先介绍神经网络剪枝的通用表述,随后介绍主要的经典剪枝主题。接着,作为本文的主体,我们对 PaI 方法进行了全面且结构化的文献回顾,包含两条主要路线(稀疏训练与稀疏选择)。最后,我们总结了 PaI 相较于 PaT 的兴起,并讨论了开放问题。除专门的文献回顾外,本文还提供了一个代码库,便于对不同 PaI 方法进行合理性检验与基准测试。

关键词

引用

@article{arxiv.2103.06460,
  title  = {Recent Advances on Neural Network Pruning at Initialization},
  author = {Huan Wang and Can Qin and Yue Bai and Yulun Zhang and Yun Fu},
  journal= {arXiv preprint arXiv:2103.06460},
  year   = {2022}
}

备注

Accepted in IJCAI'22 Survey Track. Code base: https://github.com/mingsun-tse/smile-pruning