Microbe-Drug Association Prediction Model Based on Graph Convolution and Attention Networks

Abstract

In order to overcome the long time, high cost and low efficiency of traditional experimental methods in predicting the potential association between microorganisms and drugs, a new prediction model named GCNATMDA is proposed in this paper. The model combines two deep learning models, Graph Convolutional Network and Graph Attention Network, and aims to reveal potential relationships between microbes and drugs by learning related features, Thus improve the efficiency and accuracy of prediction. We first integrated the microbe-drug association matrix from the existing dataset, and then combined the calculated microbe-drug characteristic matrix as the model input. The GCN module is used to dig deeper into the potential characterization of microbes and drugs, while the GAT module further learns the more complex interactions between them and generates the corresponding score matrix. The experimental results show that the GCNATMDA model achieves 96.59% and 93.01% in AUC and AUPR evaluation indexes, respectively, which is significantly better than the existing prediction models. In addition, the reliability of the prediction results is verified by a series of experiments.