Segmenting computed tomograms for cardiac ablation using machine learning leveraged by domain knowledge encoding

Figures

• 

  Figure 1. Concept and overview. (A) Conventional machine learning (top) can learn patterns in complex data, but requires laborious manual labeling, in large datasets which may be difficult to obtain. Conversely, our proposed approach (bottom) used natural intelligence to replace manual labeling with anatomical concepts encoded mathematically of domain knowledge, to learn rapidly from small datasets. (B) We applied mathematical encoding to segment heart CT scans via ML of small datasets. We represented heart structures as geometric primitives (“virtual dissection”). This was used to train ML on a small dataset (N = 20) and was able to accurately segment hearts in 2 larger cohorts from different institutions (N = 100, 60). In a prospective study (N = 42), the model segmented cardiac CT scans faster, but as accurately as experts. Acronyms: LA, left atrium; LSPV, left superior pulmonary vein; LIPV, left inferior pulmonary vein; RSPV, right superior pulmonary vein; RIPV, right inferior pulmonary vein; LAA, left atrial appendage.

• 

  Figure 2. Virtual Dissection algorithm. (A) The detailed pipeline. (B) The progress of the iterative erosion. The automatically selected iteration for erosion is highlighted in red. (C) The progress of the iterative dilation. The automatically selected iteration for dilation is highlighted in red. Acronyms: LA, left atrium; LSPV, left superior pulmonary vein; LIPV, left inferior pulmonary vein; RSPV, right superior pulmonary vein; RIPV, right inferior pulmonary vein; LAA, left atrial appendage.

• 

  Figure 3. Virtual dissection performance. (A) Representative samples of digital atria geometrically parsed by un-optimized algorithm. (B) Bland-Altman plots of the centroid-to-boundary of un-optimized algorithm vs. experts in 6 digital atria. After optimizing Virtual Dissection with N = 5 patient cases from the development cohort, (C) Representative patient atria from optimized algorithm in independent Test cohort (N = 100). (D) Bland–Altman plots of the centroid-to-boundary distance of optimized algorithm vs. experts in the Test cohort. (E) Success rate of virtual dissection algorithm using N={0, 5, 10, 20, 30} cases. Acronyms: LSPV, left superior pulmonary vein; LIPV, left inferior pulmonary vein; RSPV, right superior pulmonary vein; RIPV, right inferior pulmonary vein; LAA, left atrial appendage.

• 

  Figure 4. Comparison between the ML model predicted CT segmentation (left) and ground truth manual outlining (right) overlaid on the input CT scans in representative samples selected using 25th, 50th and 75th percentiles of segmentation accuracy in an independent test cohort (N = 100). Our ML model effectively captured the LA geometry, highlighting key features of PVs, LAA, and their ostia. The mitral valve plane represented in the 50th- and 25th- percentile samples showed slight variation between ML prediction and manual labeling, likely from limited image quality. Slight differences in PV measurements were found in the 25th-percentile sample, which may not be clinically relevant. Acronyms: LA, left atrium; LSPV, left superior pulmonary vein; LIPV, left inferior pulmonary vein; RSPV, right superior pulmonary vein; RIPV, right inferior pulmonary vein; LAA, left atrial appendage.

• 

  Figure 5. Accuracy CT segmentation using ML of optimized virtual dissection in two test cohorts (A) dice score of ML-based CT segmentation in the internal test cohort (N = 100; left) and an external test cohort from a different institution with different CT scanners (N = 60; right). (B) Boundary surface distances between ML-prediction and expert labelling in the Test Dataset (N = 100). (C) and (D) are Bland–Altman plots and linear regression plots of the centroid-to-boundary distance in the Test Dataset (N = 100). Acronyms: LSPV, left superior pulmonary vein; LIPV, left inferior pulmonary vein; RSPV, right superior pulmonary vein; RIPV, right inferior pulmonary vein; LAA, left atrial appendage.

• 

  Figure 6. Prospective segmentation of cardiac CT scans in 42 consecutive patients undergoing AF ablation by virtual-dissection trained ML vs. experts. (A,B) Virtual dissection trained ML significantly shortens segmentation time compared to experts. (C) Box plots of Dice similarity coefficient between ML and experts were similar. (D) and (E) LA volume and LA sphericity index marked by Virtual Dissection (red cross) accurately tracks the mean between experts (black cross).