[发明专利]一种多层次分类模型训练方法、多层次分类方法和装置有效
申请号: | 201911366785.3 | 申请日: | 2019-12-26 |
公开(公告)号: | CN111177385B | 公开(公告)日: | 2023-04-07 |
发明(设计)人: | 孙靖文;王道广;伯仲璞;于政 | 申请(专利权)人: | 北京明略软件系统有限公司 |
主分类号: | G06F16/35 | 分类号: | G06F16/35;G06F18/214;G06N3/02 |
代理公司: | 北京安信方达知识产权代理有限公司 11262 | 代理人: | 郑旭丽;栗若木 |
地址: | 100084 北京市海淀区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 多层次 分类 模型 训练 方法 装置 | ||
1.一种多层次分类模型训练方法,包括:
分别将每一个第一文本转换成对应的特征矩阵,分别将每一个第一文本对应的一级标签和第二标签转换成对应的特征向量;
分别将每一个第一文本对应的特征矩阵输入到神经网络模型中得到每一个第一文本对应的第X层特征编码,分别将每一个第一文本对应的一级标签对应的特征向量输入到神经网络模型中得到每一个第一文本对应的一级标签对应的第X层特征编码,分别将每一个第一文本对应的二级标签对应的特征向量输入到神经网络模型中得到每一个第一文本对应的二级标签对应的第X层特征编码;其中,X为神经网络模型的层数;
根据所有第一文本对应的第X层特征编码、第一文本对应的一级标签对应的第X层特征编码、第一文本对应的二级标签对应的第X层特征编码计算损失函数值;
根据损失函数值更新神经网络模型中的参数值,当神经网络模型中的参数值的变化量大于或等于预设阈值时,继续执行分别将每一个第一文本对应的特征矩阵输入到更新参数值后的神经网络模型中得到每一个第一文本对应的第X层特征编码,分别将每一个第一文本对应的一级标签对应的特征向量输入到更新参数值后的神经网络模型中得到每一个第一文本对应的一级标签对应的第X层特征编码,分别将每一个第一文本对应的二级标签对应的特征向量输入到更新参数值后的神经网络模型中得到每一个第一文本对应的二级标签对应的第X层特征编码的步骤;
所述神经网络模型中按照公式ai=Wi×ai-1+bi计算所述第一文本对应的第i层特征编码,或所述第一文本对应的一级标签对应的第i层特征编码,或所述第一文本对应的二级标签对应的第i层特征编码;其中,i为一到X之间的任意一个整数;
其中,ai为所述第一文本对应的第i层特征编码,或所述第一文本对应的一级标签对应的第i层特征编码,或所述第一文本对应的二级标签对应的第i层特征编码;
Wi为第i层的权重;
当i大于1时,ai-1为所述第一文本对应的第(i-1)层特征编码,或所述第一文本对应的一级标签对应的第(i-1)层特征编码,或所述第一文本对应的二级标签对应的第(i-1)层特征编码;当i=1时,ai-1为所述第一文本对应的特征矩阵,或所述第一文本对应的一级标签对应的特征向量,或所述第一文本对应的二级标签对应的特征向量;
bi为第i层的偏;
更新的神经网络模型中的参数包括上述Wi和bi;
其中,按照公式更新神经网络模型中的Wi,按照公式更新神经网络模型中的bi;
其中,W'i为更新后的第i层的权重,Wi为更新前的第i层的权重,α为梯度下降中的学习速率,L'为损失函数值,b'i为更新后的第i层的偏,bi为第i层的偏。
2.根据权利要求1所述的多层次分类模型训练方法,其特征在于,当所述神经网络模型中的参数值的变化量小于预设阈值时,该方法还包括:
将所述神经网络模型中更新后的参数值作为所述神经网络模型中的最佳参数值。
3.根据权利要求1或2所述的多层次分类模型训练方法,其特征在于,其中,按照公式计算损失函数值;
其中,L'为所述损失函数值,L1j为第j个第一文本对应的第X层特征编码和第j个第一文本对应的一级标签对应的第X层特征编码之间的相似度,L2j为第j个第一文本对应的经过Dropout和Dense网络层后的第X层特征编码和第j个第一文本对应的一级标签对应的经过Dropout和Dense网络层后的第X层特征编码之间的相似度,L3j为第j个第一文本对应的二级标签对应的第X层特征编码和第j个第一文本对应的一级标签对应的经过Dropout和Dense网络层后的第X层特征编码之间的相似度,L4j为第j个第一文本对应的二级标签对应的第X层特征编码和第j个第一文本对应的经过Dropout和Dense网络层后的第X层特征编码之间的相似度。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京明略软件系统有限公司,未经北京明略软件系统有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201911366785.3/1.html,转载请声明来源钻瓜专利网。