一种神经网络模型生成方法及装置

专利申请号：CN202110103858.0

公 开 号：CN112836819B

发 明 人：宇哲伦 

代 理 人：北京柏杉松知识产权代理事务所(普通合伙)项京;马敬

代理机构：北京柏杉松知识产权代理事务所(普通合伙)

专利类型：发明专利

申 请 日：20230725

公 开 日：20210126

专利主分类号：G06N3/08

关 键 词：神经网络模型 网络模型 目标神经 损失函数 目标卷 模型参数计算 结构复杂度 模型参数 模型训练 目标样本 预设结构 卷积核 标准化 图像 

摘      要：本发明实施例提供了一种神经网络模型生成方法及装置，方法包括：获取预设结构的第一神经网络模型；其中，第一神经网络模型包括至少一个批标准化BN层；基于目标样本图像和目标损失函数，对第一神经网络模型进行模型训练，得到第二神经网络模型；其中，目标损失函数为基于原始损失函数和BN层的模型参数计算得到的；确定第二神经网络模型中除目标卷积核以外的网络模型结构，作为第三神经网络模型；其中，目标卷积核为第二神经网络模型中对应的BN层的模型参数的绝对值小于目标阈值的卷积核；基于第三神经网络模型，生成目标神经网络模型。如此，能够降低目标神经网络模型的结构复杂度，提高目标神经网络模型的适用性。