论文浅尝 | DB4Trans：数据库内置知识图谱嵌入模型训练引擎

笔记整理：柳鹏凯，天津大学硕士

发表期刊：计算机学报 第45卷Vol.45, 第9期 No.9

链接：http://cjc.ict.ac.cn/online/onlinepaper/lpk-202297212908.pdf

动机

知识图谱嵌入技术主要将知识图谱中的实体和关系嵌入到连续的向量空间中，有助于完成诸如链接预测等下游任务。随着基于知识的人工智能的日益普及和应用，知识图谱的数据规模正在急剧增加。然而，大部分的知识图谱嵌入工作主要关注模型训练的结果，忽略了对于数据规模的可扩展性，在处理大规模知识图谱时表现出较差的性能。通过将知识图谱嵌入模型与数据库在数据管理上的优势进行有效的结合，能够在保证知识图谱嵌入模型训练的准确率和效率的同时，提供更好的可扩展性以支持大规模知识图谱数据的训练。基于此，本文提出一种数据库内置知识图谱嵌入模型训练引擎DB4Trans。首先，设计了一种用于知识图谱嵌入模型训练的数据存储方案，对实体和关系进行编码并建立索引结构，以实现模型训练过程中对中间结果的访问和更新；其次，提出了一种数据库内置的模型训练算法，对数据库与内存间的数据批量交换方案进行设计以支持大规模数据的训练与存储；最后，在不同数据集上进行了测试，比