Original Research ARTICLE
Graph embedding deep learning guide microbial biomarkers' identification
- 1Central China Normal University, China
- 2School of Computer, Central China Normal University, China
The microbiome-wide association studies are to figure out the relationship between microorganisms and human, with the goal of discovering relevant biomarkers to guide the disease diagnosis. However, the microbiome data is complex, high noise and dimensions. Traditional machine learning methods are limited by the models’ representation ability and cannot learn complex patterns from the data. Recently, deep learning has been widely applied to fields ranging from text processing to image recognition due to its efficient flexibility and high capacity. But the deep learning models must be trained with enough data in order to achieve good performance, which is impractical in reality. In addition, deep learning is considered as black box and hard to interpret. These factors make deep learning not widely used in microbiome-wide association studies. In this work, we construct a sparse microbial interaction network and embed this graph into deep model to alleviate the risk of overfitting and improve the performance. Further, we explore a Graph Embedding Deep Feedforward Network (GEDFN) to conduct feature selection and guide meaningful microbial markers’ identification. Based on the experimental results, we verify the feasibility of combining the microbial graph model with the deep learning, and demonstrate the feasibility of applying deep learning and feature selection on microbial data. Our main contributions are: firstly, we utilize different methods to construct a variety of microbial interaction networks and combine the network via graph embedding deep learning. Secondly, we introduce a feature selection method based on graph embedding and validate the biological meaning of microbial markers. The code is available at https://github.com/MicroAVA/GEDFN.git
Keywords: Graph embedding, deep learning, Feature Selection, biomarkers, microbiome
Received: 12 Aug 2019;
Accepted: 24 Oct 2019.
Copyright: © 2019 Zhu, Jiang, Zhu, Pan and He. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.
* Correspondence: Prof. Xingpeng Jiang, Central China Normal University, Wuhan, 430079, Hubei Province, China, email@example.com