The Application of Deep Learning for the Segmentation and Classification of Coronary Arteries

Şerife Kaba1, Huseyin Haci2, Ali Isin3

  • 1Department of Biomedical Engineering, Near East University, TRNC Mersin 10, Nicosia 99138, Turkey.

Insights

This study introduces deep learning for coronary artery disease (CAD) detection. U-Net excels at artery segmentation, while DenseNet201 accurately classifies stenosis, aiding cardiologists and reducing errors.

Area of Science:

  • Cardiology
  • Medical Imaging Analysis
  • Artificial Intelligence in Healthcare

Background:

  • Coronary artery disease (CAD) is a leading global cause of mortality.
  • Accurate detection of coronary artery stenosis is critical for effective treatment.
  • Current manual interpretation of coronary angiography is prone to high workloads, delays, and human error.

Purpose of the Study:

  • To develop and evaluate deep learning models for automated segmentation and classification of coronary arteries.
  • To compare the performance of U-Net, ResUNet-a, and UNet++ for coronary artery segmentation.
  • To assess the efficacy of DenseNet201, EfficientNet-B0, Mobilenet-v2, ResNet101, and Xception for coronary artery stenosis classification.

Main Methods:

  • Coronary artery segmentation was performed using U-Net, ResUNet-a, and UNet++.
  • Stenosis classification was conducted using DenseNet201, EfficientNet-B0, Mobilenet-v2, ResNet101, and Xception.
  • Model performance was evaluated using metrics such as Dice score, Jaccard Index, accuracy, specificity, PPV, Cohen's Kappa, and AUC.

Main Results:

  • U-Net achieved the highest segmentation performance with a 0.8467 Dice score and 0.7454 Jaccard Index.
  • DenseNet201 demonstrated superior classification performance with 0.9000 accuracy, 0.9833 specificity, 0.9556 PPV, 0.7746 Cohen's Kappa, and 0.9694 AUC.

Conclusions:

  • Automated deep learning approaches significantly enhance the accuracy and efficiency of coronary artery stenosis detection.
  • U-Net is a highly effective model for coronary artery segmentation.
  • DenseNet201 shows strong potential for reliable classification of coronary artery stenosis, aiding clinical decision-making.