UMMAN: Unsupervised Multi-graph Merge Adversarial Network for Disease Prediction Based on Intestinal Flora (2407.21714v1)
Abstract: The abundance of intestinal flora is closely related to human diseases, but diseases are not caused by a single gut microbe. Instead, they result from the complex interplay of numerous microbial entities. This intricate and implicit connection among gut microbes poses a significant challenge for disease prediction using abundance information from OTU data. Recently, several methods have shown potential in predicting corresponding diseases. However, these methods fail to learn the inner association among gut microbes from different hosts, leading to unsatisfactory performance. In this paper, we present a novel architecture, Unsupervised Multi-graph Merge Adversarial Network (UMMAN). UMMAN can obtain the embeddings of nodes in the Multi-Graph in an unsupervised scenario, so that it helps learn the multiplex association. Our method is the first to combine Graph Neural Network with the task of intestinal flora disease prediction. We employ complex relation-types to construct the Original-Graph and disrupt the relationships among nodes to generate corresponding Shuffled-Graph. We introduce the Node Feature Global Integration (NFGI) module to represent the global features of the graph. Furthermore, we design a joint loss comprising adversarial loss and hybrid attention loss to ensure that the real graph embedding aligns closely with the Original-Graph and diverges from the Shuffled-Graph. Comprehensive experiments on five classical OTU gut microbiome datasets demonstrate the effectiveness and stability of our method. (We will release our code soon.)
- E. W. Beals, “Bray-Curtis ordination: an effective strategy for analysis of multivariate ecological data,” in Advances in Ecological Research, vol. 14, pp. 1–55, Elsevier, 1984.
- L. Breiman, “Random forests,” Machine Learning, vol. 45, pp. 5–32, 2001.
- J. Cheng, G. Dasoulas, H. He, C. Agarwal, and M. Zitnik, “GNNDelete: A General Strategy for Unlearning in Graph Neural Networks,” arXiv preprint arXiv:2302.13406, 2023.
- J. C. Clemente, L. K. Ursell, L. W. Parfrey, and R. Knight, “The impact of the gut microbiota on human health: an integrative view,” Cell, vol. 148, no. 6, pp. 1258–1270, 2012.
- L. Dethlefsen, S. Huse, M. L. Sogin, and D. A. Relman, “The pervasive effects of an antibiotic on the human gut microbiota, as revealed by deep 16S rRNA sequencing,” PLoS Biology, vol. 6, no. 11, p. e280, 2008.
- J. Dicksved, J. Halfvarson, M. Rosenquist, G. Järnerot, C. Tysk, J. Apajalahti, L. Engstrand, and J. K. Jansson, “Molecular analysis of the gut microbiota of identical twins with Crohn’s disease,” The ISME Journal, vol. 2, no. 7, pp. 716–727, 2008.
- R. C. Edgar, “UPARSE: highly accurate OTU sequences from microbial amplicon reads,” Nature Methods, vol. 10, no. 10, pp. 996–998, 2013.
- D. N. Frank, A. L. St. Amand, R. A. Feldman, E. C. Boedeker, N. Harpaz, and N. R. Pace, “Molecular-phylogenetic characterization of microbial community imbalances in human inflammatory bowel diseases,” Proceedings of the National Academy of Sciences, vol. 104, no. 34, pp. 13780–13785, 2007.
- W. Hamilton, Z. Ying, and J. Leskovec, “Inductive representation learning on large graphs,” Advances in neural information processing systems, vol. 30, 2017.
- G. Jurman, S. Riccadonna, R. Visintainer, and C. Furlanello, “Canberra distance on ranked lists,” in Proceedings of Advances in Ranking NIPS 09 Workshop, pp. 22–27, Citeseer, 2009.
- T. N. Kipf and M. Welling, “Semi-supervised classification with graph convolutional networks,” arXiv preprint arXiv:1609.02907, 2016.
- A. Krizhevsky, I. Sutskever, and G. E. Hinton, “Imagenet classification with deep convolutional neural networks,” Advances in neural information processing systems, vol. 25, 2012.
- R. E. Ley, P. J. Turnbaugh, S. Klein, and J. I. Gordon, “Human gut microbes associated with obesity,” Nature, vol. 444, no. 7122, pp. 1022–1023, 2006.
- J. Liñares Blanco, C. Fernández-Lozano, J. A. Seoane Fernández, and G. Lopez-Campos, “Machine Learning Algorithms Reveals Country-Specific Metagenomic Taxa from American Gut Project Data,” Studies in Health Technology and Informatics, vol. 281, pp. 382–386, 2021.
- W. Liu, R. Zhang, R. Shu, J. Yu, H. Li, H. Long, S. Jin, S. Li, Q. Hu, F. Yao, and others, “Study of the relationship between microbiome and colorectal cancer susceptibility using 16SrRNA sequencing,” BioMed Research International, vol. 2020, 2020.
- I. Manandhar, A. Alimadadi, S. Aryal, P. Munroe, B. Joe, and X. Cheng, “Machine Learning of Gut Microbiome Composition for Diagnostic Classification of Inflammatory Bowel Diseases,” The FASEB Journal, vol. 35, 2021.
- C. Manichanh, L. Rigottier-Gois, E. Bonnaud, K. Gloux, E. Pelletier, L. Frangeul, R. Nalin, C. Jarrin, P. Chardon, P. Marteau, and others, “Reduced diversity of faecal microbiota in Crohn’s disease revealed by a metagenomic approach,” Gut, vol. 55, no. 2, pp. 205–211, 2006.
- A. McDowell, J. Kang, J. Yang, J. Jung, Y.-M. Oh, S.-M. Kym, T.-S. Shin, T.-B. Kim, Y.-K. Jee, and Y.-K. Kim, “Machine-learning algorithms for asthma, COPD, and lung cancer risk assessment using circulating microbial extracellular vesicle data and their application to assess dietary effects,” Experimental & Molecular Medicine, vol. 54, no. 9, pp. 1586–1595, 2022.
- A. Pajot, E. De Bézenac, and P. Gallinari, “Unsupervised adversarial image reconstruction,” in International Conference on Learning Representations, 2018.
- E. Pasolli, D. T. Truong, F. Malik, L. Waldron, and N. Segata, “Machine learning meta-analysis of large metagenomic datasets: tools and biological insights,” PLoS Computational Biology, vol. 12, no. 7, p. e1004977, 2016.
- D. E. Rumelhart, G. E. Hinton, and R. J. Williams, “Learning representations by back-propagating errors,” Nature, vol. 323, no. 6088, pp. 533–536, 1986.
- D. Sharma, A. D. Paterson, and W. Xu, “TaxoNN: ensemble of neural networks on stratified microbiome data for disease prediction,” Bioinformatics, vol. 36, no. 17, pp. 4544–4550, 2020.
- Q. Song, Y. Wang, L. Huang, M. Shen, Y. Yu, Q. Yu, Y. Chen, and J. Xie, “Review of the relationships among polysaccharides, gut microbiota, and human health,” Food Research International, vol. 140, p. 109858, 2021.
- C. Tana, Y. Umesaki, A. Imaoka, T. Handa, M. Kanazawa, and S. Fukudo, “Altered profiles of intestinal microbiota and organic acids may be the origin of symptoms in irritable bowel syndrome,” Neurogastroenterology & Motility, vol. 22, no. 5, pp. 512–e115, 2010.
- DJ Tena Cucala, B Cuenca Grau, EV Kostylev, and B Motik, “Explainable GNN-based models over knowledge graphs,” 2022, OpenReview.
- C. W. Wong, S. E. Yost, J. S. Lee, J. D. Gillece, M. Folkerts, L. Reining, S. K. Highlander, Z. Eftekhari, J. Mortimer, and Y. Yuan, “Analysis of gut microbiome using explainable machine learning predicts risk of diarrhea associated with tyrosine kinase inhibitor neratinib: a pilot study,” Frontiers in Oncology, vol. 11, p. 604584, 2021.
- F. Xia, K. Sun, S. Yu, A. Aziz, L. Wan, S. Pan, and H. Liu, “Graph learning: A survey,” IEEE Transactions on Artificial Intelligence, vol. 2, no. 2, p. 109-127, 2021.
- Y.-H. Zhou and P. Gallins, “A review and tutorial of machine learning methods for microbiome host trait prediction,” Frontiers in Genetics, vol. 10, p. 579, 2019.
- R. Sender, S. Fuchs and R. Milo, “Revised estimates for the number of human and bacteria cells in the body,” PLoS Biology, vol. 14, no. 8, p. e1002533, 2016.