多任务模型训练方法、车辆驾驶方法和装置、设备及介质制造方法及图纸

技术编号:42304625 阅读:16 留言:0更新日期:2024-08-14 15:51
本申请提供了一种多任务模型训练方法、车辆驾驶方法和装置、设备及介质,属于多任务学习技术领域,通过将第一图像数据集输入至初始多任务模型进行损失计算,得到每个任务的第一子损失,初始多任务模型包括模型参数和各个任务的任务权重,根据任务权重和第一子损失得到全部任务的第一目标损失,将第二图像数据集输入至初始多任务模型进行损失计算,得到每个任务的第二子损失,根据模型参数、每个任务的任务权重、第一目标损失和每个任务的第二子损失确定目标更新梯度,根据目标更新梯度对模型参数和每个任务的任务权重进行更新,根据更新后的模型参数和每个更新后的任务权重,得到目标多任务模型,能够提高多任务学习模型对不同任务的预测准确性。

【技术实现步骤摘要】

本申请涉及多任务学习,尤其涉及一种多任务模型训练方法、车辆驾驶方法和装置、设备及介质


技术介绍

1、多任务学习模型通过共享通用特征表示,同时产生多个任务的预测结果。在车载自动驾驶系统、机器人视觉处理系统、无人机视觉处理系统等计算资源有限的系统中,能同时预测多个任务的模型变得尤为重要。由于不同类型任务的损失函数类型不同,使得多任务学习模型在梯度更新时,不同任务对应的损失函数对模型参数的贡献不同,导致模型在优化过程中偏向损失值较大的任务,忽略了其他任务。这种损失函数不平衡问题增加了多任务学习模型的训练难度,影响了多任务学习模型对于不同任务的预测准确性。因此,如何提高多任务学习模型对不同任务的预测准确性,成为了亟待解决的问题。


技术实现思路

1、本申请实施例的主要目的在于提出一种多任务模型训练方法、车辆驾驶方法和装置、设备及介质,旨在提高多任务学习模型对不同任务的预测准确性。

2、为实现上述目的,本申请实施例的第一方面提出了一种多任务模型训练方法,所述方法包括:

3、获取第一图像数据集和第本文档来自技高网...

【技术保护点】

1.多任务模型训练方法,其特征在于,所述方法包括:

2.根据权利要求1所述的多任务模型训练方法,其特征在于,所述根据所述模型参数、每个所述任务的所述任务权重、所述第一目标损失和每个所述任务的所述第二子损失确定目标更新梯度,包括:

3.根据权利要求2所述的多任务模型训练方法,其特征在于,所述根据所述第一目标损失对所述模型参数进行梯度下降,得到所述初始多任务模型的伪参数,包括:

4.根据权利要求3所述的多任务模型训练方法,其特征在于,所述根据所述第一模型参数梯度、所述第一权重参数梯度、每个所述任务的所述任务权重和所述伪参数,确定所述第一更新梯度,包括:...

【技术特征摘要】

1.多任务模型训练方法,其特征在于,所述方法包括:

2.根据权利要求1所述的多任务模型训练方法,其特征在于,所述根据所述模型参数、每个所述任务的所述任务权重、所述第一目标损失和每个所述任务的所述第二子损失确定目标更新梯度,包括:

3.根据权利要求2所述的多任务模型训练方法,其特征在于,所述根据所述第一目标损失对所述模型参数进行梯度下降,得到所述初始多任务模型的伪参数,包括:

4.根据权利要求3所述的多任务模型训练方法,其特征在于,所述根据所述第一模型参数梯度、所述第一权重参数梯度、每个所述任务的所述任务权重和所述伪参数,确定所述第一更新梯度,包括:

5.根据权利要求2所述的多任务模型训练方法,其特征在于,所述根据所述第一更新梯度和每个所述任务的所述第二更新...

【专利技术属性】
技术研发人员:张宇叶飞杨
申请(专利权)人:南方科技大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1