RIANet: Recurrent interleaved attention network for cardiac MRI segmentation

Abstract

BackgroundSegmentation of anatomical structures of the heart from cardiac magnetic resonance images (MRI) has a significant impact on the quantitative analysis of the cardiac contractile function. Although deep convolutional neural networks (ConvNets) have achieved considerable success in medical imaging segmentation, it is still a challenging task for existing deep ConvNets to precisely and automatically segment multiple heart structures from cardiac MRI. This paper presents a novel recurrent interleaved attention network (RIANet) to comprehensively tackle this issue. MethodThe proposed RIANet can efficiently reuse parameters to encode richer representative features via introducing a recurrent feedback structure, Clique Block, which incorporates both forward and backward connections between different layers with the same resolution. Further, we integrate a plug-and-play interleaved attention (IA) block to modulate the information passed to the decoding stage of RIANet by effectively fusing multi-level contextual information. In addition, we improve the discrimination capability of our RIANet through a deep supervision mechanism with weighted losses. ResultsThe performance of RIANet has been extensively validated in the segmentation contest of the ACDC 2017 challenge held in conjunction with MICCAI 2017, with mean Dice scores of 0.942 (left ventricular), 0.923 (right ventricular) and 0.910 (myocardium) for cardiac MRI segmentation. Besides, we visualize intermediate features of our RIANet using guided backpropagation, which can intuitively depict the effects of our proposed components in feature representation. ConclusionExperimental results demonstrate that our RIANet have achieved competitive segmentation results with fewer parameters compared with the state-of-the-art approaches, corroborating the effectiveness and robustness of our proposed RIANet.