基于混合几何张量分解的知识图谱补全
机器学习
2025-04-04 v1 人工智能
信息检索
机器学习
摘要
本文提出了一种新的几何方法,用于通过低秩张量近似进行知识图谱补全。我们在基于 Tucker 张量分解的预训练且已建立的欧几里得模型中增添了一种新颖的双曲相互作用项。这种纠正使我们能够更细致地捕捉数据中分布属性,更贴合实际知识图谱。通过将两种几何结构结合在一起,我们的方法提高了所得模型的表达能力,实现了与以往欧几里得和双曲模型相比,在显著降低参数数量的同时实现了新的状态突破链接预测准确率。
引用
@article{arxiv.2504.02589,
title = {Knowledge Graph Completion with Mixed Geometry Tensor Factorization},
author = {Viacheslav Yusupov and Maxim Rakhuba and Evgeny Frolov},
journal= {arXiv preprint arXiv:2504.02589},
year = {2025}
}
备注
Accepted to AISTATS 2025