原创力文档- 3
- 0
- 约2.11万字
- 约 16页
- 2023-05-29 发布于四川
- 举报
本发明公开了一种基于强化学习的CNN网络剪枝率自动搜索方法及系统，属于人工智能领域。本发明给定网络整体FLOPs保留率，利用深度强化学习TD3智能体来学习卷积神经网络各层最优剪枝率，以待剪枝网络的每层特征为状态，层保留率为动作，基于孪生网络框架计算层级奖励，并利用特征图比较方法提升层级奖励函数的计算效率，与网络级奖励相结合，搜索出最优的各层剪枝率，并对完成预训练的待剪枝网络执行通道剪枝，最后通过微调获得高精度的压缩后网络。该方法及系统能够较好地发现高冗余网络及紧凑网络中各层的潜在稀疏敏感度，从而
(19)国家知识产权局
(12)发明专利申请
(10)申请公布号 CN 115829022 A
(43)申请公布日 2023.03.21
(21)申请号 202211436627.2
(22)申请日 2022.11.16
(71)申请人 西安交通大学
地址 710048
文档评论（0）