Predicting drug-disease associations with heterogeneous network embedding

The prediction of drug-disease associations holds great potential for precision medicine in the era of big data and is important for the identification of new indications for existing drugs. The associations between drugs and diseases can be regarded as a complex heterogeneous network with multiple types of nodes and links. In this paper, we propose a method, namely HED ($\mathbf{H}$ eterogeneous network $\mathbf{E}$ mbedding for $\mathbf{D}$ rug-disease association), to predict potential associations between drugs and diseases based on a drug-disease heterogeneous network. Specifically, with the heterogeneous network constructed from known drug-disease associations, HED employs network embedding to characterize drug-disease associations and then trains a classifier to predict novel potential drug-disease associations. The results on two real datasets show that HED outperforms existing popular approaches. Furthermore, some of our predictions have been verified by evidence from literature. For instance, carvedilol, a drug that was originally used for heart failure, left ventricular dysfunction, and hypertension, is predicted to be useful for atrial fibrillation by HED, which is supported by clinical trials.