跳到正文
arXiv Artificial Intelligence· Youmin Ko, Hyemin Yang, Taeuk Kim, Hyunjoon Kim·· 20 小时前AI 评分22

SATKGC:用子图感知训练与结构对比学习提升语言模型知识图谱补全

Subgraph-Aware Training of Language Models for Knowledge Graph Completion Using Structure-Aware Contrastive Learning

AI 导读

研究者提出子图感知训练框架 SATKGC,通过子图感知 mini-batching 促进难负样本采样并缓解实体出现频率不平衡,同时用新的对比学习聚焦结构上更难的批内负三元组和正三元组。该工作称首次将知识图谱的结构归纳偏置全面引入 PLM 微调,在三个 KGC 基准上验证了优越性,代码已开源。

来源:arXiv Artificial Intelligence · arxiv.org