PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
April 3, 2026IEEE Journal of Biomedical and Health Informatics0 citations

Chemical-Disease-Gene Association Prediction based on Pretraining-Prompt-Finetuning Heterogeneous Graph Neural Network for Drug Discovery

View Full Paper
XZXi ZengJCJing-Wen CaiPLPei-Yuan Lai

Key Points

  • The aim is to develop a unified framework for predicting chemical-disease-gene associations through enhanced learning methods.
  • Developed a Pretraining-Prompt-Finetuning Heterogeneous Graph Neural Network (PPF-HGNN).
  • Constructed a heterogeneous graph to represent chemical-disease-gene interactions.
  • Employed metapath-guided message passing for better semantic understanding.
  • Implemented a dual self-supervised learning objective during pretraining.
  • Utilized task-specific prompt tuning for adaptive fine-tuning on different prediction tasks.
  • Achieved an AUC of 0.9633 for chemical-disease, 0.9939 for chemical-gene, and 0.9390 for disease-gene associations.
  • F1-scores were 0.9157, 0.9668, and 0.8955 respectively, showing high accuracy.
  • Significantly outperformed six existing baseline methods.

Abstract

Chemical-Disease-Gene (CDG) association prediction-encompassing Chemical-Disease (CD), Disease-Gene (DG), and Chemical-Gene (CG) interactions-is a cornerstone of drug discovery, as it underpins target identification and drug repurposing. While these tasks are inherently synergistic, existing methods often address them in isolation, failing to capture shared heterogeneous semantics and cross-task dependencies. We hypothesize that a unified pretraining framework can learn transferable biomedical semantics across CDG tasks, with task-specific prompt tuning enabling efficient adaptive fine-tuning without full retraining. To test this hypothesis, we propose the Pretraining-Prompt-Finetuning Heterogeneous Graph Neural Network (PPF-HGNN), a two-stage framework built on heterogeneous graph neural networks (GNNs) and prompt learning. Specifically, we construct a CDG heterogeneous graph, employ parameter-free metapath-guided message passing for high-order semantic capture, and optimize generalizable representations via a dual self-supervised objective (association prediction + feature reconstruction) during pretraining. For downstream tasks, task-specific learnable prompt vectors are introduced to adapt frozen pretrained representations to CD, CG and DG association prediction tasks via additive fusion, preserving core semantics while injecting task-specific biases. Comprehensive experiments demonstrate PPF-HGNN's state-of-the-art performance: AUC of 0.9633 (CD), 0.9939 (CG), and 0.9390 (DG), with F1-scores of 0.9157, 0.9668, and 0.8955 respectively-substantially outperforming six existing baselines. This work validates the pretrain-prompt-finetune paradigm for multi-task biomedical association prediction, providing a robust AI-driven tool to accelerate translational research and decipher complex CDG relationships. The source code is available at https://github.com/ike-zengxi/PPF-HGNN.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Zeng et al. (2026) studied this question.

synapsesocial.com/papers/69cf5f305a333a821460e1bbhttps://doi.org/10.1109/jbhi.2026.3679534
Ask AI
Helpful
Bookmark
Share
View Full Paper