一种基于预训练模型和图卷积网络的医疗实体链接方法及系统

摘要

本发明公开了一种基于预训练模型和图卷积网络的医疗实体链接方法，包括建立医疗实体链接数据集及生成自监督数据集；使用自监督数据集微调预训练模型；使用微调后的预训练模型对实体提及以及实体进行嵌入，对每一个实体提及，计算其相似度，对实体相似度进行排序得到知识库中与每一个实体提及最相似的候选实体；将一个实体提及与对应的候选实体组合起来，得到数据对，使用预训练模型对其嵌入初始化图节点表示，使用实体提及和对应的候选实体构建图卷积网络，使用图卷积网络对数据对构成的图进行学习，对于用于初始化图节点初始化嵌入的预训练模型的参数使用较小学习率学习；对节点打分后排序，选择得分最大的节点中的实体作为实体链接的结果。