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