mlsanityData to 14 Oct 2026

PyKEEN 1.0: A Python Library for Training and Evaluating Knowledge Graph Embeddings

arXiv 2007.14175 ↗First posted 28 Jul 2020cs.LG cs.AI stat.MLcited by 77

Not on the map. Why?

Abstract (arXiv)

Recently, knowledge graph embeddings (KGEs) received significant attention, and several software libraries have been developed for training and evaluating KGEs. While each of them addresses specific needs, we re-designed and re-implemented PyKEEN, one of the first KGE libraries, in a community effort. PyKEEN 1.0 enables users to compose knowledge graph embedding models (KGEMs) based on a wide range of interaction models, training approaches, loss functions, and permits the explicit modeling of inverse relations. Besides, an automatic memory optimization has been realized in order to exploit the provided hardware optimally, and through the integration of Optuna extensive hyper-parameter optimization (HPO) functionalities are provided.

Key facts. Cited by 77 papers in the map (data through 14 Oct 2026). First posted on arXiv 28 Jul 2020.

Builds on

No reference list yet.

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2021: 12; 2022: 15; 2023: 17; 2024: 14; 2025: 11; 2026: 8.

77 papers.

Sort:

By area:

  1. 2021NodePiece: Compositional and Parameter-Efficient Representations of Large Knowledge Graphscited by 72
  2. 2022An Open Challenge for Inductive Link Prediction on Knowledge Graphscited by 26
  3. 2022Inductive Logical Query Answering in Knowledge Graphscited by 26
  4. 2021Improving Inductive Link Prediction Using Hyper-Relational Factscited by 22
  5. 2021Understanding the Performance of Knowledge Graph Embeddings in Drug Discoverycited by 18
  6. 2022A Unified Framework for Rank-based Evaluation Metrics for Link Prediction in Knowledge Graphscited by 15
  7. 2024Actionable Cyber Threat Intelligence using Knowledge Graphs and Large Language Modelscited by 14
  8. 2022ExpressivE: A Spatio-Functional Embedding For Knowledge Graph Completioncited by 13
  9. 2021Knowledge Sheaves: A Sheaf-Theoretic Framework for Knowledge Graph Embeddingcited by 11
  10. 2021Implications of Topological Imbalance for Representation Learning on Biomedical Knowledge Graphscited by 11
  11. 2022ChemicalX: A Deep Learning Library for Drug Pair Scoringcited by 11
  12. 2021Improving Scene Graph Classification by Exploiting Knowledge from Textscited by 10
38 more
  1. 2022KRACL: Contrastive Learning with Graph Context Modeling for Sparse Knowledge Graph Completioncited by 10
  2. 2025HypKG: Hypergraph-based Knowledge Graph Contextualization for Precision Healthcarecited by 9
  3. 2021The Tensor Brain: A Unified Theory of Perception, Memory and Semantic Decodingcited by 8
  4. 2022NeuralKG: An Open Source Library for Diverse Representation Learning of Knowledge Graphscited by 8
  5. 2024Towards Artwork Explanation in Large-scale Vision Language Modelscited by 8
  6. 2021Unified Interpretation of Softmax Cross-Entropy and Negative Sampling: With Case Study for Knowledge Graph Embeddingcited by 7
  7. 2021GRAPE for Fast and Scalable Graph Processing and random walk-based Embeddingcited by 7
  8. 2021Towards Automatic Bias Detection in Knowledge Graphscited by 6
  9. 2021MOOMIN: Deep Molecular Omics Network for Anti-Cancer Drug Combination Therapycited by 6
  10. 2021Triple Classification for Scholarly Knowledge Graph Completioncited by 5
  11. 2022LambdaKG: A Library for Pre-trained Language Model-Based Knowledge Graph Embeddingscited by 5
  12. 2023BioBLP: A Modular Framework for Learning on Multimodal Biomedical Knowledge Graphscited by 5
  13. 2022A Knowledge Graph Embeddings based Approach for Author Name Disambiguation using Literalscited by 4
  14. 2022μKG: A Library for Multi-source Knowledge Graph Embeddings and Applicationscited by 4
  15. 2022KGxBoard: Explainable and Interactive Leaderboard for Evaluation of Knowledge Graph Completion Modelscited by 4
  16. 2023Universal Preprocessing Operators for Embedding Knowledge Graphs with Literalscited by 4
  17. 2023Clinical Trial Recommendations Using Semantics-Based Inductive Inference and Knowledge Graph Embeddingscited by 4
  18. 2024SnapE -- Training Snapshot Ensembles of Link Prediction Modelscited by 4
  19. 2022Hardware-agnostic Computation for Large-scale Knowledge Graph Embeddingscited by 3
  20. 2024Identifying the Achilles' Heel: An Iterative Method for Dynamically Uncovering Factual Errors in Large Language Modelscited by 3
  21. 2024Improving Molecule Generation and Drug Discovery with a Knowledge-enhanced Generative Modelcited by 3
  22. 2022LiveSchema: A Gateway Towards Learning on Knowledge Graph Schemascited by 2
  23. 2022The URW-KG: a Resource for Tackling the Underrepresentation of non-Western Writerscited by 2
  24. 2023A Knowledge Graph Perspective on Supply Chain Resiliencecited by 2
  25. 2023Literal-Aware Knowledge Graph Embedding for Welding Quality Monitoring: A Bosch Casecited by 2
  26. 2023Know2BIO: A Comprehensive Dual-View Benchmark for Evolving Biomedical Knowledge Graphscited by 2
  27. 2023Towards a Gateway for Knowledge Graph Schemas Collection, Analysis, and Embeddingcited by 2
  28. 2024MARS: A neurosymbolic approach for interpretable drug discoverycited by 2
  29. 2024A Survey on Knowledge Graph Structure and Knowledge Graph Embeddingscited by 2
  30. 2025Knowledge-augmented Pre-trained Language Models for Biomedical Relation Extractioncited by 2
  31. 2025Augmenting generative models with biomedical knowledge graphs improves targeted drug discoverycited by 2
  32. 2022Leveraging Wikidata's edit history in knowledge graph refinement taskscited by 1
  33. 2023NeuralKG-ind: A Python Library for Inductive Knowledge Graph Representation Learningcited by 1
  34. 2023Knowledge Graphs are not Created Equal: Exploring the Properties and Structure of Real KGscited by 1
  35. 2023Does Pre-trained Language Model Actually Infer Unseen Links in Knowledge Graph Completion?cited by 1
  36. 2024Context-aware knowledge graph framework for traffic speed forecasting using graph neural networkcited by 1
  37. 2025Understanding the Embedding Models on Hyper-relational Knowledge Graphcited by 1
  38. 2025Enhancing PyKEEN with Multiple Negative Sampling Solutions for Knowledge Graph Embedding Modelscited by 1

50 most cited of 77 All 77 (JSON)