Skip to content

An interpretable deep learning model for cancer classification using high-dimensional omics data

License

Notifications You must be signed in to change notification settings

zhangxiaoyu11/XOmiVAE

 
 

Repository files navigation

XOmiVAE

GitHub license Safe GitHub stars GitHub forks

XOmiVAE: An Interpretable Deep Learning Framework for Cancer Classification using High-dimensional Omics Data

Xiaoyu Zhang ([email protected])

Data Science Institute, Imperial College London

Introduction

  • XOmiVAE is a novel interpretable deep learning model for cancer classification using high-dimensional omics data.
  • XOmiVAE provides contribution of each input molecular feature and latent dimension to the prediction.
  • XOmiVAE is able to explain unsupervised clusters produced by the VAE clustering part of the network.
  • XOmiVAE explanations of the downstream prediction were evaluated by biological annotation and literature, which aligned with current domain knowledge.
  • XOmiVAE shows great potential for novel biomedical knowledge discovery from deep learning models.

For examples of how to implement the code, see main.py. The explainable functions are in omiShapExplainer.py.

Paper Link: Briefings in Bioinformatics

Citation

If you use this code in your research, please cite our paper.

@Article{XOmiVAE2021,
    AUTHOR = {Withnell, Eloise and Zhang, Xiaoyu and Sun, Kai and Guo, Yike},
    TITLE = {XOmiVAE: an interpretable deep learning model for cancer classification using high-dimensional omics data},
    JOURNAL = {Briefings in Bioinformatics},
    YEAR = {2021},
    ISSN = {1477-4054},
    DOI = {10.1093/bib/bbab315},
    URL = {https://doi.org/10.1093/bib/bbab315},
}

OmiEmbed

Please also check our multi-task deep learning framework for multi-omics data: OmiEmbed

License

This source code is licensed under the MIT license.

About

An interpretable deep learning model for cancer classification using high-dimensional omics data

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 56.8%
  • HTML 41.9%
  • Python 1.1%
  • JavaScript 0.1%
  • C++ 0.1%
  • Batchfile 0.0%