Joint Left Atrial Segmentation and Scar Quantification Based on a DNN with Spatial Encoding and Shape Attention

TLDR: An end-to-end deep neural network (DNN) which can simultaneously segment the left atrial (LA) cavity and quantify LA scars and a shape attention (SA) mechanism is embedded into a two-task network to perform the joint LA segmentation and scar quantification.

Abstract: We propose an end-to-end deep neural network (DNN) which can simultaneously segment the left atrial (LA) cavity and quantify LA scars The framework incorporates the continuous spatial information of the target by introducing a spatially encoded (SE) loss based on the distance transform map Compared to conventional binary label based loss, the proposed SE loss can reduce noisy patches in the resulting segmentation, which is commonly seen for deep learning-based methods To fully utilize the inherent spatial relationship between LA and LA scars, we further propose a shape attention (SA) mechanism through an explicit surface projection to build an end-to-end-trainable model Specifically, the SA scheme is embedded into a two-task network to perform the joint LA segmentation and scar quantification Moreover, the proposed method can alleviate the severe class-imbalance problem when detecting small and discrete targets like scars We evaluated the proposed framework on 60 LGE MRI data from the MICCAI2018 LA challenge For LA segmentation, the proposed method reduced the mean Hausdorff distance from 364 mm to 200 mm compared to the 3D basic U-Net using the binary cross-entropy loss For scar quantification, the method was compared with the results or algorithms reported in the literature and demonstrated better performance