Dual-neighbourhood information aggregation and feature fusion for prediction of miRNA–disease association

Abstract

Studying the intricate relationship between miRNAs and diseases is crucial to prevent and treat miRNA-related disorders. Existing computational methods often overlook the importance of features of different nodes and the propagation of features among heterogeneous nodes. Many prediction models focus only on the feature coding of miRNA and diseases and ignore the importance of feature aggregation. We propose a prediction method via dual-neighbourhood feature aggregation and feature fusion, which uses multiple sources of information, aggregates information on homogeneous and heterogeneous nodes and fuses learned features to predict multiple representations of disease nodes. We constructed similarity networks of multiple homogeneous nodes based on different similarity computation methods respectively, and fused the attention mechanism by using graph convolutional networks to obtain information of different levels of importance. To alleviate the problem of sparse connectivity in the dataset, we built a two-neighbourhood heterogeneous graph neural network model to integrate the homogeneous similarity network into a miRNA-disease heterogeneous network by using known miRNA–disease association information. We used the neighbourhood information associated with the nodes in the network to perform feature aggregation. In addition, we used a feature fusion module to learn the importance of different types of nodes to predict miRNA–disease associations. Our experimental results on the Human microRNA Disease Database (HMDD v3.2) show that the model demonstrates superior performance. This work demonstrates the capability of our model to identify potential miRNAs associated with diseases through a case study of two common cancers.