xingbpshen / nested-diffusion

The official implementation of the paper "Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensembles".

Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensembles

This repository contains the official implementation of the paper:

Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensembles
Xing Shen, Hengguan Huang, Brennan Nichyporuk, Tal Arbel
arXiv Preprint, Oct 2023
Paper / BibTeX

Setup

coming soon

Training

coming soon

Evaluation

coming soon

Acknowledgement

This work is founded by the Natural Sciences and Engineering Research Council (NSERC) of Canada, the Canadian Institute for Advanced Research (CIFAR) Artificial Intelligence Chairs program, the Mila - Quebec AI Institute technology transfer program, Calcul Quebec, and the Digital Research Alliance of Canada. This repository contains code adapted from repositories CARD and SEViT. We thank to the above repositories' authors for their great work.

Citation

If you find this repository useful in your research, please cite our paper:

@article{shen2023improving,
  title={Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensembles},
  author={Shen, Xing and Huang, Hengguan and Nichyporuk, Brennan and Arbel, Tal},
  journal={arXiv preprint arXiv:2310.15952},
  year={2023}
}