KGOT: Unified Knowledge Graph and Optimal Transport Pseudo-Labeling for Molecule-Protein Interaction Prediction

Summary

KGOT is a novel framework that enhances molecule-protein interaction (MPI) prediction by integrating diverse biological data, including molecular, protein, gene, and pathway-level interactions. It employs an optimal transport-based pseudo-labeling approach to generate high-quality labels for unlabeled molecule-protein pairs. This method significantly improves prediction accuracies and zero-shot capabilities over state-of-the-art techniques, demonstrating strong performance on virtual screening and protein retrieval tasks.

Medical Relevance

This research is crucial for accelerating drug discovery by improving the accuracy and efficiency of identifying potential drug candidates and their protein targets. More precise MPI prediction can expedite the identification of novel therapeutics and enhance our understanding of molecular function in health and disease.

AI Health Application

This research applies AI techniques, specifically unified knowledge graphs and optimal transport pseudo-labeling, to enhance the prediction of molecule-protein interactions. This AI application directly supports drug discovery by improving the identification of potential drug candidates, understanding their mechanisms of action, and accelerating the virtual screening process, thereby contributing to the development of new treatments and therapies.

Key Points

• Addresses two major challenges in MPI prediction: scarcity of labeled molecule-protein pairs and the limited use of broader biological context by existing models.
• Unifies diverse biological datasets, incorporating information from molecular structures, proteins, genes, and metabolic pathways to provide a comprehensive biological context.
• Utilizes an optimal transport-based pseudo-labeling mechanism to generate high-quality labels for previously unlabeled molecule-protein interactions, guided by the underlying distribution of known interactions.
• Effectively bridges disparate biological modalities, enabling the synergistic use of heterogeneous data to enhance MPI prediction accuracy.
• Achieves substantial improvements over state-of-the-art methods in prediction accuracies and demonstrates superior zero-shot ability across unseen interactions.
• Evaluated on practical MPI tasks, including virtual screening and protein retrieval, showcasing its efficacy in relevant drug discovery scenarios.
• Proposes a new paradigm for leveraging diverse biological data sources to solve problems traditionally constrained by single- or bi-modal learning, extending beyond MPI prediction.

Clinical Impact

KGOT's improved accuracy and zero-shot learning capabilities can significantly enhance the efficiency of virtual screening in drug discovery, leading to faster and more cost-effective identification of promising drug candidates. This can accelerate the development of new therapeutics, enabling the rapid exploration of novel drug targets and compounds for various diseases, ultimately bringing new treatments to patients sooner.

Limitations

The abstract does not explicitly state limitations of the KGOT framework itself. However, it highlights two major challenges in the field that KGOT aims to overcome: the scarcity of labeled molecule-protein pairs and the limitation of existing methods relying solely on molecular and protein features, ignoring broader biological context.

Future Directions

The authors suggest that their approach provides a new paradigm for leveraging diverse biological data sources, implying its applicability extends beyond MPI prediction. This paves the way for future advances in computational biology and drug discovery by addressing other problems traditionally constrained by single- or bi-modal learning, fostering more holistic biological insights.

Medical Domains

Keywords

Abstract

Predicting molecule-protein interactions (MPIs) is a fundamental task in computational biology, with crucial applications in drug discovery and molecular function annotation. However, existing MPI models face two major challenges. First, the scarcity of labeled molecule-protein pairs significantly limits model performance, as available datasets capture only a small fraction of biological relevant interactions. Second, most methods rely solely on molecular and protein features, ignoring broader biological context such as genes, metabolic pathways, and functional annotations that could provide essential complementary information. To address these limitations, our framework first aggregates diverse biological datasets, including molecular, protein, genes and pathway-level interactions, and then develop an optimal transport-based approach to generate high-quality pseudo-labels for unlabeled molecule-protein pairs, leveraging the underlying distribution of known interactions to guide label assignment. By treating pseudo-labeling as a mechanism for bridging disparate biological modalities, our approach enables the effective use of heterogeneous data to enhance MPI prediction. We evaluate our framework on multiple MPI datasets including virtual screening tasks and protein retrieval tasks, demonstrating substantial improvements over state-of-the-art methods in prediction accuracies and zero shot ability across unseen interactions. Beyond MPI prediction, our approach provides a new paradigm for leveraging diverse biological data sources to tackle problems traditionally constrained by single- or bi-modal learning, paving the way for future advances in computational biology and drug discovery.