一种神经网络剪枝方法及装置

摘要

本发明公开了一种神经网络剪枝方法及装置，以解决现有技术中网络剪枝不能兼顾压缩、加速和精度的技术问题。该方法：根据待剪枝网络层中神经元的激活值，确定神经元的重要度值；根据所述待剪枝网络层中神经元与下一个网络层中神经元的连接权重，确定神经元的多样性值；根据所述待剪枝网络层中的神经元的重要度值和多样性值，采用体积最大化神经元选取策略从所述待剪枝网络层中选取保留的神经元；将所述待剪枝网络层中的其他神经元剪掉，得到剪枝网络层。本发明技术方案能够在确保神经网络精度的同时取得了较好的压缩和加速效果。

主权项

一种神经网络剪枝方法，其特征在于，所述方法包括：根据待剪枝网络层中神经元的激活值，确定神经元的重要度值；根据所述待剪枝网络层中神经元与下一个网络层中神经元的连接权重，确定神经元的多样性值；根据所述待剪枝网络层中的神经元的重要度值和多样性值，采用体积最大化神经元选取策略从所述待剪枝网络层中选取保留的神经元；将所述待剪枝网络层中的其他神经元剪掉，得到剪枝网络层。