Unsupervised Detection of Fetal Brain Anomalies using Denoising Diffusion Models
Authors:
Markus Ditlev Sjøgren Olsen,
Jakob Ambsdorf,
Manxi Lin,
Caroline Taksøe-Vester,
Morten Bo Søndergaard Svendsen,
Anders Nymark Christensen,
Mads Nielsen,
Martin Grønnebæk Tolsgaard,
Aasa Feragen,
Paraskevas Pegios
Abstract:
Congenital malformations of the brain are among the most common fetal abnormalities that impact fetal development. Previous anomaly detection methods on ultrasound images are based on supervised learning, rely on manual annotations, and risk missing underrepresented categories. In this work, we frame fetal brain anomaly detection as an unsupervised task using diffusion models. To this end, we empl…
▽ More
Congenital malformations of the brain are among the most common fetal abnormalities that impact fetal development. Previous anomaly detection methods on ultrasound images are based on supervised learning, rely on manual annotations, and risk missing underrepresented categories. In this work, we frame fetal brain anomaly detection as an unsupervised task using diffusion models. To this end, we employ an inpainting-based Noise Agnostic Anomaly Detection approach that identifies the abnormality using diffusion-reconstructed fetal brain images from multiple noise levels. Our approach only requires normal fetal brain ultrasound images for training, addressing the limited availability of abnormal data. Our experiments on a real-world clinical dataset show the potential of using unsupervised methods for fetal brain anomaly detection. Additionally, we comprehensively evaluate how different noise types affect diffusion models in the fetal anomaly detection domain.
△ Less
Submitted 7 August, 2024;
originally announced August 2024.
