Multi-scale information fusion and decoupled representation learning for robust microbe-disease interaction prediction.

Abstract

Research indicates that microbe activity within the human body significantly influences health by being closely linked to various diseases. Accurately predicting microbe-disease interactions (MDIs) offers critical insights for disease intervention and pharmaceutical research. Current advanced AI-based technologies automatically generate robust representations of microbes and diseases, enabling effective MDI predictions. However, these models continue to face significant challenges. A major issue is their reliance on complex feature extractors and classifiers, which substantially diminishes the models' generalizability. To address this, we introduce a novel graph autoencoder framework that utilizes decoupled representation learning and multi-scale information fusion strategies to efficiently infer potential MDIs. Initially, we randomly mask portions of the input microbe-disease graph based on Bernoulli distribution to boost self-supervised training and minimize noise-related performance degradation. Secondly, we employ decoupled representation learning technology, compelling the graph neural network (GNN) to independently learn the weights for each feature subspace, thus enhancing its expressive power. Finally, we implement multi-scale information fusion technology to amalgamate the multi-layer outputs of GNN, reducing information loss due to occlusion. Extensive experiments on public datasets demonstrate that our model significantly surpasses existing top MDI prediction models. This indicates that our model can accurately predict unknown MDIs and is likely to aid in disease discovery and precision pharmaceutical research. Code and data are accessible at: https://github.com/shmildsj/MDI-IFDRL.