训练卷积神经网络的方法及装置、计算机设备和存储介质

基本信息

申请号 CN201910578722.8 申请日 -
公开(公告)号 CN110288090A 公开(公告)日 2019-09-27
申请公布号 CN110288090A 申请公布日 2019-09-27
分类号 G06N3/08;G06N3/04 分类 计算;推算;计数;
发明人 张韵东;黎明;夏昌盛;李国新;昝劲文 申请(专利权)人 广东中星微电子有限公司
代理机构 北京布瑞知识产权代理有限公司 代理人 广东中星微电子有限公司
地址 519031 广东省珠海市横琴新区宝华路6号105室-15070
法律状态 -

摘要

摘要 本发明提供了一种训练卷积神经网络的方法及装置、计算机设备和存储介质,该方法包括在利用包含K个样品的训练集对卷积神经网络进行推理运算时,对卷积神经网络的每个卷积层的M个输出通道中的每个输出通道的特征图进行平均计算,得到K个M维特征向量;对K个M维特征向量进行主成分分析PCA操作以获得M个新的空间主轴和K个M维特征向量在新的空间主轴上的M个方差值;根据M个方差值,确定缩减后的每个卷积层的输出通道数;以及根据缩减后的每个卷积层的输出通道数设计缩减后的卷积神经网络,并对缩减后的卷积神经网络进行再训练以获得新的卷积神经网络,能够精确地确定出缩减后的每个卷积层的输出通道数,减少卷积神经网络的设计花销时间。