一种基于贪心策略的图像分类模型的压缩方法.pdfVIP

  • 2
  • 0
  • 约1.26万字
  • 约 10页
  • 2023-05-30 发布于四川
  • 举报

一种基于贪心策略的图像分类模型的压缩方法.pdf

本发明公开了一种基于贪心策略的图像分类模型的压缩方法。本发明通过迭代贪心的方式,设定较小的训练次数、精度和剪枝率阈值,减轻了剪枝的痛点。而且将初始神经网络作为知识蒸馏中的教师网络,用于模型微调,将网络结构交给剪枝流程来选择,减少了知识蒸馏对于网络结构选择的依赖性。本发明经过图像分类小数据集CIFAR‑10与大数据ImageNet验证可靠性和泛用性,并且选用不同的剪枝策略和模型,实验结果表明,本发明提出的压缩方法与现有的压缩方法相比,能够在几乎不损失很高精度的情况下,最大程度地搜索轻量化的网络模型

(19)国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 115810128 A (43)申请公布日 2023.03.17 (21)申请号 202211650626.8 (22)申请日 2022.12.21 (71)申请人 杭州电子科技大学 地址 31002

文档评论(0)

1亿VIP精品文档

相关文档