An Efficient Capsule-based Network for 2D Left Ventricle Segmentation in Echocardiography Images

The segmentation of cardiac chambers is essential for the clinical diagnosis and treatment of cardiovascular diseases. It is demonstrated that in cardiac disease, the left ventricle (LV) is extensively involved. Therefore, segmentation of the LV in echocardiographic images is critical for the precise evaluation of factors that influence cardiac function such as LV volume, ejection fraction, and LV mass. Although these measurements could be obtained by manual segmentation of the LV, it would be time-consuming and inaccurate because of the poor quality and low contrast of these images. Convolutional neural networks, commonly referred to as CNNs, have emerged as a highly favored deep learning technique for medical image segmentation. Despite their popularity, the pooling layers in CNNs ignore the spatial information and do not consider the part-whole hierarchy relationships. Furthermore, they require a large training dataset and a large number of parameters. Therefore, Capsule Networks are proposed to address the CNNs limitations. In this study, for the first time, an optimized capsule-based network for object segmentation called SegCaps is proposed to achieve accurate LV segmentation on echocardiography images applied to the CAMUS dataset. The result was compared against the standard 2D-UNet. The modified SegCaps and 2D-UNet achieved an average Dice similarity coefficient (DSC) of 84.48% and 83.28% on LV segmentation, respectively. The capabilities of the CapsNet led to an improvement of 1.44% in DSC with 92.77% fewer parameters than the U-Net. The results indicate that the proposed method leads to accurate and efficient LV segmentation.Clinical Relevance— From a clinical point of view, our findings lead to more precise evaluations of critical cardiac parameters, including ejection fraction as well as left ventricle volume at end-diastole and end-systole.