Skip to main content

ORIGINAL RESEARCH article

Front. Pharmacol.
Sec. Ethnopharmacology
Volume 15 - 2024 | doi: 10.3389/fphar.2024.1484639
This article is part of the Research Topic Artificial Intelligence in Traditional Medicine Research and Application View all articles

MKAN-MMI: Empowering Traditional Medicine-Microbe Interaction Prediction with Masked Graph Autoencoders and KANs

Provisionally accepted
Sheng Ye Sheng Ye 1*Jue Wang Jue Wang 2Mingmin Zhu Mingmin Zhu 1Sisi Yuan Sisi Yuan 3Linlin Zhuo Linlin Zhuo 4Tiancong Chen Tiancong Chen 5Jinjian Gao Jinjian Gao 1
  • 1 Wenzhou Medical University, Wenzhou, China
  • 2 Shandong University, Jinan, Shandong Province, China
  • 3 University of North Carolina at Charlotte, Charlotte, North Carolina, United States
  • 4 Wenzhou University of Technology, Wenzhou, China
  • 5 Wenzhou People’s Hospital, Wenzhou, Zhejiang Province, China

The final, formatted version of the article will be published soon.

    The growing microbial resistance to traditional medicines necessitates in-depth analysis of medicine-microbe interactions (MMIs) to develop new therapeutic strategies. Widely used artificial intelligence models are limited by sparse observational data and prevalent noise, leading to over-reliance on specific data for feature extraction and reduced generalization ability. To address these limitations, we integrate Kolmogorov-Arnold Networks (KANs), independent subspaces, and collaborative decoding techniques into the masked graph autoencoder (Mask GAE) framework, creating an innovative MMI prediction model with enhanced accuracy, generalization, and interpretability. First, we apply Bernoulli distribution to randomly mask parts of the medicinemicrobe graph, advancing self-supervised training and reducing noise impact. Additionally, the independent subspace technique enables graph neural networks (GNNs) to learn weights independently across different feature subspaces, enhancing feature expression. Fusing the multi-layer outputs of GNNs effectively reduces information loss caused by masking. Moreover, using KANs for advanced nonlinear mapping enhances the learnability and interpretability of weights, deepening the understanding of complex MMIs.These measures significantly enhanced the accuracy, generalization, and interpretability of our model in MMI prediction tasks. We validated our model on three public datasets with results showing that our model 1 Sample et al. outperformed existing leading models. The relevant data and code are publicly accessible at: https://github.com/zhuoninnin1992/MKAN-MMI.

    Keywords: Traditional medicine (TM), medicine-microbe interactions (MMIs), Artificial intelligence models, masked graph autoencoder (Mask GAE), Kolmogorov-Arnold Networks (KANs)

    Received: 22 Aug 2024; Accepted: 08 Oct 2024.

    Copyright: © 2024 Ye, Wang, Zhu, Yuan, Zhuo, Chen and Gao. 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) or licensor 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: Sheng Ye, Wenzhou Medical University, Wenzhou, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.