Heart and great vessels segmentation in congenital heart disease via CNN and conditioned energy function

Jiaxuan Liu1, Bolun Zeng1, Xiaojun Chen2,3

  • 1Institute of Biomedical Manufacturing and Life Quality Engineering, School of Mechanical Engineering, Shanghai Jiao Tong University, Shanghai, China.

Insights

This study introduces a novel two-stage AI method for segmenting the heart and great vessels in CT scans of congenital heart disease (CHD). The approach improves diagnostic accuracy for cardiac anomalies.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Cardiology

Background:

  • Accurate segmentation of the heart and great vessels in CT images is crucial for diagnosing congenital heart disease (CHD).
  • Diverse CHD abnormalities pose significant challenges to existing segmentation methods.
  • Clinical assessment and diagnosis of CHD rely heavily on precise cardiac imaging analysis.

Purpose of the Study:

  • To develop and validate a novel two-stage segmentation approach for the heart and great vessels in CT images of CHD.
  • To address the challenges posed by diverse cardiac anomalies in CHD segmentation.
  • To enhance the accuracy and reliability of cardiac structure segmentation for improved CHD diagnosis.

Main Methods:

  • A two-stage segmentation strategy was proposed, combining a Convolutional Neural Network (CNN) with a gated self-attention mechanism and a conditioned energy function.
  • The initial stage uses a CNN for segmenting five heart structures and two major vessels.
  • The second stage refines the segmentation of the pulmonary artery and aorta using a conditioned energy function to ensure vascular continuity.

Main Results:

  • The proposed method was evaluated on a public dataset of 110 3D CT volumes with 16 CHD variants.
  • It demonstrated superior performance compared to U-Net, V-Net, Unetr, and dynUnet, with improvements in Dice Coefficient (DSC), Intersection over Union (IOU), and Hausdorff Distance (HD95).
  • Enhancements in segmentation metrics for both heart structures and great vessels confirmed the method's efficacy.

Conclusions:

  • The study validates the effectiveness of the proposed two-stage segmentation method for CHD.
  • Precise segmentation of the heart and great vessels is clinically relevant for CHD diagnosis and treatment.
  • The findings highlight the potential of advanced AI techniques in improving cardiovascular imaging analysis.
Abstract