[发明专利]蒸馏系统的训练方法、装置、设备及存储介质在审
| 申请号: | 202010887422.0 | 申请日: | 2020-08-28 |
| 公开(公告)号: | CN112101545A | 公开(公告)日: | 2020-12-18 |
| 发明(设计)人: | 董大祥;覃安;黄鑫杰;王曦;刘毅;巩伟宝;侯珏 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
| 主分类号: | G06N3/08 | 分类号: | G06N3/08;G09B9/00 |
| 代理公司: | 北京清亦华知识产权代理事务所(普通合伙) 11201 | 代理人: | 王萌 |
| 地址: | 100085 北京市*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 蒸馏 系统 训练 方法 装置 设备 存储 介质 | ||
1.一种蒸馏系统的训练方法,包括:
获取第一终端设备发送的蒸馏训练请求,其中,所述蒸馏训练请求包括训练样本集和教师资源的目标类型;
根据所述目标类型获取与所述教师资源对应的目标教师模型;
将所述训练样本集输入至所述目标教师模型,生成目标训练结果,并将所述目标训练结果发送至所述第一终端设备,以使所述第一终端设备根据所述训练样本集和所述目标训练结果训练学生模型。
2.根据权利要求1所述的蒸馏系统的训练方法,其中,所述根据所述目标类型获取与所述教师资源对应的目标教师模型,包括:
获取第二终端设备发送的教师模型的存活列表,并根据所述目标类型从所述存活列表中获取所述目标教师模型。
3.根据权利要求2所述的蒸馏训练系统的训练方法,在所述获取第二终端设备发送的教师模型的存活列表之前,还包括:
获取所述第二终端设备的设备资源,根据所述设备资源在所述第二终端设备上部署所述教师模型,并生成所述教师模型的注册列表;
分别向所述注册列表中的每个所述教师模型发送探活信号,并根据探活成功的所述教师模型生成所述存活列表。
4.根据权利要求1所述的蒸馏训练系统的训练方法,还包括:
在所述学生模型进行模型训练过程中,根据第二终端设备的设备资源增加部署所述目标教师模型;
通过所述第一终端设备,分别获取增加部署所述目标教师模型之前所述学生模型的第一训练速度和增加部署所述目标教师模型之后所述学生模型的第二训练速度;
根据所述第一训练速度和所述第二训练速度,配置所述目标教师模型的个数。
5.根据权利要求4所述的蒸馏训练系统的训练方法,其中,所述根据所述第一训练速度和所述第二训练速度,配置所述目标教师模型的个数,包括:
如果所述第一训练速度小于所述第二训练速度,则增加部署所述目标教师模型;
如果所述第一训练速度等于所述第二训练速度,则减少部署所述目标教师模型。
6.根据权利要求4所述的蒸馏训练系统的训练方法,其中,所述将所述训练样本集输入至所述目标教师模型,包括:
获取所述目标教师模型的当前数量;
根据所述当前数量对所述训练样本集进行划分,以生成训练样本子集;
将所述训练样本子集分别输入至所述目标教师模型。
7.根据权利要求1所述的蒸馏训练系统的训练方法,其中,在所述将所述目标训练结果发送至所述第一终端设备之前,还包括:
识别未接收到的所述目标训练结果的持续时间;
如果所述持续时间达到预设时间,则重新获取所述蒸馏训练请求。
8.一种蒸馏系统的训练装置,包括:
获取模块,用于获取第一终端设备发送的蒸馏训练请求,其中,所述蒸馏训练请求包括训练样本集和教师资源的目标类型;
识别模块,用于根据所述目标类型获取与所述教师资源对应的目标教师模型;
训练模块,用于将所述训练样本集输入至所述目标教师模型,生成目标训练结果,并将所述目标训练结果发送至所述第一终端设备,以使所述第一终端设备根据所述训练样本集和所述目标训练结果训练学生模型。
9.根据权利要求8所述的蒸馏系统的训练装置,所述识别模块,具体用于:
获取第二终端设备发送的教师模型的存活列表,并根据所述目标类型从所述存活列表中获取所述目标教师模型。
10.根据权利要求9所述的蒸馏系统的训练装置,所述识别模块,还包括:
注册子模块,用于获取所述第二终端设备的设备资源,根据所述设备资源在所述第二终端设备上部署所述教师模型,并生成所述教师模型的注册列表;
探活子模块,用于分别向所述注册列表中的每个所述教师模型发送探活信号,并根据探活成功的所述教师模型生成所述存活列表。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010887422.0/1.html,转载请声明来源钻瓜专利网。
- 上一篇:一种试制汽车后尾门上段外板的回弹量控制方法
- 下一篇:移动式污泥处理系统





