Skip to content

Drug-Drug Interaction Prediction Based on Knowledge Graph Embeddings and Convolutional-LSTM Network

Notifications You must be signed in to change notification settings

rezacsedu/Drug-Drug-Interaction-Prediction

Repository files navigation

Drug-Drug Interaction Prediction using Knowledge Graph Embeddings & Conv-LSTM Network

Implementation of our paper titled "Drug-Drug Interaction Prediction Based on Knowledge Graph Embeddings and Convolutional-LSTM Network", proc. of The 10th ACM Conference on Bioinformatics, Computational Biology, and Health Informatics(ACM BCB), 2019.

In this paper, we proposed a new method for predicting potential DDIs by encompassing over 12,000 drug features from DrugBank, PharmGKB, and KEGG drugs with the help of knowledge graph(KGs).

In our pipeline, we extract feature vector representation of drugs from the KGs, using various embedding techniques such as RDF2Vec, TransE, KGloVe, SimplE, CrossE, and PyTorch-BigGraph(PBG). The embedded vectors are then used to train different prediction models.

Requirements

  • Python 3
  • PySpark
  • Scikit-learn
  • Keras
  • TensorFlow.

How to reproduce the results?

Example DDI prediction

An example of using the graph embeddings generated by RDF2Vec can be found in https://github.com/rezacsedu/DDI-prediction-KG-embeddings-Conv-LSTM/blob/master/Sample_DDI_Prediction_RDF2Vec.ipynb, which shows DDI prediction about 2500 drugs.

Knowledge graphs

Due to data sharing restrictions from DrugBank, KEGG, and PharmGKB, created RDF graphs are not publicly accessible. However, soon the SPARQL endpoint will be made public for querying. Please check back to http://cloud39.dbis.rwth-aachen.de:9999/blazegraph/#splash.

Graph embeddings

Already prepared embeddings can be downloaded from the following links with a password of '123':

Acknowledgement

Some concepts are based on https://github.com/rcelebi/GraphEmbedding4DDI by Remzi Celebi et al.

Citation request

If you use the code of this repository for your reserch, please consider citing the following paper:

@inproceedings{karim2019ddiconvlstm,
    title={Drug-Drug Interaction Prediction Based on Knowledge Graph Embeddings and Convolutional-LSTM Network},
    author={Md. Rezaul Karim, Michael Cochez, Joao Bosco Jares, Mamtaz Uddin, Stefan Decker, and Oya Beyan},
    booktitle={Proceedings of ACM BCB, ACM, New York, NY, USA, 10 pages},
    year={2019}
}

Contributing

For any questions, feel free to open an issue or contact at [email protected]

About

Drug-Drug Interaction Prediction Based on Knowledge Graph Embeddings and Convolutional-LSTM Network

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published