LightCAKE:一种用于上下文感知知识图谱嵌入的轻量级框架
人工智能
2021-03-05 v2 计算与语言
机器学习
摘要
知识图谱嵌入(KGE)模型基于观测到的三元组学习将符号实体和关系投影到连续向量空间中。然而,现有 KGE 模型无法在图上下文与模型复杂度之间做出恰当权衡,使其仍远不能令人满意。本文提出一种名为 LightCAKE 的轻量级框架用于上下文感知 KGE。LightCAKE 在不引入冗余可训练参数的情况下显式建模图上下文,并使用迭代聚合策略将上下文信息整合进实体/关系嵌入中。作为一个通用框架,它可与许多简单 KGE 模型结合使用以取得优异结果。最后,在公开基准上的大量实验证明了我们框架的效率和有效性。
引用
@article{arxiv.2102.10826,
title = {LightCAKE: A Lightweight Framework for Context-Aware Knowledge Graph Embedding},
author = {Zhiyuan Ning and Ziyue Qiao and Hao Dong and Yi Du and Yuanchun Zhou},
journal= {arXiv preprint arXiv:2102.10826},
year = {2021}
}
备注
Accepted by PAKDD 2021