可左右滑动选省市

模型训练方法、模型训练装置及终端设备实用新型专利

更新时间:2024-10-01
模型训练方法、模型训练装置及终端设备实用新型专利 专利申请类型:实用新型专利;
地区:广东-深圳;
源自:深圳高价值专利检索信息库;

专利名称:模型训练方法、模型训练装置及终端设备

专利类型:实用新型专利

专利申请号:CN202011504757.6

专利申请(专利权)人:中国平安人寿保险股份有限公司
权利人地址:广东省深圳市福田区福华三路星河发展中心办公9、10、11层

专利发明(设计)人:梁宸,马龙,周元笙,蒋佳惟,陈思姣,李炫

专利摘要:本申请适用于人工智能技术领域,提供了模型训练方法、模型训练装置、终端设备及存储介质,其中,一种模型训练方法,包括:对第一待训练语句进行掩码处理;根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型;根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型;基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型。通过上述方法,可以获得准确性较高的文本分类模型。

主权利要求:
1.一种模型训练方法,其特征在于,包括:
对第一待训练语句进行掩码处理,其中,所述掩码处理指将第一待训练语句中的指定文字进行遮盖,以将所述指定文字作为待预测文字,若所述指定文字涉及第一产品名,则将涉及的每一个第一产品名作为一组待预测文字;
根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型,其中,第一训练中,所述文本分类模型用于对进行掩码处理后的第一待训练语句中的待预测文字进行预测;
根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型,其中,各个所述第二待训练语句中分别包括至少一个第一产品名,并且,每个所述第二待训练语句对应有业务标签,第二训练中,所述文本分类模型用于确定所述第二待训练语句所对应的业务类别;各个所述第一产品名和所述业务标签为根据产品图谱而得到,所述产品图谱中的每一个节点包括至少一个第一产品名,一个所述节点中的第一产品名对应同一个产品,并且,每一个节点所对应的属性信息包含所述节点所对应的产品的业务信息;
基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,其中,所述待预测语句与所述第一待训练语句以及所述第二待训练语句不同,第三训练中,基于所述文本分类模型中的至少两个指定层的输出确定所述第三训练是否完成;
所述基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,包括:在所述第三训练的每一次迭代过程中,针对一个待预测语句,将所述待预测语句输入第二训练完成的预设模型,获得第二训练完成的预设模型的至少两个指定层针对所述待预测语句所分别输出的第一词向量;
针对第二训练完成的预设模型所输出的关于所述待预测语句的输出向量,获取与所述输出向量相关联的第二词向量,其中,所述输出向量所指示的业务类别与所述第二词向量所指示的业务类别相同;
根据所述待预测语句中除所述第一词向量相关联的文字以外的其他部分,获得第三词向量;
对至少两个所述第一词向量进行池化处理,获得第一向量;
对所述第二词向量进行池化处理,获得第二向量;
对所述第三词向量进行池化处理,获得第三向量;
根据所述第一向量、第二向量、第三向量和预设分类网络,进行业务分类;
根据业务分类结果,确定所述第三训练是否完成;
在对第一待训练语句进行掩码处理之前,还包括:
从预设语料中获取待识别产品名;
针对每一个待识别产品名,确定所述待识别产品名分别相对于各个标准产品名的相似度信息;
根据各个所述相似度信息,生成产品名关联列表,所述产品名关联列表中包括至少两个第一产品名和各个第一产品名之间的关联关系,任一所述第一产品名为标准产品名或者待识别产品名;
根据预设业务信息和所述产品名关联列表,生成产品图谱;
在获得第三训练完成的文本分类模型之后,还包括:
通过所述第三训练完成的文本分类模型,对待检测文本进行文本分类,以确定所述待检测文本的业务类别。
2.如权利要求1所述的模型训练方法,其特征在于,所述针对每一个待识别产品名,确定所述待识别产品名分别相对于各个标准产品名的相似度信息,包括:将所述待识别产品名与预设的标准产品名输入第一自然语言处理模型,获得所述第一自然语言处理模型的输出结果,所述输出结果包括每一个所述待识别产品名分别相对于各个标准产品名的相似度信息。
3.一种模型训练装置,其特征在于,包括:
掩码处理模块,用于对第一待训练语句进行掩码处理,其中,所述掩码处理指将第一待训练语句中的指定文字进行遮盖,以将所述指定文字作为待预测文字,若所述指定文字涉及第一产品名,则将涉及的每一个第一产品名作为一组待预测文字;
第一训练模块,用于根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型,其中,第一训练中,所述文本分类模型用于对进行掩码处理后的第一待训练语句中的待预测文字进行预测;
第二训练模块,用于根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型,其中,各个所述第二待训练语句中分别包括至少一个第一产品名,并且,每个所述第二待训练语句对应有业务标签,第二训练中,所述文本分类模型用于确定所述第二待训练语句所对应的业务类别;各个所述第一产品名和所述业务标签为根据产品图谱而得到,所述产品图谱中的每一个节点包括至少一个第一产品名,一个所述节点中的第一产品名对应同一个产品,并且,每一个节点所对应的属性信息包含所述节点所对应的产品的业务信息;
第三训练模块,用于基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,其中,所述待预测语句与所述第一待训练语句以及所述第二待训练语句不同,第三训练中,基于所述文本分类模型中的至少两个指定层的输出确定所述第三训练是否完成;
训练单元,用于在所述第三训练的每一次迭代过程中,针对一个待预测语句,将所述待预测语句输入第二训练完成的预设模型,获得第二训练完成的预设模型的至少两个指定层针对所述待预测语句所分别输出的第一词向量;
第一处理单元,用于针对第二训练完成的预设模型所输出的关于所述待预测语句的输出向量,获取与所述输出向量相关联的第二词向量,其中,所述输出向量所指示的业务类别与所述第二词向量所指示的业务类别相同;
第二处理单元,用于根据所述待预测语句中除所述第一词向量相关联的文字以外的其他部分,获得第三词向量;
第一处理子单元,用于对至少两个所述第一词向量进行池化处理,获得第一向量;
第二处理子单元,用于对所述第二词向量进行池化处理,获得第二向量;
第三处理子单元,用于对所述第三词向量进行池化处理,获得第三向量;
分类子单元,用于根据所述第一向量、第二向量、第三向量和预设分类网络,进行业务分类;
第三处理单元,用于根据业务分类结果,确定所述第三训练是否完成;
模型训练装置还包括:
获取模块,用于从预设语料中获取待识别产品名;
确定模块,用于针对每一个待识别产品名,确定所述待识别产品名分别相对于各个标准产品名的相似度信息;
第一生成模块,用于根据各个所述相似度信息,生成产品名关联列表,所述产品名关联列表中包括至少两个第一产品名和各个第一产品名之间的关联关系,任一所述第一产品名为标准产品名或者待识别产品名;
第二生成模块,用于根据预设业务信息和所述产品名关联列表,生成产品图谱;
该模型训练装置还包括:
文本分类模块,用于通过所述第三训练完成的文本分类模型,对待检测文本进行文本分类,以确定所述待检测文本的业务类别。
4.一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1至2任一项所述的模型训练方法。
5.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至2任一项所述的模型训练方法。 说明书 : 模型训练方法、模型训练装置及终端设备技术领域[0001] 本申请涉及人工智能技术领域,尤其涉及模型训练方法、模型训练装置、终端设备及计算机可读存储介质。背景技术[0002] 文本分类是自然语言处理中一项十分基础且重要的工作。它的应用也非常广泛,比如:垃圾信息过滤,新闻分类,词性标注等等。[0003] 目前,用于常用的机器学习相关的文本分类方法有基于逻辑回归、支持向量机的文本分类方法等。而这些机器学习算法在根据输入的文本进行特征提取时,提取的文本点信息都较为独立,不能很好地利用上下文信息,从而导致文本分类的准确性较低。发明内容[0004] 有鉴于此,本申请实施例提供了模型训练方法、模型训练装置、终端设备及计算机可读存储介质,可以获得准确性较高的文本分类模型。[0005] 第一方面,本申请实施例提供了一种模型训练方法,包括:[0006] 对第一待训练语句进行掩码处理,其中,所述掩码处理指将第一待训练语句中的指定文字进行遮盖,以将所述指定文字作为待预测文字,若所述指定文字涉及第一产品名,则将涉及的每一个第一产品名作为一组待预测文字;[0007] 根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型,其中,第一训练中,所述文本分类模型用于对进行掩码处理后的第一待训练语句中的待预测文字进行预测;[0008] 根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型,其中,各个所述第二待训练语句中分别包括至少一个第一产品名,并且,每个所述第二训练语句对应有业务标签,第二训练中,所述文本分类模型用于确定所述第二待训练语句所对应的业务类别;[0009] 基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,其中,所述待预测语句与所述第一待训练语句以及所述第二训练语句不同,第三训练中,基于所述文本分类模型中的至少两个指定层的输出确定所述第三训练是否完成。[0010] 第二方面,本申请实施例提供了一种模型训练装置,包括:[0011] 掩码处理模块,用于对第一待训练语句进行掩码处理,其中,所述掩码处理指将第一待训练语句中的指定文字进行遮盖,以将所述指定文字作为待预测文字,若所述指定文字涉及第一产品名,则将涉及的每一个第一产品名作为一组待预测文字;[0012] 第一训练模块,用于根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型,其中,第一训练中,所述文本分类模型用于对进行掩码处理后的第一待训练语句中的待预测文字进行预测;[0013] 第二训练模块,用于根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型,其中,各个所述第二待训练语句中分别包括至少一个第一产品名,并且,每个所述第二训练语句对应有业务标签,第二训练中,所述文本分类模型用于确定所述第二待训练语句所对应的业务类别;[0014] 第三训练模块,用于基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,其中,所述待预测语句与所述第一待训练语句以及所述第二训练语句不同,第三训练中,基于所述文本分类模型中的至少两个指定层的输出确定所述第三训练是否完成。[0015] 第三方面,本申请实施例提供了一种终端设备,包括存储器、处理器、显示器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如第一方面所述的模型训练方法。[0016] 第四方面,本申请实施例提供了一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现如第一方面所述的模型训练方法。[0017] 实施本申请实施例提供的模型训练方法具有以下有益效果:本申请实施例中,可以将产品名等业务术语通过掩码处理后的第一待训练语句通过第一训练融入进文本分类模型中,获得第一训练完成的文本分类模型;然后,再通过第二待训练语句的业务标签和第一产品名,将业务类别和产品名等业务特征融入到第一训练完成的文本分类模型中,获得第二训练完成的文本分类模型;接着,可以通过第三训练和待预测语句,第二训练完成的文本分类模型进行精调,从而获得第三训练完成的预设模型,此时,所述第三训练完成的预设模型能够较为准确地识别文本中的业务信息。可见,通过本申请实施例,能够通过多层次的模型训练,获得文本分类准确率较高的第三训练完成的文本分类模型,以在后续能够通过该第三训练完成的文本分类模型,较为准确地检测文本类别。附图说明[0018] 为了更清楚地说明本申请实施例中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。[0019] 图1是本申请一实施例提供的一种模型训练方法的流程示意图;[0020] 图2是本申请一实施例提供的步骤S104的一种流程示意图;[0021] 图3是本申请一实施例提供的一种模型训练装置的结构示意图;[0022] 图4是本申请一实施例提供的终端设备的结构示意图。具体实施方式[0023] 为了使本申请的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本申请进行进一步详细说明。应当理解,此处所描述的具体实施例仅仅用以解释本申请,并不用于限定本申请。[0024] 本申请实施例提供的模型训练方法可以应用于服务器、台式电脑、手机、平板电脑、可穿戴设备、车载设备、增强现实(augmentedreality,AR)/虚拟现实(virtualreality,VR)设备、笔记本电脑、超级移动个人计算机(ultra‑mobilepersonalcomputer,UMPC)、上网本、个人数字助理(personaldigitalassistant,PDA)等终端设备上,本申请实施例对终端设备的具体类型不作任何限制。[0025] 请参阅图1,图1示出了本申请实施例提供的一种模型训练方法的流程图。[0026] 如图1所示,本实施例提供的一种模型训练方法可以包括:[0027] 步骤S101,对第一待训练语句进行掩码处理,其中,所述掩码处理指将第一待训练语句中的指定文字进行遮盖,以将所述指定文字作为待预测文字,若所述指定文字涉及第一产品名,则将涉及的每一个第一产品名作为一组待预测文字。[0028] 本申请实施例中,所述掩码处理可以随机地确定指定文字的位置和包含的文字个数,当然,可以根据其他预设的规则确定指定文字的位置和包含的文字个数等。若任一所述掩码处理中涉及第一产品名,则该掩码处理即对该第一产品名打上掩码。[0029] 例如,“请问产品A适合有小孩的家庭”,会对“产品A”打上连续的掩码进行遮盖以用于预测,而不会仅仅对“A”打上掩码。[0030] 步骤S102,根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型,其中,第一训练中,所述文本分类模型用于对进行掩码处理后的第一待训练语句中的待预测文字进行预测。[0031] 所述文本分类模型的具体结构在此不作限定。示例性的,所述文本分类模型可以为分层注意网络(HierarchicalAttentionNetwork)、动态记忆网络(DynamicMemoryNetwork)、文本卷积神经网络(TextConvolutionalNeuralNetwork)或者12层transformer模型等等。[0032] 其中,所述第一训练完成的判断方式可以有多种。例如,可以是第一训练的迭代次数达到预设次数,也可以是第一训练的某一次迭代过程中,所述文本分类模型对第一待训练语句的待预测文字的预测精度达到第一精度阈值等等。所述第一训练中的损失函数等在此也不做限定。示例性的,所述第一训练中的损失函数可以为交叉熵函数、均方差函数等等。[0033] 本申请实施例中,可以将产品名等业务术语通过掩码处理后的第一待训练语句通过第一训练融入进文本分类模型中,获得第一训练完成的文本分类模型。[0034] 步骤S103,根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型,其中,各个所述第二待训练语句中分别包括至少一个第一产品名,并且,每个所述第二训练语句对应有业务标签,第二训练中,所述文本分类模型用于确定所述第二待训练语句所对应的业务类别。[0035] 第二训练过程可以用于区分业务用语(即产品名)在文本分类模型中的融入效果,使得[0036] 示例性的,所述业务标签可以包括业务类别标签、业务对象标签、业务内容标签等等中的至少一种。所述业务标签可以用于判断所述第二训练是否完成,也可以用于将业务信息通过第二训练的方式融入第二训练完成的文本分类模型中,使得第二训练完成的文本分类模型能够识别业务标签中的信息,从而使得第一产品名、业务术语、业务内容以及指向的业务类别等业务信息在第二训练完成的文本分类模型中得到突显。[0037] 其中,所述第二训练完成的判断方式可以有多种。例如,可以是第二训练的迭代次数达到预设次数,也可以是第二训练的某一次迭代过程中,所述文本分类模型对第二待训练语句的待预测文字的预测精度达到第二精度阈值等等。所述第二训练中的损失函数等在此也不做限定。示例性的,所述第二训练中的损失函数可以为交叉熵函数、均方差函数等等。[0038] 步骤S104,基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,其中,所述待预测语句与所述第一待训练语句以及所述第二训练语句不同,第三训练中,基于所述文本分类模型中的至少两个指定层的输出确定所述第三训练是否完成。[0039] 本申请实施例中,所述待预测语句可以与所述第一待训练语句以及第二待训练语句不同。此时,所述待预测语句可以认为是实际应用场景中的具体应用数据,因此,在针对具体业务,可以通过所述第三训练过程和所述待预测语句,将第二训练完成的文本分类模型迁移到新数据(即待预测语句)中。因此,所述待预测语句可以针对不同的应用场景而变化。[0040] 可见,所述第三训练可以用于针对业务场景的迁移学习。也即是说,在获得第二训练完成的文本分类模型之后,针对与第一待训练语句和第二待训练语句存在不同的业务场景,可以获取包含该业务场景的待预测语句,并通过包含该业务场景的待预测语句对所述第二训练完成的文本分类模型进行第三训练,以对所述第二训练完成的文本分类模型进行有针对性地精调,获得第三训练完成的文本分类模型。此时,第三训练完成的文本分类模型可以较好地识别待预测语句所对应的业务场景下的文本类别,同时,在第二训练完成的文本分类模型的基础上进行第三训练时,训练难度大大减小,训练效率较高。[0041] 在一些实施例中,所述基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,包括:[0042] 步骤S201,在所述第三训练的每一次迭代过程中,针对一个待预测语句,将所述待预测语句输入第二训练完成的预设模型,获得第二训练完成的预设模型的至少两个指定层针对所述待预测语句所分别输出的第一词向量;[0043] 步骤S202,针对第二训练完成的预设模型所输出的关于所述待预测语句的输出向量,获取与所述输出向量相关联的第二词向量,其中,所述输出向量所指示的业务类别与所述第二词向量所指示的业务类别相同;[0044] 步骤S203,根据所述待预测语句中除所述第一词向量相关联的文字以外的其他部分,获得第三词向量;[0045] 步骤S204,根据所述第一词向量、第二词向量、第三词向量和预设分类网络,进行业务分类;[0046] 步骤S205,根据业务分类结果,确定所述第三训练是否完成。[0047] 本申请实施例中,所述第二训练完成的文本分类模型的至少两个指定层的类型、个数以及结构形式在此不做限制。示例性的,所述指定层可以包括特征提取层、激活层和/或全连接层等等。[0048] 在一些实施例中,所述针对第二训练完成的预设模型所输出的关于所述待预测语句的输出向量,获取与所述输出向量相关联的第二词向量可以包括:[0049] 针对第二训练完成的预设模型所输出的关于所述待预测语句的输出向量,基于产品图谱获取与所述输出向量相关联的第二词向量。[0050] 其中,所述产品图谱中的每一个节点包括至少一个第一产品名,一个所述节点中的第一产品名对应同一个产品,并且,每一个节点所对应的属性信息包含所述节点所对应的产品的业务信息。因此,可以通过所述产品图谱,获取到与所述输出向量所指示的业务类别相关联的第一产品名,再通过诸如第二训练完成的文本分类模型等,获得与所述输出向量所指示的业务类别相关联的第一产品名的词向量作为所述第二词向量。[0051] 所述第一词向量可以认为是表征待预测语句中的第一产品名的特征向量,第二训练完成的预设模型所输出的关于所述待预测语句的输出向量可以用于指示所述待预测语句的业务类别,而所述第二词向量可以认为是与所述输出向量所指示的业务类别相关联的第一产品名的词向量。所述第三词向量可以包含所述待预测语句中的第一产品名的上下文信息。[0052] 此时,通过所述预设分类网络可以整合所述第一词向量、第二词向量和第三词向量的信息,从而根据多维度的信息进行业务分类。示例性的,所述预设分类网络可以包括全连接层。[0053] 在一些实施例中,所述根据所述第一词向量、第二词向量、第三词向量和预设分类网络,进行业务分类,包括:[0054] 对至少两个所述第一词向量进行池化处理,获得第一向量;[0055] 对所述第二词向量进行池化处理,获得第二向量;[0056] 对所述第三词向量进行池化处理,获得第三向量;[0057] 根据所述第一向量、第二向量、第三向量和预设分类网络,进行业务分类。[0058] 示例性的,所述池化处理可以为最大池化处理。所述第二训练完成的文本分类模型的至少两个指定层的类型、个数以及结构形式在此不做限制。[0059] 示例性的,所述第二训练完成的文本分类模型的至少两个指定层可以为所述第二训练完成的文本分类模型的最后四层。该最后四层针对所述待预测语句所分别输出的第一词向量可以称为CLS(classification)向量;[0060] 该第一词向量经过最大池化处理后形成一个融合四层CLS向量的第一向量,记为h(CLS’)。[0061] 针对第二训练完成的文本分类模型基于所述待预测语句的输出向量,基于所述产品图谱查找与所述输出向量相关联的第二词向量,该第二词向量经过最大池化处理后获得一个与CLS向量维度相同的第二向量记为h(A)。[0062] 根据所述待预测语句中除所述第一词向量相关联的文字以外的其他部分,获得第三词向量;[0063] 该第二词向量经过最大池化处理后形成一个与CLS向量维度相同的第三向量记为h(B);[0064] h(CLS’)、h(A)和h(B)经过预设分类网络,可以进行分类操作,以进行业务分类,获得业务分类结果,从而根据该业务分类结果和预设损失函数(如交叉熵函数),确定是否训练完成。[0065] 在一些实施例中,各个所述第一产品名和所述业务标签为根据产品图谱而得到,所述产品图谱中的每一个节点包括至少一个第一产品名,一个所述节点中的第一产品名对应同一个产品,并且,每一个节点所对应的属性信息包含所述节点所对应的产品的业务信息。[0066] 在一些实施例中,在对第一待训练语句进行掩码处理之前,还包括:[0067] 从预设语料中获取待识别产品名;[0068] 针对每一个待识别产品名,确定所述待识别产品名分别相对于各个标准产品名的相似度信息;[0069] 根据各个所述相似度信息,生成产品名关联列表,所述产品名关联列表中包括至少两个第一产品名和各个第一产品名之间的关联关系,任一所述第一产品名为标准产品名或者待识别产品名;[0070] 根据预设业务信息和所述产品名关联列表,生成产品图谱。[0071] 其中,所述语料可以为特定业务领域的文本资料。从所述预设语料中获取待识别产品名的方式可以有多种。例如,可以是通过人工提取得到,也可以是通过预设的关键词提取算法等等方式从所述预设语料的标题、预设产品列表、文件描述部分内容等位置提取得到。[0072] 所述标准产品名可以从预设的标准产品名列表中提取得到。示例性的,所述标准产品名列表可以是从特定业务领域的标准产品名录等文件中提取得到。[0073] 例如,对于保险业务,所述标准产品名可以从目前在售的保险条目文件等等。[0074] 所述待识别产品名可以认为是业务预料中可能存在的对所述标准产品名的别称。针对每一个待识别产品名,若所述待识别产品名与任一标准产品名的相似度大于预设相似度阈值,则可以认为所述待识别产品名为该标准产品名的相似说法,也即是说,可以认为该待识别产品名与所述标准产品名相关联,并指向同一产品。[0075] 可见,通过生成产品名关联列表,从而可以确定哪一些产品名与同一个业务相关联,以为后续的业务分类提供分类基础。[0076] 所述业务信息可以从预设的业务文件中提取得到,所述业务信息可以标识对应的产品的定义、业务范围、目标客户、业务内容以及业务类别等等中的至少一种。其中,目标客户还可以包括目标客户年龄、职业等客户群体信息。[0077] 需要说明的是,所述属性信息的具体获取形式可以有多种。例如,可以通过人工获取,也可以通过关键词匹配的方式提取得到,或者,还可以通过特定的深度学习模型获取到。[0078] 在一些实施例中,示例性的,所述属性信息可以通过预先训练的BERT+CRF模型从预设的业务文件中提取得到。[0079] 其中,所述条件随机场(ConditionalRandomFields,CRF)模型是一种基于遵循马尔可夫性的概率图模型。所述BERT模型是一个预训练语言表示模型(pre‑trainedlanguagerepresentationsmodel),全称为BidirectionalEncoderRepresentationsfromTransformers。[0080] BERT+CRF模型为命名实体识别模型,命名实体识别(NamedEntityRecognition,简称NER)模型可以在信息提取、问答系统、句法分析、机器翻译等应用领域中进行应用。[0081] 对所述BERT+CRF模型的预训练可以通过预设产品名和对应的预设属性信息共同构成的预设语句进行初步训练,再通过业务文件中的文本对初步训练后的BERT+CRF模型进行精调,然后通过精调后的BERT+CRF模型,从所述业务文件中,提取各个产品名所分别对应的属性信息。[0082] 在一些实施例中,所述针对每一个待识别产品名,确定所述待识别产品名分别相对于各个标准产品名的相似度信息,包括:[0083] 将所述待识别产品名与预设的标准产品名输入第一自然语言处理模型,获得所述第一自然语言处理模型的输出结果,所述输出结果包括每一个所述待识别产品名分别相对于各个标准产品名的相似度信息。[0084] 所述第一自然语言处理模型在此不作限制。示例性的,所述第一自然语言处理模型可以为BERT模型。所述BERT模型是一个预训练语言表示模型(pre‑trainedlanguagerepresentationsmodel),全称为BidirectionalEncoderRepresentationsfromTransformers。[0085] 通过所述BERT模型,可以计算每一个所述待识别产品名分别相对于每一个标准产品名之间的相似度信息。[0086] 在一些实施例中,在获得第三训练完成的文本分类模型之后,还包括:[0087] 通过所述第三训练完成的文本分类模型,对待检测文本进行文本分类,以确定所述待检测文本的业务类别。[0088] 本申请实施例中,所述第三训练完成的文本分类模型能够对特定业务的业务术语进行识别,从而可以根据识别得到的业务数据信息,对涉及特定业务场景的待检测文本进行较为准确的文本分类,获得较为准确的业务类别。[0089] 本申请实施例中,可以将产品名等业务术语通过掩码处理后的第一待训练语句通过第一训练融入进文本分类模型中,获得第一训练完成的文本分类模型;然后,再通过第二待训练语句的业务标签和第一产品名,将业务类别和产品名等业务特征融入到第一训练完成的文本分类模型中,获得第二训练完成的文本分类模型;接着,可以通过第三训练和待预测语句,第二训练完成的文本分类模型进行精调,从而获得第三训练完成的预设模型,此时,所述第三训练完成的预设模型能够较为准确地识别文本中的业务信息。可见,通过本申请实施例,能够通过多层次的模型训练,获得文本分类准确率较高的第三训练完成的文本分类模型,以在后续能够通过该第三训练完成的文本分类模型,较为准确地检测文本类别。[0090] 请参阅图3,图3是本申请实施例提供的一种模型训练装置的结构框图。本实施例中该终端设备包括的各单元用于执行上述各个模型训练方法实施例中的各步骤。具体请参阅上述模型训练方法所对应的实施例中的相关描述。为了便于说明,仅示出了与本实施例相关的部分。[0091] 参照图3,该模型训练装置3包括:[0092] 掩码处理模块301,用于对第一待训练语句进行掩码处理,其中,所述掩码处理指将第一待训练语句中的指定文字进行遮盖,以将所述指定文字作为待预测文字,若所述指定文字涉及第一产品名,则将涉及的每一个第一产品名作为一组待预测文字;[0093] 第一训练模块302,用于根据进行掩码处理后的第一待训练语句对文本分类模型进行第一训练,获得第一训练完成的文本分类模型,其中,第一训练中,所述文本分类模型用于对进行掩码处理后的第一待训练语句中的待预测文字进行预测;[0094] 第二训练模块303,用于根据第二待训练语句对所述第一训练完成的文本分类模型进行第二训练,获得第二训练完成的文本分类模型,其中,各个所述第二待训练语句中分别包括至少一个第一产品名,并且,每个所述第二训练语句对应有业务标签,第二训练中,所述文本分类模型用于确定所述第二待训练语句所对应的业务类别;[0095] 第三训练模块304,用于基于待预测语句,对所述第二训练完成的文本分类模型进行第三训练,获得第三训练完成的文本分类模型,其中,所述待预测语句与所述第一待训练语句以及所述第二训练语句不同,第三训练中,基于所述文本分类模型中的至少两个指定层的输出确定所述第三训练是否完成。[0096] 可选的,所述第三训练模块304包括:[0097] 训练单元,用于在所述第三训练的每一次迭代过程中,针对一个待预测语句,将所述待预测语句输入第二训练完成的预设模型,获得第二训练完成的预设模型的至少两个指定层针对所述待预测语句所分别输出的第一词向量;[0098] 第一处理单元,用于针对第二训练完成的预设模型所输出的关于所述待预测语句的输出向量,获取与所述输出向量相关联的第二词向量,其中,所述输出向量所指示的业务类别与所述第二词向量所指示的业务类别相同;[0099] 第二处理单元,用于根据所述待预测语句中除所述第一词向量相关联的文字以外的其他部分,获得第三词向量;[0100] 分类单元,用于根据所述第一词向量、第二词向量、第三词向量和预设分类网络,进行业务分类;[0101] 第三处理单元,用于根据业务分类结果,确定所述第三训练是否完成。[0102] 可选的,所述分类单元具体包括:[0103] 第一处理子单元,用于对至少两个所述第一词向量进行池化处理,获得第一向量;[0104] 第二处理子单元,用于对所述第二词向量进行池化处理,获得第二向量;[0105] 第三处理子单元,用于对所述第三词向量进行池化处理,获得第三向量;[0106] 分类子单元,用于根据所述第一向量、第二向量、第三向量和预设分类网络,进行业务分类。[0107] 可选的,各个所述第一产品名和所述业务标签为根据产品图谱而得到,所述产品图谱中的每一个节点包括至少一个第一产品名,一个所述节点中的第一产品名对应同一个产品,并且,每一个节点所对应的属性信息包含所述节点所对应的产品的业务信息。[0108] 可选的,该模型训练装置3还包括:[0109] 获取模块,用于从预设语料中获取待识别产品名;[0110] 确定模块,用于针对每一个待识别产品名,确定所述待识别产品名分别相对于各个标准产品名的相似度信息;[0111] 第一生成模块,用于根据各个所述相似度信息,生成产品名关联列表,所述产品名关联列表中包括至少两个第一产品名和各个第一产品名之间的关联关系,任一所述第一产品名为标准产品名或者待识别产品名;[0112] 第二生成模块,用于根据预设业务信息和所述产品名关联列表,生成产品图谱。[0113] 可选的,所述确定模块具体用于:[0114] 将所述待识别产品名与预设的标准产品名输入第一自然语言处理模型,获得所述第一自然语言处理模型的输出结果,所述输出结果包括每一个所述待识别产品名分别相对于各个标准产品名的相似度信息。[0115] 可选的,该模型训练装置3还包括:[0116] 文本分类模块,用于通过所述第三训练完成的文本分类模型,对待检测文本进行文本分类,以确定所述待检测文本的业务类别。[0117] 本申请实施例中,可以将产品名等业务术语通过掩码处理后的第一待训练语句通过第一训练融入进文本分类模型中,获得第一训练完成的文本分类模型;然后,再通过第二待训练语句的业务标签和第一产品名,将业务类别和产品名等业务特征融入到第一训练完成的文本分类模型中,获得第二训练完成的文本分类模型;接着,可以通过第三训练和待预测语句,第二训练完成的文本分类模型进行精调,从而获得第三训练完成的预设模型,此时,所述第三训练完成的预设模型能够较为准确地识别文本中的业务信息。可见,通过本申请实施例,能够通过多层次的模型训练,获得文本分类准确率较高的第三训练完成的文本分类模型,以在后续能够通过该第三训练完成的文本分类模型,较为准确地检测文本类别。[0118] 需要说明的是,上述装置/单元之间的信息交互、执行过程等内容,由于与本申请方法实施例基于同一构思,其具体功能及带来的技术效果,具体可参见方法实施例部分,此处不再赘述。[0119] 图4是本申请另一实施例提供的一种终端设备的结构框图。如图4所示,该实施例的终端设备4包括:处理器41、存储器42以及存储在所述存储器42中并可在所述处理器41上运行的计算机程序43,例如模型训练方法的程序。处理器41执行所述计算机程序43时实现上述各个模型训练方法各实施例中的步骤,例如图1所示的S101至S104,或者图2所示的S201至S205。或者,所述处理器41执行所述计算机程序43时实现上述图3对应的实施例中各单元的功能,例如,图3所示的单元301至304的功能,具体请参阅图3对应的实施例中的相关描述,此处不赘述。[0120] 示例性的,所述计算机程序43可以被分割成一个或多个单元,所述一个或者多个单元被存储在所述存储器42中,并由所述处理器41执行,以完成本申请。所述一个或多个单元可以是能够完成特定功能的一系列计算机程序指令段,该指令段用于描述所述计算机程序43在所述终端设备40中的执行过程。例如,所述计算机程序43可以被分割成第一获取单元、第一确定单元、第一调整单元、第二调整单元以及执行单元,各单元具体功能如上所述。[0121] 所述终端设备可包括,但不仅限于,处理器41、存储器42。本领域技术人员可以理解,图4仅仅是终端设备4的示例,并不构成对终端设备4的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件,例如所述转台设备还可以包括输入输出设备、网络接入设备、总线等。[0122] 所称处理器41可以是中央处理单元(CentralProcessingUnit,CPU),还可以是其他通用处理器、数字信号处理器(DigitalSignalProcessor,DSP)、专用集成电路(ApplicationSpecificIntegratedCircuit,ASIC)、现成可编程门阵列(Field‑ProgrammableGateArray,FPGA)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。[0123] 所述存储器42可以是所述终端设备4的内部存储单元,例如终端设备4的硬盘或内存。所述存储器42也可以是所述终端设备4的外部存储设备,例如所述终端设备4上配备的插接式硬盘,智能存储卡(SmartMediaCard,SMC),安全数字(SecureDigital,SD)卡,闪存卡(FlashCard)等。进一步地,所述存储器42还可以既包括所述终端设备4的内部存储单元也包括外部存储设备。所述存储器42用于存储所述计算机程序以及所述转台设备所需的其他程序和数据。所述存储器42还可以用于暂时地存储已经输出或者将要输出的数据。[0124] 本申请实施例中,上述处理器40执行上述计算机程序42以实现上述任意各个模型训练方法实施例中的步骤时,可以将产品名等业务术语通过掩码处理后的第一待训练语句通过第一训练融入进文本分类模型中,获得第一训练完成的文本分类模型;然后,再通过第二待训练语句的业务标签和第一产品名,将业务类别和产品名等业务特征融入到第一训练完成的文本分类模型中,获得第二训练完成的文本分类模型;接着,可以通过第三训练和待预测语句,第二训练完成的文本分类模型进行精调,从而获得第三训练完成的预设模型,此时,所述第三训练完成的预设模型能够较为准确地识别文本中的业务信息。可见,通过本申请实施例,能够通过多层次的模型训练,获得文本分类准确率较高的第三训练完成的文本分类模型,以在后续能够通过该第三训练完成的文本分类模型,较为准确地检测文本类别。[0125] 本申请实施例还提供了一种计算机可读存储介质,上述计算机可读存储介质存储有计算机程序,上述计算机程序被处理器执行时实现可实现上述各个模型训练方法实施例中的步骤。[0126] 本申请实施例提供了一种计算机程序产品,当计算机程序产品在终端设备上运行时,使得终端设备执行时实现可实现上述各个模型训练方法实施例中的步骤。[0127] 以上所述实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围,均应包含在本申请的保护范围之内。

专利地区:广东

专利申请日期:2020-12-18

专利公开日期:2024-07-26

专利公告号:CN112579774B


以上信息来自国家知识产权局,如信息有误请联系我方更正!
电话咨询
读内容
搜本页
回顶部