Abstract:Understanding complex interactions in biomedical networks is crucial for advancements in biomedicine. Traditional link prediction (LP) methods, using similarity metrics like Personalized PageRank, are limited in capturing the complexity of biological networks. Recently, representation-based learning techniques have emerged, mapping nodes to low-dimensional embeddings to enhance prediction accuracy. However, these methods often face challenges with interpretability and scalability in large, complex networks. Based on a representation of biological systems as knowledge graphs (KGs), which encode entities and their relationships as triplets, we propose here BioKGC, a novel graph neural network framework which builds upon the Neural Bellman-Ford Network (NBFNet). It addresses the limitations of previous methods by utilizing path-based reasoning for LP in biomedical knowledge graphs (KGs). Unlike node-embedding learning frameworks that optimize the embedding space based on single triplets, BioKGC learns representations between nodes by considering all relations along paths. This approach enhances prediction accuracy and interpretability, allowing for the visualization of influential paths and facilitating the validation of biological plausibility. BioKGC leverages a background regulatory graph (BRG) for enhanced message passing and implements a stringent negative sampling strategy to improve learning precision. In evaluations across various LP tasks, gene function annotation, drug-disease interaction prediction, synthetic lethality prediction, and lncRNA-mRNA regulatory relationship inference, BioKGC consistently outperformed state-of-the art methods. BioKGC outperformed knowledge graph embedding and GNN-based methods in gene function prediction, especially with BRG information. We demonstrated that BioKGC effectively predicts drug-disease interactions in zero-shot learning scenarios, surpassing state-of-the-art models like TxGNN. Additionally, BioKGC demonstrated robust performance in synthetic lethality prediction and the capacity for scoring novel lncRNA-mRNA interactions, showcasing its versatility in diverse biomedical applications. One of BioKGC's key advantages is its interpretability, enabling researchers to trace prediction paths and gain insights into molecular mechanisms. Combined with its use of regulatory information for message passing, BioKGC is a powerful tool for predicting complex biological interactions, making it valuable for drug discovery and personalized medicine.

PT-KGNN: A framework for pre-training biomedical knowledge graphs with graph neural networks

Pre-training graph neural networks for link prediction in biomedical networks

Path-based reasoning in biomedical knowledge graphs

A knowledge-guided pre-training framework for improving molecular representation learning

BioBLP: A Modular Framework for Learning on Multimodal Biomedical Knowledge Graphs

DeepKG: an end-to-end deep learning-based workflow for biomedical knowledge graph extraction, optimization and applications

KGNN: Distributed Framework for Graph Neural Knowledge Representation

A data-centric framework of improving graph neural networks for knowledge graph embedding

Pretrain-KGEs: Learning Knowledge Representation from Pretrained Models for Knowledge Graph Embeddings

Structure Pre-training and Prompt Tuning for Knowledge Graph Transfer

MEGA: Meta-Graph Augmented Pre-Training Model for Knowledge Graph Completion

PTGB: Pre-Train Graph Neural Networks for Brain Network Analysis

PharmKG: a dedicated knowledge graph benchmark for bomedical data mining

Biomedical Knowledge Graph Embedding with Householder Projection (student Abstract)

KGNN: Knowledge Graph Neural Network for Drug-Drug Interaction Prediction

Knowledge Graph Embeddings in the Biomedical Domain: Are They Useful? A Look at Link Prediction, Rule Learning, and Downstream Polypharmacy Tasks

MKGE: Knowledge Graph Embedding with Molecular Structure Information.

MPTN: A message-passing transformer network for drug repurposing from knowledge graph

Pretrain-KGE - Learning Knowledge Representation from Pretrained Language Models.

MegaKG: Toward an explainable knowledge graph for early drug development

STonKGs: a sophisticated transformer trained on biomedical text and knowledge graphs