Deep diffusion models for anomaly detection
Recrutement en cours/passé: 
Recrutement passé


Deep Learning, Weakly supervised segmentation, Diffusion models

Scientific context

Deep learning is now an established approach for medical image processing, providingstate of the art results for image segmentation. However, best results are obtained with supervised learning,requiring a large annotated dataset. Especially for medical image segmentation, annotation is a tediousand time consuming task, requiring experienced clinician as annotator. In a previous work, we propose touse constrained learning with attribution maps to segment tumor or multiple sclerosis (MS) lesions in aweakly supervised way. Deep unsupervised learning is often based on generative models such as generativeadversarial network (GAN) or variational autoencoder (VAE) but these models has strong limitations: GANare notoriously difficult to train and VAE produces blurry images. Recently, diffusion model [Song et al] haveshown there ability to generate images that can be highly resolved and trained in a stable manner. Someauthors already proposed to use them for anomaly detection on medical images [Wolleb et al].


The objective of the internship would be to propose a new weakly supervised anomaly detectionmethod based on diffusion models. A work program could be: 1/ deeply understand the mechanism ofdiffusion models, 2/ reproduce the results of [Wolleb et al] and apply them to different data and differentpathology (MS), 3/ combine the methods of [Wargnier et al] and [Wolleb et al] and evaluate the performanceof the combined method. This work program can evolve based on the results obtained during the internshipand new ideas of the intern.


Several datasets are already available for use in the lab.


The successful candidate is expected to be autonomous, to show strong motivation and in-terest in multidisciplinary research (image processing and machine learning for medical applications), to behighly proficient in python (pytorch is a plus) and to have a background in either• Image analysis• deep learning• or applied mathematicsInterested applicants are required to send a cover letter, resume and any other relevant documents (ref-erence letter, recent transcripts of marks,...) to: michael.sdika[at], valentine.wargnier[at] and thomas.grenier[at]


  1. J. Song et al, Denoising Diffusion Implicit Models, ICLR 2021
  2. J. Wolleb et al, Diffusion Models for Medical Anomaly Detection, MICCAI 2022,
  3. V. Wargnier Dauchelle, A Weakly-Supervised Gradient Attribution Constraint for Interpretable Classification and Anomaly Detection, et al, IEEE TMI (submitted)