11 天前

PyKEEN 1.0：一个用于训练和评估知识图谱嵌入的 Python 库

Mehdi Ali, Max Berrendorf, Charles Tapley Hoyt, Laurent Vermue, Sahand Sharifzadeh, Volker Tresp, Jens Lehmann

摘要

近年来，知识图谱嵌入（Knowledge Graph Embeddings, KGEs）受到广泛关注，为此已开发出多个软件库以支持KGE的训练与评估。尽管各类工具各有侧重，我们仍通过社区协作对早期KGE库之一——PyKEEN进行了重新设计与重实现。PyKEEN 1.0 允许用户基于多种交互模型、训练方法、损失函数构建知识图谱嵌入模型（KGEMs），并支持显式建模逆关系。此外，系统实现了自动内存优化机制，以充分挖掘硬件资源的潜力；同时通过集成Optuna，提供了强大的自动超参数优化（Hyper-Parameter Optimization, HPO）功能。