[发明专利]命名实体识别方法、训练方法、装置、设备及介质

权利要求书

1.一种命名实体识别方法，其特征在于，所述方法包括：

获取输入文本，所述输入文本包括至少一个字符；

对所述输入文本中的每个字符进行字符编码，得到每个字符的第一字符编码向量；

基于所述每个字符的第一字符编码向量，预测所述每个字符为候选实体类型的开始字符的第一概率，以及所述每个字符为所述候选实体类型的结束字符的第二概率；

基于所述第一概率和所述第二概率识别出属于所述候选实体类型的命名实体。

2.根据权利要求1所述的方法，其特征在于，所述根据所述字符的第一字符编码向量，预测所述字符为候选实体类型的开始字符的第一概率，以及所述字符为所述候选实体类型的结束字符的第二概率，包括：

调用二元模型中的第一分类层对所述字符的第一字符编码向量进行预测，得到所述字符的第二字符编码向量，以及所述字符为所述候选实体类型的开始字符的第一概率；

对所述字符的第一字符编码向量和所述第二字符编码向量进行级联，得到所述字符的合并字符编码向量；

调用所述二元模型的第二分类层对所述字符的所述合并字符编码向量进行预测，得到所述字符为所述候选实体类型的结束字符的第二概率。

3.根据权利要求2所述的方法，其特征在于，所述第一分类层为第一全连接层，所述第二分类层为第二全连接层；

所述调用二元模型中的第一分类层对所述字符的第一字符编码向量进行预测，得到所述字符的第二字符编码向量，以及所述字符为所述候选实体类型的开始字符的第一概率，包括：

调用所述第一全连接层对所述字符的第一字符编码向量进行预测，得到所述字符的第二字符编码向量，以及所述字符为所述候选实体类型的开始字符的第一概率；

所述调用所述二元模型的第二分类层对所述字符的所述合并字符编码向量进行预测，得到所述字符为所述候选实体类型的结束字符的第二概率，包括：

调用所述第二全连接层对所述字符的所述合并字符编码向量进行预测，得到所述字符为所述候选实体类型的结束字符的第二概率。

4.根据权利要求1至3任一所述的方法，其特征在于，所述对所述输入文本中的每个字符进行字符编码，得到每个字符的字符编码向量，包括：

调用语言模型对所述输入文本中的每个字符进行编码，得到所述每个字符的字符编码向量。

5.根据权利要求4所述的方法，其特征在于，所述调用语言模型对所述输入文本中的每个字符进行编码，得到所述每个字符的字符编码向量，包括：

调用来自变压器的双向编码器表示BERT模型中的编码器对所述输入文本中的每个字符进行编码，得到所述每个字符的字符编码向量。

6.一种命名实体模型的训练方法，其特征在于，所述命名实体模型包括：语言模型和二元模型，所述方法包括：

获取训练集，所述训练集包括样本文本、以及所述样本文本中的样本命名实体；

调用所述预训练语言模型对所述样本文本中的每个字符进行字符编码，得到每个字符的第一字符编码向量；

调用所述二元模型基于所述每个字符的第一字符编码向量，预测所述每个字符为候选实体类型的开始字符的第一概率，以及所述每个字符为所述候选实体类型的结束字符的第二概率；基于所述第一概率和所述第二概率识别出属于所述候选实体类型的预测命名实体；

基于所述预测命名实体和所述样本命名实体之间的误差，对所述语言模型和所述二元模型的网络参数进行训练。

7.根据权利要求6所述的方法，其特征在于，所述调用所述预训练语言模型对所述样本文本中的每个字符进行字符编码，得到每个字符的第一字符编码向量，包括：

调用来自变压器的双向编码器表示BERT模型中的编码器对所述样本文本中的每个字符进行字符编码，得到每个字符的第一字符编码向量。