Denoising Autoencoders for Unsupervised Anomaly Detection in Brain MRI

Kascenas, A., Pugeault, N. and O'Neil, A. Q. (2022) Denoising Autoencoders for Unsupervised Anomaly Detection in Brain MRI. In: Medical Imaging with Deep Learning (MIDL 2022), Zurich, Switzerland, 6-8 July 2022,

[img] Text
277118.pdf - Published Version

2MB

Publisher's URL: https://proceedings.mlr.press/v172/kascenas22a.html

Abstract

Pathological brain lesions exhibit diverse appearance in brain images, making it difficult to design specialized detection solutions due to the lack of comprehensive data and annotations. Thus, in this work we tackle unsupervised anomaly detection, using only healthy data for training with the aim of detecting unseen anomalies at test time. Many current approaches employ autoencoders with restrictive architectures (i.e. containing information bottlenecks) that tend to give poor reconstructions of not only the anomalous but also the normal parts of the brain. Instead, we investigate classical denoising autoencoder models that do not require bottlenecks and can employ skip connections to give high resolution fidelity. We design a simple noise generation method of upscaling low-resolution noise that enables high-quality reconstructions, reducing false positive noise in reconstruction errors. We find that with appropriate noise generation, denoising autoencoder reconstruction errors generalize to hyperintense lesion segmentation and can reach state of the art performance for unsupervised tumor detection in brain MRI data, beating more complex methods such as variational autoencoders. We believe this provides a strong and easy-to-implement baseline for further research into unsupervised anomaly detection.

Item Type:Conference Proceedings
Status:Published
Refereed:Yes
Glasgow Author(s) Enlighten ID:Pugeault, Dr Nicolas and Kascenas, Antanas
Authors: Kascenas, A., Pugeault, N., and O'Neil, A. Q.
College/School:College of Science and Engineering
College of Science and Engineering > School of Computing Science
ISSN:2640-3498
Copyright Holders:Copyright © 2022 A. Kascenas, N. Pugeault and A.Q. O’Neil
First Published:First published in Proceedings of The 5th International Conference on Medical Imaging with Deep Learning, PMLR 172:653-664
Publisher Policy:Reproduced in accordance with publisher policy
Related URLs:

University Staff: Request a correction | Enlighten Editors: Update this record