HyperAIHyperAI

Command Palette

Search for a command to run...

StarGraph:基于不完整两跳子图的知识表示学习

Hongzhu Li Xiangrui Gao* Linhui Feng Yafeng Deng† Yuhui Yin†

摘要

传统的知识图谱(KG)表示学习算法将每个实体映射到一个唯一的嵌入向量,忽略了邻域中包含的丰富信息。我们提出了一种名为StarGraph的方法,为大规模知识图谱提供了一种新颖的利用邻域信息获取实体表示的方式。首先生成每个目标节点的不完整的两跳邻域子图,然后通过改进的自注意力网络对其进行处理,以获得实体表示,该表示用于替代传统方法中的实体嵌入。我们在ogbl-wikikg2数据集上达到了最佳性能(SOTA),并在fb15k-237数据集上获得了具有竞争力的结果。实验结果证明了StarGraph在参数效率方面的优势,而其在ogbl-wikikg2上的改进则展示了其在大规模知识图谱表示学习中的有效性。代码现已发布在\url{https://github.com/hzli-ucas/StarGraph}


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
StarGraph:基于不完整两跳子图的知识表示学习 | 论文 | HyperAI超神经