Abstract:Abstract The prediction of molecular interactions is vital for drug discovery. Existing methods often focus on individual prediction tasks and overlook the relationships between them. Additionally, certain tasks encounter limitations due to insufficient data availability, resulting in limited performance. To overcome these limitations, we propose KGE-UNIT, a unified framework that combines knowledge graph embedding (KGE) and multi-task learning, for simultaneous prediction of drug–target interactions (DTIs) and drug–drug interactions (DDIs) and enhancing the performance of each task, even when data availability is limited. Via KGE, we extract heterogeneous features from the drug knowledge graph to enhance the structural features of drug and protein nodes, thereby improving the quality of features. Additionally, employing multi-task learning, we introduce an innovative predictor that comprises the task-aware Convolutional Neural Network-based (CNN-based) encoder and the task-aware attention decoder which can fuse better multimodal features, capture the contextual interactions of molecular tasks and enhance task awareness, leading to improved performance. Experiments on two imbalanced datasets for DTIs and DDIs demonstrate the superiority of KGE-UNIT, achieving high area under the receiver operating characteristics curves (AUROCs) (0.942, 0.987) and area under the precision-recall curve ( AUPRs) (0.930, 0.980) for DTIs and high AUROCs (0.975, 0.989) and AUPRs (0.966, 0.988) for DDIs. Notably, on the LUO dataset where the data were more limited, KGE-UNIT exhibited a more pronounced improvement, with increases of 4.32$\%$ in AUROC and 3.56$\%$ in AUPR for DTIs and 6.56$\%$ in AUROC and 8.17$\%$ in AUPR for DDIs. The scalability of KGE-UNIT is demonstrated through its extension to protein–protein interactions prediction, ablation studies and case studies further validate its effectiveness.

KG-MTL: Knowledge Graph Enhanced Multi-Task Learning for Molecular Interaction

KGE-UNIT: toward the unification of molecular interactions prediction based on knowledge graph and multi-task learning on drug discovery

Learning to Denoise Biomedical Knowledge Graph for Robust Molecular Interaction Prediction

MKGE: Knowledge Graph Embedding with Molecular Structure Information.

Multi-task learning models for predicting active compounds

MolTC: Towards Molecular Relational Modeling In Language Models

KGML-xDTD: a knowledge graph–based machine learning framework for drug treatment prediction and mechanism description

KPGT: Knowledge-Guided Pre-training of Graph Transformer for Molecular Property Prediction

MolTrans: Molecular Interaction Transformer for Drug Target Interaction Prediction

KGML-xDTD: A Knowledge Graph-based Machine Learning Framework for Drug Treatment Prediction and Mechanism Description

AttentionMGT-DTA: A multi-modal drug-target affinity prediction using graph transformer and attention mechanism

KnoMol: A Knowledge-Enhanced Graph Transformer for Molecular Property Prediction

Kinome-wide polypharmacology profiling of small molecules by multi-task graph isomorphism network approach

Knowledge-enhanced Relation Graph and Task Sampling for Few-shot Molecular Property Prediction

Predicting Molecule-Target Interaction by Learning Biomedical Network and Molecule Representations

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

MGNDTI: A Drug-Target Interaction Prediction Framework Based on Multimodal Representation Learning and the Gating Mechanism

Pushing the Boundaries of Molecular Property Prediction for Drug Discovery with Multitask Learning BERT Enhanced by SMILES Enumeration

MKDTI: Predicting drug-target interactions via multiple kernel fusion on graph attention network

A general hypergraph learning algorithm for drug multi-task predictions in micro-to-macro biomedical networks