多任务神经网络的训练方法、装置、计算设备及存储介质
实质审查的生效
摘要
本发明公开一种多任务神经网络的训练方法、装置、计算设备及存储介质,该方法包括:获取训练数据集,并且对于每个任务随机采集训练数据集中一定数量的图片进行标注;对训练数据集中的图片进行预处理,若图片具有所有任务的标签,采用第一方式得到第一图片,反之采用第一方式和第二方式得到第一图片和第二图片;对于每一任务,如果图片具有该任务的标签,则采用交叉熵方法计算该任务的损失函数,反之则计算对应的第一图片和第二图片在该任务上的KL散度值作为损失函数;根据所有损失函数计算梯度值,并根据梯度值调整预设的多任务神经网络模型。根据本发明实施例可以大大减少训练阶段人力成本和时间成本的投入,提升模型对于多任务的预测泛化能力。
基本信息
专利标题 :
多任务神经网络的训练方法、装置、计算设备及存储介质
专利标题(英):
暂无
公开(公告)号 :
CN114332558A
申请号 :
CN202111533167.0
公开(公告)日 :
2022-04-12
申请日 :
2021-12-15
授权号 :
暂无
授权日 :
暂无
发明人 :
陈博文吴鸿伟翟永强林淑强徐晶吴顺平
申请人 :
厦门市美亚柏科信息股份有限公司
申请人地址 :
福建省厦门市思明区软件园二期观日路12号102-402单元
代理机构 :
厦门福贝知识产权代理事务所(普通合伙)
代理人 :
郭涵炜
优先权 :
CN202111533167.0
主分类号 :
G06V10/774
IPC分类号 :
G06V10/774 G06V10/82 G06N3/08 G06V40/16 G06K9/62
法律状态
2022-04-29 :
实质审查的生效
IPC(主分类) : G06V 10/774
申请日 : 20211215
申请日 : 20211215
2022-04-12 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载