Automated image segmentation for cardiac septal defects based on contour region with convolutional neural networks: A preliminary study

Ria Nova, Siti Nurmaini, Radiyati Umi Partan, Sukman Tulus Putra

Research output: Contribution to journalArticlepeer-review

Abstract

Echocardiogram examination is important for diagnosing cardiac septal defects. With the development of AI-based technology, an echocardiogram examination previously performed manually by cardiologists can be done automatically. Automatic segmentation of cardiac septal defects can help a physician to make an initial diagnosis before referring a pediatric cardiologist for further treatment. In previous studies, automatic object segmentation using convolutional neural networks (CNNs) was one of the DL applications that have been developed for cardiac abnormalities. In this study, we propose a CNN-based U-Net architecture to automatically segment the cardiac chamber to detect abnormalities (holes) in the heart septum. In this study, echocardiogram examinations were performed on atrial septal defects (ASDs), ventricular septal defects (VSDs), atrioventricular septal defects (AVSDs), and normal hearts with patients undergoing echocardiogram examination at Moh Hoesin Hospital in Palembang. The results show that even for the relatively small number of datasets, the proposed technique can produce superior performance in the detection of the cardiac septal defects. Using the proposed segmentation model for four classes produces a pixel accuracy of 99.15%, mean intersection over union (IoU) of 94.69%, mean accuracy of 97.73%, sensitivity of 96.02%, and F1 score of 94.88%, respectively. The plots of the loss and accuracy curve show that all the errors were small, with accuracy rates reaching 99.05%, 98.62%, 99.39%, and 98.97% for ASD, VSD, AVSD, and normal heart, respectively. The comparison accuracy of contour prediction for U-Net was 99.01%, while V-Net was 93.70%. This shows that the U-Net has better accuracy than the V-Net model architecture. It can be proven that the architecture of CNNs has been successful in segmenting the cardiac chamber to detect defects in the heart septum and support the work of cardiologists.

Original languageEnglish
Article number100601
JournalInformatics in Medicine Unlocked
Volume24
DOIs
Publication statusPublished - Jan 2021

Keywords

  • Cardiac septal defect
  • CNNs
  • Contour segmentation
  • Deep learning

Fingerprint

Dive into the research topics of 'Automated image segmentation for cardiac septal defects based on contour region with convolutional neural networks: A preliminary study'. Together they form a unique fingerprint.

Cite this