在审

深度神经网络模型压缩方法及装置、电子设备、存储介质

李晶、郑哲、刘瑞、崔文朋、聂玉虎、池颖英、徐鲲鹏、李腾浩、杨玎、黄桂林、胡戈飚、习雨同、曹波
北京智芯微电子科技有限公司

摘要

本发明涉及人工智能领域,提供一种深度神经网络模型压缩方法及装置。所述方法包括:针对选定的深度神经网络模型提供随机初始化参数;基于所述随机初始化参数对所述深度神经网络模型的通道进行选择;对通道选择后的深度神经网络模型进行剪枝得到裁剪后的网络模型;利用量化参数对裁剪后的网络模型进行量化得到压缩的网络模型。本发明利用随机初始化参数对网络模型结构进行剪枝,无需预训练模型,节省了预训练模型的时间成本,增加了网络模型结构的剪枝空间,保证模型精度,提升模型剪枝效率,能够快速压缩模型。

暂无引用专利