Graph Diffusion Network for Drug-Gene Prediction

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wu, Jiayang, Gan, Wensheng, Yu, Philip S.
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910825613099008
author Wu, Jiayang
Gan, Wensheng
Yu, Philip S.
author_facet Wu, Jiayang
Gan, Wensheng
Yu, Philip S.
contents Predicting drug-gene associations is crucial for drug development and disease treatment. While graph neural networks (GNN) have shown effectiveness in this task, they face challenges with data sparsity and efficient contrastive learning implementation. We introduce a graph diffusion network for drug-gene prediction (GDNDGP), a framework that addresses these limitations through two key innovations. First, it employs meta-path-based homogeneous graph learning to capture drug-drug and gene-gene relationships, ensuring similar entities share embedding spaces. Second, it incorporates a parallel diffusion network that generates hard negative samples during training, eliminating the need for exhaustive negative sample retrieval. Our model achieves superior performance on the DGIdb 4.0 dataset and demonstrates strong generalization capability on tripartite drug-gene-disease networks. Results show significant improvements over existing methods in drug-gene prediction tasks, particularly in handling complex heterogeneous relationships. The source code is publicly available at https://github.com/csjywu1/GDNDGP.
format Preprint
id arxiv_https___arxiv_org_abs_2502_09335
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Graph Diffusion Network for Drug-Gene Prediction
Wu, Jiayang
Gan, Wensheng
Yu, Philip S.
Machine Learning
Artificial Intelligence
Predicting drug-gene associations is crucial for drug development and disease treatment. While graph neural networks (GNN) have shown effectiveness in this task, they face challenges with data sparsity and efficient contrastive learning implementation. We introduce a graph diffusion network for drug-gene prediction (GDNDGP), a framework that addresses these limitations through two key innovations. First, it employs meta-path-based homogeneous graph learning to capture drug-drug and gene-gene relationships, ensuring similar entities share embedding spaces. Second, it incorporates a parallel diffusion network that generates hard negative samples during training, eliminating the need for exhaustive negative sample retrieval. Our model achieves superior performance on the DGIdb 4.0 dataset and demonstrates strong generalization capability on tripartite drug-gene-disease networks. Results show significant improvements over existing methods in drug-gene prediction tasks, particularly in handling complex heterogeneous relationships. The source code is publicly available at https://github.com/csjywu1/GDNDGP.
title Graph Diffusion Network for Drug-Gene Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2502.09335