###
计算机系统应用英文版:2025,34(2):206-215
本文二维码信息
码上扫一扫!
融合DeepE和对比学习的链路预测模型
(1.福建师范大学 计算机与网络空间安全学院, 福州 350117;2.福建医科大学 临床医学部, 福州 350122;3.福州大学附属省立医院, 福州 350001)
Link Prediction Model Integrating DeepE and Contrastive Learning
(1.College of Computer and Cyber Security, Fujian Normal University, Fuzhou 350117, China;2.School of Clinical, Fujian Medical University, Fuzhou 350122, China;3.Fuzhou University Affiliated Provincial Hospital, Fuzhou 350001, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 493次   下载 1134
Received:July 15, 2024    Revised:August 13, 2024
中文摘要: 现有的知识图谱链路预测方法在学习语义信息的过程中大多只关注单个三元组中头实体h、关系r和尾实体t之间的语义关系, 没有考虑不同三元组中相关实体和实体关系之间的联系. 针对此问题, 本文提出了DeepE_CL模型. 首先, 通过DeepE模型学习相关三元组的语义信息和具有相同实体关系对的实体或具有相同实体的实体关系对的语义信息. 其次, 利用提取的相关三元组语义信息计算相应的评分函数和交叉熵损失, 并采用对比学习模型对提取的具有相同实体关系对的实体或具有相同实体的实体关系对的语义信息进行优化, 从而实现对相关三元组缺失信息的预测. 本文在4个常见的数据集上进行验证, 运用MRMRRHit@1和Hit@10这4个评价指标对所提方法和其他基线模型进行对比, 实验结果表明DeepE_CL模型在所有指标上都取得了最好的结果. 为了进一步验证模型的实用性, 本文还将模型应用到了1个真实的中成药数据集, 实验结果显示DeepE_CL模型比DeepE模型在MR指标上降低了18, 在MRRHit@1指标上分别提升了0.8%、1.1%, 在Hit@10指标上维持不变. 实验证明了引入对比学习模型的DeepE_CL模型在提升知识图谱链路预测性能方面的有效性.
Abstract:Most of the existing knowledge graph link prediction methods focus only on the semantic relationships between a head entity h, a relationship r, and a tail entity t in a single triad in learning semantic information. They do not consider the links between related entities and entity relationships in different triads. To address this problem, this study proposes the DeepE_CL model. Firstly, the study uses the DeepE model to learn the semantic information of related triads and entities with the same entity relationship pairs or entity relationship pairs with the same entities. Secondly, the extracted semantic information of the related triads is used to calculate the corresponding scoring function and cross-entropy loss, and the extracted semantic information of entities with the same entity relationship pairs or entity relationship pairs with the same entities is optimized through the comparative learning model, so as to predict the missing information of the related triads. This paper validates the proposed method through four common datasets and compares the proposed method with other baseline models by applying four evaluation indicators, including MR, MRR, Hit@1, and Hit@10. The experimental results show that the DeepE_CL model achieves the best results in all indicators. To further validate the usefulness of the model, this study also applies the model to a real traditional Chinese medicine (TCM) dataset, and the experimental results show that compared with the DeepE model, the DeepE_CL model reduces the MR indicators by 18, and improves the MRR, Hit@1 indicators by 0.8%, 1.1%, and the Hit@10 indicators remain unchanged. The experiments demonstrate that the DeepE_CL model, introducing a comparative learning model, is very effective in improving the performance of knowledge graph link prediction.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金(62171131); 福建省自然科学基金(2022J01398)
引用文本:
翁慧敏,郭躬德,林世水.融合DeepE和对比学习的链路预测模型.计算机系统应用,2025,34(2):206-215
WENG Hui-Min,GUO Gong-De,LIN Shi-Shui.Link Prediction Model Integrating DeepE and Contrastive Learning.COMPUTER SYSTEMS APPLICATIONS,2025,34(2):206-215