基于类别掩码的神经网络剪枝方法.pdfVIP

  • 2
  • 0
  • 约7.84千字
  • 约 9页
  • 2023-07-06 发布于四川
  • 举报
基于类别掩码的神经网络剪枝方法,涉及人工神经网络的压缩与加速。通过对不同通道激活的特征图的深入可视化,观察到不同通道对图像分类中不同类别的贡献不同。选择保留有助于大多数类别的渠道。具体来说,为模拟每个通道对差异类别的贡献,为每个通道分配一个类别掩码,从而得到每个通道对不同类识别的贡献。包括以下步骤:1)为神经网络中每一个通道分配一个类别掩码,其维度等于数据集图像类别数;2)基于输入图像的类别激活相应的掩码来进行训练网络;3)基于训练后的类别掩码进行全局投票,剪枝分数较小的通道;4)对保留下来的卷

(19)中华人民共和国国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 113837377 A (43)申请公布日 2021.12.24 (21)申请号 202111005877.6 (22)申请日 2021.08.30 (71)申请人 厦门大学 地址 36

您可能关注的文档

文档评论(0)

1亿VIP精品文档

相关文档