Invention Grant
- Patent Title: 分布式训练系统、神经网络模型训练方法、设备和介质
-
Application No.: CN202310928151.2Application Date: 2023-07-27
-
Publication No.: CN116644804BPublication Date: 2024-01-12
- Inventor: 闫瑞栋 , 郭振华 , 李仁刚 , 邱志勇
- Applicant: 浪潮电子信息产业股份有限公司
- Applicant Address: 山东省济南市高新区浪潮路1036号
- Assignee: 浪潮电子信息产业股份有限公司
- Current Assignee: 浪潮电子信息产业股份有限公司
- Current Assignee Address: 山东省济南市高新区浪潮路1036号
- Agency: 北京集佳知识产权代理有限公司
- Agent 李慧引
- Main IPC: G06N3/098
- IPC: G06N3/098

Abstract:
本发明公开了一种分布式训练系统、神经网络模型训练方法、设备和介质,涉及计算机技术领域,该系统包括控制器和多个计算节点;控制器用于获取目标神经网络模型和训练数据集,并向多个计算节点发送对应的节点训练模型和节点训练数据集;目标神经网络模型为初始的神经网络模型去除第i个神经网络层与第i+1个神经网络层之间链接重要程度最低的第一数量个链接、去除神经元重要程度最低的第二数量个神经元以及对应的链接得到神经网络模型;计算节点用于基于接收到的节点训练数据集和接收到的节点训练模型的当前模型参数计算局部信息,并基于全局信息更新节点训练模型的模型参数,实现节点训练模型的迭代训练,得到训练完成的节点训练模型。
Public/Granted literature
- CN116644804A 分布式训练系统、神经网络模型训练方法、设备和介质 Public/Granted day:2023-08-25
Information query