Weakly supervised classification of aortic valve malformations using unlabeled cardiac MRI sequences

Nat Commun. 2019 Jul 15;10(1):3111. doi: 10.1038/s41467-019-11012-3.

Abstract

Biomedical repositories such as the UK Biobank provide increasing access to prospectively collected cardiac imaging, however these data are unlabeled, which creates barriers to their use in supervised machine learning. We develop a weakly supervised deep learning model for classification of aortic valve malformations using up to 4,000 unlabeled cardiac MRI sequences. Instead of requiring highly curated training data, weak supervision relies on noisy heuristics defined by domain experts to programmatically generate large-scale, imperfect training labels. For aortic valve classification, models trained with imperfect labels substantially outperform a supervised model trained on hand-labeled MRIs. In an orthogonal validation experiment using health outcomes data, our model identifies individuals with a 1.8-fold increase in risk of a major adverse cardiac event. This work formalizes a deep learning baseline for aortic valve classification and outlines a general strategy for using weak supervision to train machine learning models using unlabeled medical images at scale.

Publication types

  • Research Support, N.I.H., Extramural
  • Research Support, Non-U.S. Gov't
  • Research Support, U.S. Gov't, Non-P.H.S.

MeSH terms

  • Aortic Valve / abnormalities*
  • Aortic Valve / diagnostic imaging
  • Aortic Valve / pathology
  • Heart Diseases / pathology
  • Heart Valve Diseases / diagnostic imaging
  • Heart Valve Diseases / pathology*
  • Humans
  • Machine Learning*
  • Magnetic Resonance Imaging
  • Neural Networks, Computer
  • Supervised Machine Learning