[发明专利]卷积神经网络的训练方法及装置有效
申请号: | 201710539510.X | 申请日: | 2017-07-04 |
公开(公告)号: | CN107480785B | 公开(公告)日: | 2021-01-29 |
发明(设计)人: | 陈志军 | 申请(专利权)人: | 北京小米移动软件有限公司 |
主分类号: | G06N3/08 | 分类号: | G06N3/08;G06N3/04 |
代理公司: | 北京尚伦律师事务所 11477 | 代理人: | 代治国 |
地址: | 100085 北京市海淀区清河*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 卷积 神经网络 训练 方法 装置 | ||
1.一种卷积神经网络的训练方法,其特征在于,包括:
在接收到向所述卷积神经网络输入的图片样本时,获取所述卷积神经网络对所述图片样本的当前识别结果;所述图片样本包括N个三元组,其中,N为正整数;所述各三元组图片样本包括:三张图片样本,其中,所述三张图片样本中有两张图片样本属于同类图片样本,剩余的一张图片样本属于异类图片样本;
根据各三元组图片样本中异类图片样本的预设识别结果与所述同类图片样本的预设识别结果之间的预设识别差值,确定所述三元组损失函数中所述各三元组图片样本对应的识别间隔参数;所述当前识别结果包括当前识别的人体特征,人体特征包括年龄或颜值,所述预设识别结果包括预设人体特征,所述同类图片样本中的两张图片样本各自的预设人体特征的人体特征差不大于第一预设人体特征差,所述异类图片样本与所述同类图片样本中较小人体特征/平均人体特征的人体特征差大于第二预设人体特征差,其中,所述第二预设人体特征差大于所述第一预设人体特征差;
根据所述当前识别结果和所述识别间隔参数,确定所述图片样本的目标识别损失值;
根据所述图片样本的目标识别损失值,调整所述卷积神经网络的网络参数;
所述各三元组图片样本的预设识别差值越大,所述各三元组图片样本对应的识别间隔参数越大。
2.根据权利要求1所述的方法,其特征在于,
所述根据所述各三元组图片样本中异类图片样本的预设识别结果与所述同类图片样本的预设识别结果之间的预设识别差值,确定三元组损失函数中所述各三元组图片样本对应的识别间隔参数,包括:
根据预设损失阈值函数,确定所述各三元组图片样本对应的识别间隔参数,其中,所述预设损失阈值函数包括:
α=β*D+b
其中,α代表所述各三元组图片样本对应的识别间隔参数,D代表所述各三元组图片样本的预设识别差值,β为预设系数,b为预设偏置值。
3.一种卷积神经网络的训练装置,其特征在于,包括:
获取模块,用于在接收到向所述卷积神经网络输入的图片样本时,获取所述卷积神经网络对所述图片样本的当前识别结果;
第一确定模块,用于根据所述图片样本的预设识别结果,确定所述图片样本的识别间隔参数;
第二确定模块,用于根据所述当前识别结果和所述识别间隔参数,确定所述图片样本的目标识别损失值;
调整模块,用于根据所述图片样本的目标识别损失值,调整所述卷积神经网络的网络参数;
所述图片样本包括N个三元组,其中,N为正整数;
所述第一确定模块包括:
确定子模块,用于根据各三元组图片样本的预设识别结果,确定三元组损失函数中所述各三元组图片样本对应的识别间隔参数;
所述各三元组图片样本包括:三张图片样本,其中,所述三张图片样本中有两张图片样本属于同类图片样本,剩余的一张图片样本属于异类图片样本;
所述确定子模块包括:
确定单元,用于根据所述各三元组图片样本中异类图片样本的预设识别结果与所述同类图片样本的预设识别结果之间的预设识别差值,确定所述三元组损失函数中所述各三元组图片样本对应的识别间隔参数;
所述当前识别结果包括当前识别的人体特征,人体特征包括年龄或颜值,所述预设识别结果包括预设人体特征,所述同类图片样本中的两张图片样本各自的预设人体特征的人体特征差不大于第一预设人体特征差,所述异类图片样本与所述同类图片样本中较小人体特征/平均人体特征的人体特征差大于第二预设人体特征差,其中,所述第二预设人体特征差大于所述第一预设人体特征差;
所述各三元组图片样本的预设识别差值越大,所述各三元组图片样本对应的识别间隔参数越大。
4.根据权利要求3所述的装置,其特征在于,
所述确定单元包括:
确定子单元,用于根据预设损失阈值函数,确定所述各三元组图片样本对应的识别间隔参数,其中,所述预设损失阈值函数包括:
α=β*D+b
其中,α代表所述各三元组图片样本对应的识别间隔参数,D代表所述各三元组图片样本的预设识别差值,β为预设系数,b为预设偏置值。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京小米移动软件有限公司,未经北京小米移动软件有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201710539510.X/1.html,转载请声明来源钻瓜专利网。