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...
Main Authors: | , , , , , , |
---|---|
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 |