Unsupervised pathology detection in medical images using conditional variational autoencoders.

Hristina Uzunova, Sandra Schultz, Heinz Handels, Jan Ehrhardt
Author Information
  1. Hristina Uzunova: Institute of Medical Informatics, University of Lübeck, Lübeck, Germany. uzunova@imi.uni-luebeck.de. ORCID
  2. Sandra Schultz: Institute of Medical Informatics, University of Lübeck, Lübeck, Germany.
  3. Heinz Handels: Institute of Medical Informatics, University of Lübeck, Lübeck, Germany.
  4. Jan Ehrhardt: Institute of Medical Informatics, University of Lübeck, Lübeck, Germany.

Abstract

PURPOSE: Pathology detection in medical image data is an important but a rather complicated task. In particular, the big variability of the pathologies is a challenge to automatic detection methods and even to machine learning methods. Supervised algorithms would usually learn the appearance of a single pathological structure based on a large annotated dataset. As such data is not usually available, especially in large amounts, in this work we pursue a different unsupervised approach.
METHODS: Our method is based on learning the entire variability of healthy data and detect pathologies by their differences to the learned norm. For this purpose, we use conditional variational autoencoders which learn the reconstruction and encoding distribution of healthy images and also have the ability to integrate certain prior knowledge about the data (condition).
RESULTS: Our experiments on different 2D and 3D datasets show that the approach is suitable for the detection of pathologies and deliver reasonable Dice coefficients and AUCs. Also this method can estimate missing correspondences in pathological images and thus can be used as a pre-step to a registration method. Our experiments show improving registration results on pathological data when using this approach.
CONCLUSIONS: Overall the presented approach is suitable for a rough pathology detection in medical images and can be successfully used as a preprocessing step to other image processing methods.

Keywords

References

  1. Neuroimage. 2008 Feb 1;39(3):1064-80 [PMID: 18037310]
  2. Neuroimage. 2009 Jul 1;46(3):786-802 [PMID: 19195496]
  3. Phys Med Biol. 2014 Aug 7;59(15):4247-60 [PMID: 25017631]
  4. IEEE Trans Med Imaging. 2015 Oct;34(10):1993-2024 [PMID: 25494501]
  5. Med Image Anal. 2017 Jan;35:250-269 [PMID: 27475911]
  6. IEEE Trans Med Imaging. 2017 Oct;36(10):2057-2067 [PMID: 28574348]
  7. Med Image Anal. 2017 Dec;42:60-88 [PMID: 28778026]

MeSH Term

Algorithms
Area Under Curve
Brain Neoplasms
Glioblastoma
Humans
Image Processing, Computer-Assisted
Imaging, Three-Dimensional
Lung
Machine Learning
Pathology
Phantoms, Imaging

Word Cloud

Created with Highcharts 10.0.0detectiondataapproachimagesmedicalpathologiesmethodspathologicalmethodvariationalcanregistrationpathologyimagevariabilitylearningusuallylearnbasedlargedifferenthealthyconditionalautoencodersexperimentsshowsuitableusedusingUnsupervisedPURPOSE:PathologyimportantrathercomplicatedtaskparticularbigchallengeautomaticevenmachineSupervisedalgorithmsappearancesinglestructureannotateddatasetavailableespeciallyamountsworkpursueunsupervisedMETHODS:entiredetectdifferenceslearnednormpurposeusereconstructionencodingdistributionalsoabilityintegratecertainpriorknowledgeconditionRESULTS:2D3DdatasetsdeliverreasonableDicecoefficientsAUCsAlsoestimatemissingcorrespondencesthuspre-stepimprovingresultsCONCLUSIONS:OverallpresentedroughsuccessfullypreprocessingstepprocessingConditionalautoencoderImage

Similar Articles

Cited By