- 专利标题: 一种基于分布式数据的多任务模型训练方法及系统
-
申请号: CN202010849745.0申请日: 2020-08-21
-
公开(公告)号: CN111813869B公开(公告)日: 2021-03-23
- 发明人: 方伟琪
- 申请人: 支付宝(杭州)信息技术有限公司
- 申请人地址: 浙江省杭州市西湖区西溪路556号8层B段801-11
- 专利权人: 支付宝(杭州)信息技术有限公司
- 当前专利权人: 支付宝(杭州)信息技术有限公司
- 当前专利权人地址: 浙江省杭州市西湖区西溪路556号8层B段801-11
- 代理机构: 北京博思佳知识产权代理有限公司
- 代理商 周嗣勇
- 主分类号: G06F16/27
- IPC分类号: G06F16/27 ; G06Q40/02
摘要:
本说明书公开了一种基于分布式数据的多任务模型训练方法及系统。所述方法包括:循环执行以下步骤,直到满足循环停止条件:针对任一任务,训练节点确定存储目标数据节点;对目标数据节点当前的本地模型进行独立训练,基于独立训练后的目标数据节点本地模型损失函数的梯度确定该任务的梯度;训练节点对多个任务的梯度进行汇总,并利用梯度汇总结果更新训练节点当前的本地模型参数集;训练节点向所述系统中的数据节点下发当前的本地模型参数集,以使所述系统中每个数据节点基于接收到的模型参数集更新本地模型;循环停止后,针对任一任务,将训练节点当前的本地模型参数集作为模型初始参数集,以得到针对该任务的特定模型。
公开/授权文献
- CN111813869A 一种基于分布式数据的多任务模型训练方法及系统 公开/授权日:2020-10-23