[发明专利]一种基于GPT网络模型的中文问题重写方法在审
申请号: | 202211603685.X | 申请日: | 2022-12-14 |
公开(公告)号: | CN116011425A | 公开(公告)日: | 2023-04-25 |
发明(设计)人: | 张峻崎;曹肖攀;陈先磊;吴磊;赵凯文;赵俊容 | 申请(专利权)人: | 中电万维信息技术有限责任公司 |
主分类号: | G06F40/194 | 分类号: | G06F40/194;G06F40/289;G06F18/214;G06N3/0464 |
代理公司: | 兰州嘉诺知识产权代理事务所(普通合伙) 62202 | 代理人: | 郭海 |
地址: | 730000 甘肃省兰州市城关*** | 国省代码: | 甘肃;62 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 基于 gpt 网络 模型 中文 问题 重写 方法 | ||
1.一种基于GPT网络模型的中文问题重写方法,其特征在于包括如下步骤:
S1.构造训练数据集
参考github开源项目公开的相似度匹配数据集构造训练数据集,挑选出中文文本数据集中两个文本相似的文本组合;中文问题匹配对放在一个train.txt文件中,每一行放置匹配的两个中文问题,中间用’sep’分开,对数据进行删除过滤操作,设定文本最大长度max_len为72,即删除总长度大于72的文本,同时交换文本匹配的两条数据顺序进行扩增两条训练样本,经过整理得到中文问题匹配训练数据集;
S2.建立词表
将步骤S1中得到的中文问题匹配训练数据所有字符去重后,建立词表序列,记为Dict,Dict的前项key为字符索引编号,Dict的后项value为具体的单个字符, Dict为{字符索引编号0:‘[CLS]’,1:’[PAD]’,2:’[SEP]’,.......},其中[CLS]为文本起始符,[PAD]代表当文本长度不够最大长度时,采用[PAD]填充到max_len长度,[SEP]为中文问题和该中文问题重写后的问题的分割符,max_len为步骤S1中所提及的每条文本数据最大长度;
S3.数据和模型适配
假如现在在步骤S1中得到的训练数据集中的某条文本样本记为列表A,本条文本不够最大长度72,则将列表A通过[PAD]填充到该文本最大长度72,然后通过字典Dict,映射成索引编号列表,进而变成输入GPT模型Tensor张量X,对应的Y输出为缺少起始符通过索引编号列表,变成Tensor张量,作为模型输出的优化目标Y,将训练数据集中的其它文本执行同样操作,得到整个输入数据和优化目标,作为后续模型的输入和优化目标;
S4.构建GPT网络模型结构
模型约定文本最大长度为72,嵌入维度为256,针对在步骤S3中模型输入的每条输入,先经过token_embedding、postional_embedding将文本进行嵌入表示,此时文本嵌入表示的矩阵形状为[72,256],将该文本嵌入表示输入到GPT网络模型,即:具有遮掩mask的多头注意力机制Multi_Head_Attention、前馈全连接神经网络Feed_forward作为一个单元,重复6次,即6层相同的网络,得到整体网络模型结构;
S5.训练模型
根据步骤S4中已设定好的模型结构,并加入模型优化器和损失函数进行模型训练;具体训练过程为将步骤S1、S2和S3得到的模型输入数据,喂入步骤S4中得到的整体网络模型结构,模型最终目标即最小化损失函数,先根据网络结构进行前向传播,根据预测输出与真实输出计算损失,然后进行反向传播,通过Adam优化器不断更新网络模型参数,以此往复运行,直到损失函数值降低到最小并趋于稳定;将此时的参数矩阵进行保存,即可得到最优网络模型;
S6.模型预测
在模型预测阶段,模型参数随机失活比率dropout为0.1,解码阶段采取在模型预测的每个位置所属字的可能性按大小排序,仅保留最前面两个最大的标记,然后在该两个标记中保留最大的,在文本生成阶段,也即采样出可能性最高的两个中一个,然后通过词典获取该索引,该索引即为Dict的键Key值,然后根据Key值得到对应的字, '[SEP]'记为List_input,其长度为len_List,然后通过’[PAD]’填充到最大长度72,再通过步骤S2中提及的字典Dict,映射成索引编号张量,将其作为模型输入,使用步骤S5中得到的网络模型,进入步骤S4中的网络结构,进行前向传播,得到logits值,再使用softmax计算模型输出的第len_List个位置可能性最高的字符,然后随机采样,然后再次通过’[PAD]’填充到最大长度max_len,再通过步骤S2中提及的字典Dict,映射成索引编号张量,将其作为模型输入喂入步骤S5中得到的最优网络模型,此时计算模型输出的第len_List、1个位置可能性最高的字符采样,如此反复操作依次输出对应的改写后的中文问题,最后将所有预测的[SEP]后的字符拼接在一起,作为文本预测最终输出结果;
S7.模型应用
当前数据库中有问题Qa和答案Answer为问题和对应答案匹配对,通过将Qa多次输入GPT中文问题重写模型,则生成Qa的多种表示。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中电万维信息技术有限责任公司,未经中电万维信息技术有限责任公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202211603685.X/1.html,转载请声明来源钻瓜专利网。