有效

一种融合原型知识的关系抽取方法

庞宁、赵翔、肖卫东、葛斌、胡艳丽、谭真、张翀
中国人民解放军国防科技大学

摘要

本发明公开了一种融合原型知识的关系抽取方法,在输入层,输入训练样本;在编码层,利用预训练语言模型,对关系所涉及的头尾实体进行符号标记,并将标记所对应的隐藏层向量串联作为句子编码向量;在原型向量层,首先初始化各关系的原型向量,然后通过对比学习,最小化样本与其对应关系的原型向量之间的距离,最大化样本与其他关系类别的原型向量之间的距离,学习更新各关系的原型向量,获得原型知识;在记忆融合层,通过多头自注意力的方法,将原型知识融合到当前样本的编码向量中,用于特征增强;在输出层,通过一个多分类器根据融合了原型知识的句子编码向量对实体之间的关系进行分类预测。