有效

神经网络的压缩方法和装置、设备、介质

洪文静、李皈颖、杨鹏、唐珂
南方科技大学

摘要

本实施例提供一种神经网络的压缩方法和装置、设备、介质,属于计算机应用技术领域。该方法包括:获取在硬件上运行的预设神经网络;对所述预设神经网络进行压缩处理,以生成至少两个第一压缩神经网络;获取每一所述第一压缩神经网络对应于所述硬件的硬件性能指标;计算所述预设神经网络的第一准确率和每一所述第一压缩神经网络的第二准确率;根据所述硬件性能指标、所述第一准确率、多个所述第二准确率对多个所述第一压缩神经网络进行初步筛选,得到初始神经网络集;根据预设条件对所述初始神经网络集进行二次筛选,得到目标压缩神经网络。根据硬件性能指标和准确率来指导神经网络的压缩,能够实现神经网络在多种硬件上的快速部署。

暂无引用专利