[发明专利]计算方法、计算装置、电子设备和计算机可读存储介质在审
申请号: | 201910822448.4 | 申请日: | 2019-08-30 |
公开(公告)号: | CN110533172A | 公开(公告)日: | 2019-12-03 |
发明(设计)人: | 向辉;王奇刚;王鹏;师忠超 | 申请(专利权)人: | 联想(北京)有限公司 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08 |
代理公司: | 11021 中科专利商标代理有限责任公司 | 代理人: | 王晓晗<国际申请>=<国际公布>=<进入 |
地址: | 100085 北京市海淀*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 计算模型 丢弃概率 损失函数 运算单元 样本集 神经网络模型 训练神经网络 电子设备 计算数据 计算机可读存储介质 计算装置 训练过程 运算 样本 更新 | ||
1.一种由电子设备执行的计算方法,包括:
获得计算模型和计算数据;以及
将所述计算数据输入所述计算模型以得到计算结果;
其中,所述获得计算模型包括:
获得多个样本集;以及
利用所述多个样本集训练神经网络模型,训练完成后得到所述计算模型,所述神经网络模型包括多个运算单元,
其中,在利用所述多个样本集中的每一个样本集训练所述神经网络模型的过程中:基于所述神经网络模型的丢弃概率确定所述多个运算单元中的每个运算单元在训练过程中是否参与运算;以及获得损失函数的值,并基于所述损失函数的值更新所述丢弃概率,所述损失函数包括附加项,所述附加项与所述丢弃概率相关。
2.根据权利要求1所述的方法,其中:
所述神经网络模型还包括对应于每个运算单元的丢弃系数,所述丢弃系数基于所述丢弃概率确定,所述附加项包括所述丢弃系数;
所述基于所述神经网络模型的丢弃概率确定所述多个运算单元中的每个运算单元在训练过程中是否参与运算包括:基于所述丢弃概率确定所述每个运算单元的丢弃系数,基于所述丢弃系数确定对应的运算单元在训练过程中是否参与运算。
3.根据权利要求2所述的方法,其中:
所述丢弃系数具有第一系数值和第二系数值,所述第一系数值小于所述第二系数值,所述丢弃概率为使所述丢弃系数取第一系数值的概率;
在所述丢弃系数取所述第一系数值的情况下,与所述丢弃系数对应的运算单元不参与运算。
4.根据权利要求2所述的方法,其中:
所述附加项包括第一部分,所述第一部分为每个运算单元的丢弃系数与各自的运算时间的乘积绝对值的加和;
所述附加项还包括调节参数,所述附加项为所述调节参数与所述第一部分的乘积,其中,所述调节参数为非负常数。
5.根据权利要求4所述的方法,其中:
在所述调节参数不同的情况下,训练得到的计算模型的运行时间不同;
所述利用所述多个样本集训练神经网络模型,训练完成后得到所述计算模型还包括:
获得所述调节参数的多个取值;
基于所述多个取值中的每个取值训练得到一个计算模型,并得到每个计算模型的运行时间;
确定多个计算模型中运行时间接近预设的期望运行时间的计算模型。
6.根据权利要求2所述的方法,其中:
所述多个运算单元共用同一个丢弃概率,在训练所述神经网络模型的过程中,基于所述损失函数的值更新所述丢弃概率的值;或者
所述多个运算单元中的每个运算单元具有各自的丢弃概率,在训练所述神经网络模型的过程中,基于所述损失函数的值更新所述每个运算单元的丢弃概率的值。
7.根据权利要求1所述的方法,其中:
所述运算单元包括一个神经元或者一组神经元;
所述神经网络模型包括输入层、输出层和中间层,所述多个运算单元属于所述神经网络模型的中间层。
8.一种计算装置,包括:
获取模块,用于获得计算模型和计算数据;以及
计算模块,用于将所述计算数据输入所述计算模型以得到计算结果;
其中,所述获取模块包括:
获取单元,用于获得多个样本集;以及
训练单元,用于用所述多个样本集训练神经网络模型,训练完成后得到所述计算模型,所述神经网络模型包括多个运算单元,
其中,在利用所述多个样本集中的每一个样本集训练所述神经网络模型的过程中:基于所述神经网络模型的丢弃概率确定所述多个运算单元中的每个运算单元在训练过程中是否参与运算;以及获得损失函数的值,并基于所述损失函数的值更新所述丢弃概率,所述损失函数包括附加项,所述附加项与所述丢弃概率相关。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于联想(北京)有限公司,未经联想(北京)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910822448.4/1.html,转载请声明来源钻瓜专利网。