A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest

Abstract Background In recent years, the extensive use of drugs and antibiotics has led to increasing microbial resistance. Therefore, it becomes crucial to explore deep connections between drugs and microbes. However, traditional biological experiments are very expensive and time-consuming. Therefo...

Full description

Bibliographic Details
Main Authors: Haiyue Kuang, Zhen Zhang, Bin Zeng, Xin Liu, Hao Zuo, Xingye Xu, Lei Wang
Format: Article
Language:English
Published: BMC 2024-02-01
Series:BMC Bioinformatics
Subjects:
Online Access:https://doi.org/10.1186/s12859-024-05687-9
_version_ 1797273010017665024
author Haiyue Kuang
Zhen Zhang
Bin Zeng
Xin Liu
Hao Zuo
Xingye Xu
Lei Wang
author_facet Haiyue Kuang
Zhen Zhang
Bin Zeng
Xin Liu
Hao Zuo
Xingye Xu
Lei Wang
author_sort Haiyue Kuang
collection DOAJ
description Abstract Background In recent years, the extensive use of drugs and antibiotics has led to increasing microbial resistance. Therefore, it becomes crucial to explore deep connections between drugs and microbes. However, traditional biological experiments are very expensive and time-consuming. Therefore, it is meaningful to develop efficient computational models to forecast potential microbe-drug associations. Results In this manuscript, we proposed a novel prediction model called GARFMDA by combining graph attention networks and bilayer random forest to infer probable microbe-drug correlations. In GARFMDA, through integrating different microbe-drug-disease correlation indices, we constructed two different microbe-drug networks first. And then, based on multiple measures of similarity, we constructed a unique feature matrix for drugs and microbes respectively. Next, we fed these newly-obtained microbe-drug networks together with feature matrices into the graph attention network to extract the low-dimensional feature representations for drugs and microbes separately. Thereafter, these low-dimensional feature representations, along with the feature matrices, would be further inputted into the first layer of the Bilayer random forest model to obtain the contribution values of all features. And then, after removing features with low contribution values, these contribution values would be fed into the second layer of the Bilayer random forest to detect potential links between microbes and drugs. Conclusions Experimental results and case studies show that GARFMDA can achieve better prediction performance than state-of-the-art approaches, which means that GARFMDA may be a useful tool in the field of microbe-drug association prediction in the future. Besides, the source code of GARFMDA is available at https://github.com/KuangHaiYue/GARFMDA.git
first_indexed 2024-03-07T14:37:29Z
format Article
id doaj.art-133c0ab8619d40e8b88eaebcea780ded
institution Directory Open Access Journal
issn 1471-2105
language English
last_indexed 2024-03-07T14:37:29Z
publishDate 2024-02-01
publisher BMC
record_format Article
series BMC Bioinformatics
spelling doaj.art-133c0ab8619d40e8b88eaebcea780ded2024-03-05T20:32:03ZengBMCBMC Bioinformatics1471-21052024-02-0125111610.1186/s12859-024-05687-9A novel microbe-drug association prediction model based on graph attention networks and bilayer random forestHaiyue Kuang0Zhen Zhang1Bin Zeng2Xin Liu3Hao Zuo4Xingye Xu5Lei Wang6Big Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityBig Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityBig Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityBig Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityBig Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityBig Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityBig Data Innovation and Entrepreneurship Education Center of Hunan Province, Changsha UniversityAbstract Background In recent years, the extensive use of drugs and antibiotics has led to increasing microbial resistance. Therefore, it becomes crucial to explore deep connections between drugs and microbes. However, traditional biological experiments are very expensive and time-consuming. Therefore, it is meaningful to develop efficient computational models to forecast potential microbe-drug associations. Results In this manuscript, we proposed a novel prediction model called GARFMDA by combining graph attention networks and bilayer random forest to infer probable microbe-drug correlations. In GARFMDA, through integrating different microbe-drug-disease correlation indices, we constructed two different microbe-drug networks first. And then, based on multiple measures of similarity, we constructed a unique feature matrix for drugs and microbes respectively. Next, we fed these newly-obtained microbe-drug networks together with feature matrices into the graph attention network to extract the low-dimensional feature representations for drugs and microbes separately. Thereafter, these low-dimensional feature representations, along with the feature matrices, would be further inputted into the first layer of the Bilayer random forest model to obtain the contribution values of all features. And then, after removing features with low contribution values, these contribution values would be fed into the second layer of the Bilayer random forest to detect potential links between microbes and drugs. Conclusions Experimental results and case studies show that GARFMDA can achieve better prediction performance than state-of-the-art approaches, which means that GARFMDA may be a useful tool in the field of microbe-drug association prediction in the future. Besides, the source code of GARFMDA is available at https://github.com/KuangHaiYue/GARFMDA.githttps://doi.org/10.1186/s12859-024-05687-9Graph attention networksBilayer random forestMicrobial-drug networksContribution value
spellingShingle Haiyue Kuang
Zhen Zhang
Bin Zeng
Xin Liu
Hao Zuo
Xingye Xu
Lei Wang
A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest
BMC Bioinformatics
Graph attention networks
Bilayer random forest
Microbial-drug networks
Contribution value
title A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest
title_full A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest
title_fullStr A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest
title_full_unstemmed A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest
title_short A novel microbe-drug association prediction model based on graph attention networks and bilayer random forest
title_sort novel microbe drug association prediction model based on graph attention networks and bilayer random forest
topic Graph attention networks
Bilayer random forest
Microbial-drug networks
Contribution value
url https://doi.org/10.1186/s12859-024-05687-9
work_keys_str_mv AT haiyuekuang anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT zhenzhang anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT binzeng anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT xinliu anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT haozuo anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT xingyexu anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT leiwang anovelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT haiyuekuang novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT zhenzhang novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT binzeng novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT xinliu novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT haozuo novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT xingyexu novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest
AT leiwang novelmicrobedrugassociationpredictionmodelbasedongraphattentionnetworksandbilayerrandomforest