摘要
本发明提供了一种基于交互式概率编码的实体抽取方法、装置、介质和设备;其中方法包括如下步骤:将原始文段与目标实体类型对应的问题进行拼接做BPE编码分词,并输入到预训练模块中进行词嵌入,得到词向量表征;将词向量表征进行两次独立的初步交互注意力计算,得到目标实体的起始与结束位置1D概率分布;将词向量表征做进一步门控交互注意力计算,得到目标实体坐标的2D概率分布;对2D分布进行掩码以及行、列过滤,利用过滤后的预测坐标抽取目标实体并进行BPE解码,得到最终的目标实体集合。本发明可增强对目标实体的理解与识别,并增强对目标实体所有可能边界的检测,使得抽取的准确性与查全率更高。
- 单位