Retraining Convolutional Neural Networks for Specialized Cardiovascular Imaging Tasks: Lessons from Tetralogy of Fallot

Animesh Tandon, Navina Mohan, Cory Jensen, Barbara E.U. Burkhardt, Vasu Gooty, Daniel A. Castellanos, Paige L. McKenzie, Riad Abou Zahr, Abhijit Bhattaru, Mubeena Abdulkarim, Alborz Amir-Khalili, Alireza Sojoudi, Stephen M. Rodriguez, Jeanne Dillenbeck, Gerald F. Greil, Tarique Hussain

Research output: Contribution to journalArticlepeer-review

Abstract

Ventricular contouring of cardiac magnetic resonance imaging is the gold standard for volumetric analysis for repaired tetralogy of Fallot (rTOF), but can be time-consuming and subject to variability. A convolutional neural network (CNN) ventricular contouring algorithm was developed to generate contours for mostly structural normal hearts. We aimed to improve this algorithm for use in rTOF and propose a more comprehensive method of evaluating algorithm performance. We evaluated the performance of a ventricular contouring CNN, that was trained on mostly structurally normal hearts, on rTOF patients. We then created an updated CNN by adding rTOF training cases and evaluated the new algorithm’s performance generating contours for both the left and right ventricles (LV and RV) on new testing data. Algorithm performance was evaluated with spatial metrics (Dice Similarity Coefficient (DSC), Hausdorff distance, and average Hausdorff distance) and volumetric comparisons (e.g., differences in RV volumes). The original Mostly Structurally Normal (MSN) algorithm was better at contouring the LV than the RV in patients with rTOF. After retraining the algorithm, the new MSN + rTOF algorithm showed improvements for LV epicardial and RV endocardial contours on testing data to which it was naïve (N = 30; e.g., DSC 0.883 vs. 0.905 for LV epicardium at end diastole, p < 0.0001) and improvements in RV end-diastolic volumetrics (median %error 8.1 vs 11.4, p = 0.0022). Even with a small number of cases, CNN-based contouring for rTOF can be improved. This work should be extended to other forms of congenital heart disease with more extreme structural abnormalities. Aspects of this work have already been implemented in clinical practice, representing rapid clinical translation. The combined use of both spatial and volumetric comparisons yielded insights into algorithm errors.

Original languageEnglish (US)
Pages (from-to)578-589
Number of pages12
JournalPediatric Cardiology
Volume42
Issue number3
DOIs
StatePublished - Mar 2021

Keywords

  • Congenital heart disease
  • Convolutional neural network
  • Machine learning
  • Tetralogy of fallot
  • Ventricular contouring

ASJC Scopus subject areas

  • Pediatrics, Perinatology, and Child Health
  • Cardiology and Cardiovascular Medicine

Fingerprint Dive into the research topics of 'Retraining Convolutional Neural Networks for Specialized Cardiovascular Imaging Tasks: Lessons from Tetralogy of Fallot'. Together they form a unique fingerprint.

Cite this