This repo is the official PyTorch implementation of "KGETCDA: an efficient representation learning framework based on knowledge graph encoder from transformer for predicting circRNA-disease associations"
KGETCDA (Knowledge Graph Encoder from Transformer for predicting CircRNA-Disease Associations) is an efficient knowledge-based representation learning framework. Built upon knowledge graph and Transformer, KGETCDA can effectively produce high-quality embeddings with accurately captured low-order and high-order interaction information.
KGETCDA consistently achieves remarkable performance on three datasets (Dataset1---a small dataset focusing on non-cancer used in previous works; Dataset2---a larger heterogeneous dataset constructed by us; Dataset3---a small dataset focusing on cancer used in previous works)
We also provide a user-friendly interactive web-based platform (named HNRBase), which is publicly available at http://lab-fly.site/KGETCDA.
The code has been tested running under Python 3.9.15. The required packages are as follows:
- numpy == 1.24.1
- pandas == 1.5.2
- torch == 1.11.0
The expected structure of files is:
── KGETCDA
├── datasets
│ ├── Dataset1
│ ├── Dataset2
│ └── Dataset3
├── data_loader
│ └── loader_KGETCDA.py
├── models
│ └── KGETCDA.py
├── parsers
│ └── parser_KGETCDA.py
├── utils
│ ├── helper_init.py
│ ├── helper_log.py
│ ├── helper_metrics.py
│ └── helper_model.py
├── main.py
└── test.py
We use three datasets here, and provide 4 entities (circRNA, miRNA, lncRNA, disease) and 5 relations (circRNA-disease, miRNA-disease, lncRNA-disease, circRNA-miRNA, miRNA-lncRNA) file and all pairs. The summary information is listed as follows:
| Dataset | circ-dis | mir-dis | lnc-dis | circ-mir | mir-lnc | total |
|---|---|---|---|---|---|---|
| Dataset1 | 346 | 106 | 527 | 146 | 202 | 1327 |
| Dataset2 | 1399 | 10154 | 3280 | 1129 | 9506 | 25468 |
| Dataset3 | 647 | 732 | 1066 | 756 | 308 | 3509 |
-
entity.txt- All entities file.
- Each line is an entity with its ID: (
nameandID).
-
relation.txt- All relations file.
- Each line is a relation with its ID: (
nameandID).
-
pair_circ_dis.txt- All circRNA-disease positve associations file.
- Each line is a circRNA name with its positive interactions with diseases: (
nameandname).
-
pair_mir_dis.txt- All miRNA-disease positve associations file.
- Each line is a miRNA name with its positive interactions with diseases: (
nameandname).
-
pair_lnc_dis.txt- All lncRNA-disease positve associations file.
- Each line is a lncRNA name with its positive interactions with diseases: (
nameandname).
-
pair_circ_mir.txt- All circRNA-miRNA positve associations file.
- Each line is a circRNA name with its positive interactions with miRNAs: (
nameandname).
-
pair_mir_lnc.txt- All miRNA-lncRNA positve associations file.
- Each line is a miRNA name with its positive interactions with lncRNAs: (
nameandname).
You can directly run the above model KGETCDA.
We also recommend users use our KGETCDA Webserver (http://lab-fly.site/KGETCDA), which is user-friendly and easy to use. Consisting of 4 core functions (intelligent search and browse, model prediction, information visualization, and advanced interaction), our web-based platform enables novel visualization, accessible resources and user-friendly interaction. Everyone could upload or typein the candidate circRNAs or diseases of interest in our web without further installation, our backend server will calculate and give the prediction results to the user. Users can also choose to download the predict csv file results.
In this paper, we compare our model with 8 SOTAs including: GMNN2CD, KGANCDA, RNMFLP, AE-RF, DMFCDA, CD-LNLP, RWR, KATZHCDA, which are compared under the same experiment settings. The parameters of the other 8 models maintain consistency with their original papers.
-
GMNN2CD (2022)
- Proposed in GMNN2CD: identification of circRNA–disease associations based on variational inference and graph Markov neural networks, Bioinformatics 2022.
-
KGANCDA (2022)
- Proposed in KGANCDA: predicting circRNA-disease associations based on knowledge graph attention network, Briefings in bioinformatics 2022.
-
RNMFLP (2022)
- Proposed in RNMFLP: Predicting circRNA–disease associations based on robust nonnegative matrix factorization and label propagation, Briefings in bioinformatics 2022.
-
AE-RF (2021)
- Proposed in Inferring Potential CircRNA-Disease Associations via Deep Autoencoder-Based Classification, Molecular diagnosis & therapy 2021.
-
DMFCDA (2021)
- Proposed in Deep Matrix Factorization Improves Prediction of Human CircRNA-Disease Associations, IEEE Journal of Biomedical and Health Informatics 2021.
-
CD-LNLP (2019)
- Proposed in Predicting CircRNA-Disease Associations Through Linear Neighborhood Label Propagation Method, IEEE Access 2019.
-
RWR (2019)
- Proposed in A Model Based on Random Walk with Restart toPredict CircRNA-Disease Associations onHeterogeneous Network, IEEE/ACM International Conference on Advances in Social Networks Analysis and Mining (ASONAM) 2019.
-
KATZHCDA (2018)
- Proposed in Prediction of CircRNA-Disease Associations Using KATZ Model Based on Heterogeneous Networks, Biomolecules 2018.
If you use this model, please cite as follows:
@article{10.1093/bib/bbad292,
author = {Wu, Jinyang and Ning, Zhiwei and Ding, Yidong and Wang, Ying and Peng, Qinke and Fu, Laiyi},
title = {KGETCDA: an efficient representation learning framework based on knowledge graph encoder from transformer for predicting circRNA-disease associations},
journal = {Briefings in Bioinformatics},
pages = {bbad292},
year = {2023},
month = {08},
issn = {1477-4054},
doi = {10.1093/bib/bbad292},
eprint = {https://academic.oup.com/bib/advance-article-pdf/doi/10.1093/bib/bbad292/51117888/bbad292.pdf},
}