- 专利标题: 基于均值迭代的多任务学习模型训练以及预测方法
-
申请号: CN201911003056.1申请日: 2019-10-22
-
公开(公告)号: CN110738270B公开(公告)日: 2022-03-11
- 发明人: 周鋆 , 孙立健 , 符鹏涛 , 朱先强 , 张维明
- 申请人: 中国人民解放军国防科技大学
- 申请人地址: 湖南省长沙市开福区德雅路109号
- 专利权人: 中国人民解放军国防科技大学
- 当前专利权人: 中国人民解放军国防科技大学
- 当前专利权人地址: 湖南省长沙市开福区德雅路109号
- 代理机构: 长沙国科天河知识产权代理有限公司
- 代理商 周达
- 主分类号: G06K9/62
- IPC分类号: G06K9/62
摘要:
一种基于均值迭代的多任务学习模型训练以及预测方法,先获取多个任务的样本数据集,并将各任务的样本数据集均划分为训练集和测试集。对于每一个任务,利用其对应的训练集得到各任务上各类标签的先验概率以及各任务中各特征变量在各类标签上的条件概率,计算每一个任务对应的测试集中各实例的类标签。基于均值迭代方法更新各任务上各类标签的先验概率。不断循环迭代,直至不同任务上类标签的先验概率误差的绝对值之和Δ小于设定的阈值时收敛,得到训练好的多任务学习模型。本发明可以显著提高多任务学习的效率。同时,本发明可以更加充分利用任务之间的共享信息和数据的先验知识,利用较少的计算资源便可以达到更佳的分类效果。
公开/授权文献
- CN110738270A 基于均值迭代的多任务学习模型训练以及预测方法 公开/授权日:2020-01-31