f-AnoGAN: Fast unsupervised anomaly detection with generative adversarial networks

Med Image Anal. 2019 May:54:30-44. doi: 10.1016/j.media.2019.01.010. Epub 2019 Jan 31.

Abstract

Obtaining expert labels in clinical imaging is difficult since exhaustive annotation is time-consuming. Furthermore, not all possibly relevant markers may be known and sufficiently well described a priori to even guide annotation. While supervised learning yields good results if expert labeled training data is available, the visual variability, and thus the vocabulary of findings, we can detect and exploit, is limited to the annotated lesions. Here, we present fast AnoGAN (f-AnoGAN), a generative adversarial network (GAN) based unsupervised learning approach capable of identifying anomalous images and image segments, that can serve as imaging biomarker candidates. We build a generative model of healthy training data, and propose and evaluate a fast mapping technique of new data to the GAN's latent space. The mapping is based on a trained encoder, and anomalies are detected via a combined anomaly score based on the building blocks of the trained model - comprising a discriminator feature residual error and an image reconstruction error. In the experiments on optical coherence tomography data, we compare the proposed method with alternative approaches, and provide comprehensive empirical evidence that f-AnoGAN outperforms alternative approaches and yields high anomaly detection accuracy. In addition, a visual Turing test with two retina experts showed that the generated images are indistinguishable from real normal retinal OCT images. The f-AnoGAN code is available at https://github.com/tSchlegl/f-AnoGAN.

Keywords: Anomaly detection; Optical coherence tomography; Unsupervised learning; Wasserstein generative adversarial network.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Diagnostic Techniques, Ophthalmological*
  • Humans
  • Image Processing, Computer-Assisted / methods*
  • Information Theory
  • Neural Networks, Computer*
  • Retina / diagnostic imaging*
  • Tomography, Optical Coherence*