LambdaKG: A Library for Pre-trained Language Model-Based Knowledge Graph Embeddings
Xin Xie,Zhoubo Li,Xiaohan Wang,Zekun Xi,Ningyu Zhang
2023-09-14
Abstract:Knowledge Graphs (KGs) often have two characteristics: heterogeneous graph structure and text-rich entity/relation information. Text-based KG embeddings can represent entities by encoding descriptions with pre-trained language models, but no open-sourced library is specifically designed for KGs with PLMs at present. In this paper, we present LambdaKG, a library for KGE that equips with many pre-trained language models (e.g., BERT, BART, T5, GPT-3), and supports various tasks (e.g., knowledge graph completion, question answering, recommendation, and knowledge probing). LambdaKG is publicly open-sourced at <a class="link-external link-https" href="https://github.com/zjunlp/PromptKG/tree/main/lambdaKG" rel="external noopener nofollow">this https URL</a>, with a demo video at <a class="link-external link-http" href="http://deepke.zjukg.cn/lambdakg.mp4" rel="external noopener nofollow">this http URL</a> and long-term maintenance.
Computation and Language,Artificial Intelligence,Databases,Information Retrieval,Machine Learning