- 2
- 0
- 约1.26万字
- 约 10页
- 2023-05-30 发布于四川
- 举报
本发明公开了一种基于贪心策略的图像分类模型的压缩方法。本发明通过迭代贪心的方式,设定较小的训练次数、精度和剪枝率阈值,减轻了剪枝的痛点。而且将初始神经网络作为知识蒸馏中的教师网络,用于模型微调,将网络结构交给剪枝流程来选择,减少了知识蒸馏对于网络结构选择的依赖性。本发明经过图像分类小数据集CIFAR‑10与大数据ImageNet验证可靠性和泛用性,并且选用不同的剪枝策略和模型,实验结果表明,本发明提出的压缩方法与现有的压缩方法相比,能够在几乎不损失很高精度的情况下,最大程度地搜索轻量化的网络模型
(19)国家知识产权局
(12)发明专利申请
(10)申请公布号 CN 115810128 A
(43)申请公布日 2023.03.17
(21)申请号 202211650626.8
(22)申请日 2022.12.21
(71)申请人 杭州电子科技大学
地址 31002
原创力文档

文档评论(0)