Two-Stage CNN Whole Heart Segmentation Combining Image Enhanced Attention Mechanism and Metric Classification

Xuchu Wang1, Fusheng Wang2, Yanmin Niu3

  • 1Key Laboratory of Optoelectronic Technology and Systems of Ministry of Education, College of Optoelectronic Engineering, Chongqing University, Chongqing, 400040, China. xcwang@cqu.edu.cn.

Journal of Digital Imaging
|September 29, 2022
PubMed

Insights

This study introduces a novel two-stage convolutional neural network (CNN) for cardiac MRI segmentation. The method enhances tissue feature extraction and boundary refinement for improved computer-aided cardiovascular diagnosis.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Cardiovascular Diagnostics

Background:

  • Accurate segmentation of cardiac tissues and organs in magnetic resonance imaging (MRI) is crucial for computer-aided cardiovascular diagnosis.
  • Challenges include complex tissue distribution, low feature discriminability, and large organ sizes in cardiac MRI slices.
  • Existing methods struggle with segmenting subtle and adherent tissue boundaries.

Purpose of the Study:

  • To develop an advanced segmentation method for cardiac MRI to address current segmentation challenges.
  • To improve the accuracy and robustness of automated cardiac tissue and organ segmentation.
  • To enhance computer-aided diagnosis through precise delineation of cardiac structures.

Main Methods:

  • A two-stage convolutional neural network (CNN) segmentation approach combining a Log-Gabor filter attention mechanism and metric classification.
  • Log-Gabor filterbank enhances tissue texture and contour information.
  • Spatial and channel attention mechanisms with varying kernel sizes adaptively extract features and focus on discriminative information.
  • A metric classification network refines segmentation of difficult boundaries.

Main Results:

  • The proposed method was validated on a cardiac MRI dataset for segmenting 7 cardiac tissues.
  • The approach demonstrated effectiveness in enhancing feature extraction and refining segmentation boundaries.
  • Achieved competitive performance compared to existing deep learning-based segmentation models.

Conclusions:

  • The proposed two-stage CNN method effectively addresses challenges in cardiac MRI segmentation.
  • The integration of Log-Gabor filters, attention mechanisms, and metric classification improves segmentation accuracy, especially for intricate boundaries.
  • This method shows significant potential for advancing computer-aided cardiovascular diagnosis.

Related Concept Videos