[发明专利]指代消解模型的训练方法及装置、电子设备

权利要求书

1.一种指代消解模型的训练方法，其特征在于，所述方法包括：

在预设语料池中筛选符合目标条件的语料；其中，符合所述目标条件的语料中至少存在第一候选名词、第二候选名词以及目标名词，所述目标名词在所述语料中位于所述第一候选名词以及所述第二候选名词之后，所述第一候选名词与所述目标名词相同，所述第二候选名词与所述目标名词不同；

将所述符合目标条件的语料中的所述目标名词替换为预设标识，所述第一候选名词替换为第一预设名词，所述第二候选名词替换为第二预设名词，得到目标语料，其中，所述第一预设名词和所述第二预设名词为包含预设数量名词的目标词库中两个不同的名词；

根据所述第一预设名词以及所述第二预设名词，生成对应所述目标语料的标注信息；

根据所述目标语料以及所述对应目标语料的标注信息，对指代消解模型进行训练。

2.根据权利要求1所述的方法，其特征在于，将所述符合目标条件的语料中的所述目标名词替换为预设标识，所述第一候选名词替换为第一预设名词，所述第二候选名词替换为第二预设名词，得到目标语料，包括：

分别针对每条符合目标条件的语料，在所述目标词库中随机选择一个名词作为第一预设名词，并再次随机选择一个与所述第一预设名词不同的名词作为第二预设名词；

分别针对每条符合目标条件的语料，采用第一预设名词对所述第一候选名词进行替换、采用第二预设名词对所述第二候选名词进行替换以及采用预设标识对所述目标名词进行替换，生成所述目标语料。

3.根据权利要求2所述的方法，其特征在于，在所述目标词库中的名词、第一候选名词和第二候选名词均为人名的情况下，所述在所述目标词库中随机选择一个名词作为第一预设名词，并再次随机选择一个与所述第一预设名词不同的名词作为第二预设名词，包括：

根据所述符合目标条件的语料所属的文学作品，确定所述第一候选名词对应的性别为第一性别，所述第二候选名词对应的性别为第二性别；

在所述目标词库的对应第一性别的名词中随机选择一个名词作为第一预设名词；

在所述目标词库的对应第二性别的名词中随机选择一个与所述第一预设名词不同的名词作为第二预设名词。

4.根据权利要求1所述的方法，其特征在于，所述在预设语料池中筛选符合目标条件的语料，包括：

基于命名实体识别确定所述预设语料池中每条语料包含的名词；

根据每条语料包含的名词，筛选至少存在所述第一候选名词、所述第二候选名词以及所述目标名词的语料。

5.根据权利要求1所述的方法，其特征在于，所述根据所述第一预设名词以及所述第二预设名词，生成对应所述目标语料的标注信息，包括：

将所述第一预设名词以及所述第二预设名词组成候选名词集合；

记录所述预设标识指代所述候选名词集合中的第一预设名词的目标信息。

6.根据权利要求5所述的方法，其特征在于，所述根据所述目标语料以及所述对应目标语料的标注信息，对指代消解模型进行训练，包括：

对所述目标语料中的所述第一预设名词添加第一标注标识，并对所述第二预设名词添加第二标注标识，得到中间语料；

将所述中间语料输入所述指代消解模型，根据所述第一标注标识以及所述第二标注标识，生成指示所述第一预设名词的第一语义向量、指示第二预设名词的第二语义向量以及目标语义向量，其中，所述目标语义向量为根据所述中间语料中预设标识的上下文信息确定的语义向量；

根据所述第一语义向量、所述第二语义向量分别与所述目标语义向量的相似度，确定预测结果，其中，所述预测结果包括所述预设标识指代所述第一预设名词或者所述预设标识指代所述第二预设名词；

根据所述预测结果以及所述目标信息，调整所述指代消解模型中对应的参数。