@inproceedings{
rosnati2023robust,
title={Robust semi-supervised segmentation with timestep ensembling diffusion models},
author={Margherita Rosnati and M{\'e}lanie Roschewitz and Ben Glocker},
booktitle={Deep Generative Models for Health Workshop NeurIPS 2023},
year={2023},
url={https://openreview.net/forum?id=OBzxDn1XEy}
}
Medical image segmentation is challenging due to limited data and annotations. Denoising diffusion probabilistic models (DDPM) show promise in modelling natural image distributions and are successfully applied in medical imaging. Our research focuses on semi-supervised image segmentation using diffusion models' latent representations and addressing domain generalisation. We found that optimal performance depends on the choice of diffusion steps and ensembling. Our model outperformed in domain-shifted settings while remaining competitive within domain, highlighting DDPMs' potential for medical image segmentation.
Robust semi-supervised segmentation with timestep ensembling diffusion models
Margherita Rosnati Mélanie Roschewitz Ben Glocker