Title: Graph embedding and ensemble learning for predicting gene-disease associations

Authors: Haorui Wang; Xiaochan Wang; Zhouxin Yu; Wen Zhang

Addresses: College of Informatics, Huazhong Agricultural University, Wuhan 430070, Hubei, China; School of Computer Science, Wuhan University, Wuhan 430070, Hubei, China ' School of Computer Science, Wuhan University, Wuhan 430070, Hubei, China ' College of Informatics, Huazhong Agricultural University, Wuhan 430070, Hubei, China ' College of Informatics, Huazhong Agricultural University, Wuhan 430070, Hubei, China; Hubei Engineering Technology Research Centre of Agricultural Big Data, Wuhan 430070, Hubei, China

Abstract: The discovery of gene-disease associations is important for preventing, diagnosing, and treating diseases. In this paper, we propose two heterogeneous network-based methods that enhance gene-disease association prediction by using graph embedding and ensemble learning, abbreviated as 'HNEEM' and 'HNEEM-PLUS'. We integrate gene-disease associations, gene-chemical associations, gene-gene associations and disease-chemical associations to construct a heterogeneous network, and adopt six graph embedding methods respectively to learn the representative vectors of genes and diseases from the network. We build individual prediction models using each graph embedding representation and random forest, and then combine them by average scoring to construct the ensemble model HNEEM. To increase the diversity of base predictors, we further introduce the multilayer perceptron as an additional classifier and generate more base predictors, and thus propose an extended method named 'HNEEM-PLUS'. Computational experiments show that HNEEM has better results than individual methods and HNEEM-PLUS makes more improvement than HNEEM.

Keywords: gene-disease association; heterogeneous network; graph embedding.

DOI: 10.1504/IJDMB.2020.108704

International Journal of Data Mining and Bioinformatics, 2020 Vol.23 No.4, pp.360 - 379

Received: 24 Apr 2020
Accepted: 26 Apr 2020

Published online: 27 Jul 2020 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article