[发明专利]模型训练、文本预测方法、装置、电子设备及介质有效
申请号: | 202111479311.7 | 申请日: | 2021-12-07 |
公开(公告)号: | CN113901799B | 公开(公告)日: | 2022-03-08 |
发明(设计)人: | 李仁刚;赵雅倩;郭振华;徐聪;范宝余;金良;刘璐;闫瑞栋 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06F40/274 | 分类号: | G06F40/274;G06F40/205;G06N3/08 |
代理公司: | 北京集佳知识产权代理有限公司 11227 | 代理人: | 吕鑫 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 模型 训练 文本 预测 方法 装置 电子设备 介质 | ||
1.一种模型训练方法,其特征在于,包括:
获取训练文本,并利用所述训练文本和预设文本模板生成训练输入数据;
将所述训练输入数据输入copynet模型,得到训练输出数据;
将所述训练输出数据输入预训练模型,得到训练预测值;
利用所述训练预测值与所述训练文本的训练标签计算损失值,并利用所述损失值调节所述copynet模型的参数;
若检测到满足训练完成条件,则将参数调节后的copynet模型确定为文本特征提取模型。
2.根据权利要求1所述的模型训练方法,其特征在于,所述copynet模型具有dropout机制,所述copynet模型的dropout机制处于激活状态;所述将参数调节后的copynet模型确定为文本特征提取模型,包括:
将所述参数调节后的copynet模型的dropout机制设置为未激活状态,得到所述文本特征提取模型。
3.根据权利要求2所述的模型训练方法,其特征在于,所述将所述训练输入数据输入copynet模型,得到训练输出数据,包括:
利用所述copynet模型对所述训练输入数据进行两次基于随机dropout的特征提取处理,得到第一训练输出数据和第二训练输出数据;
相应的,所述将所述训练输出数据输入预训练模型,得到训练预测值,包括:
将所述第一训练输出数据和所述第二训练输出数据分别输入所述预训练模型,得到第一训练预测值和第二训练预测值。
4.根据权利要求3所述的模型训练方法,其特征在于,所述利用所述训练预测值与所述训练文本的训练标签计算损失值,并利用所述损失值调节所述copynet模型的参数,包括:
利用所述第一训练预测值和所述第二训练预测值计算相对熵损失值;
利用所述训练标签分别与所述第一训练预测值和所述第二训练预测值进行损失计算,得到第一损失值和第二损失值;
利用所述相对熵损失值、所述第一损失值和所述第二损失值生成所述损失值。
5.根据权利要求1所述的模型训练方法,其特征在于,所述利用所述训练文本和预设文本模板生成训练输入数据,包括:
将所述训练文本插入所述预设文本模板的指定位置,得到训练输入序列;
对所述训练输入序列进行词向量编码,得到所述训练输入数据。
6.一种文本预测方法,其特征在于,包括:
获取待预测文本,利用所述待预测文本和预设文本模板生成待预测输入数据;
将所述待预测输入数据输入文本特征提取模型,得到输入特征;所述文本特征提取模型根据如权利要求1至5任一项所述的模型训练方法生成;
将所述输入特征输入预训练模型,得到预测结果;所述预训练模型预先利用无标注文本训练得到。
7.一种模型训练装置,其特征在于,包括:
训练获取模块,用于获取训练文本,并利用所述训练文本和预设文本模板生成训练输入数据;
训练输入模块,用于将所述训练输入数据输入copynet模型,得到训练输出数据;
训练预测模块,用于将所述训练输出数据输入预训练模型,得到训练预测值;
损失计算模块,用于利用所述训练预测值与所述训练文本的训练标签计算损失值,并利用所述损失值调节所述copynet模型的参数;
模型确定模块,用于若检测到满足训练完成条件,则将参数调节后的copynet模型确定为文本特征提取模型。
8.一种文本预测装置,其特征在于,包括:
获取模块,用于获取待预测文本,利用所述待预测文本和预设文本模板生成待预测输入数据;
特征提取模块,用于将所述待预测输入数据输入文本特征提取模型,得到输入特征;所述文本特征提取模型根据如权利要求1至5任一项所述的模型训练方法生成;
预测模块,用于将所述输入特征输入预训练模型,得到预测结果;所述预训练模型预先利用无标注文本训练得到。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111479311.7/1.html,转载请声明来源钻瓜专利网。