## 1 Introduction

Atrial Fibrillation (AF) is a common electro-physiological cardiac disorder with a large prevalence worldwide [1] that causes changes in the anatomy of the atria. A better characterization of these changes as a consequence of the substrate that causes and sustains fibrillation is desirable for the definition of clinical biomarkers. These biomarkers can be directly started from the image (i.e. the shape of the atria [2], or the fibrotic burden from late gadolinium enhanced (LGE) magnetic resonance imaging (MRI) [3]) or from mechanistic simulations of the function (i.e. computation of the risk of arrhythmia perpetuation [4]).

There is thus a need for a fully automatic segmentation of the atria from clinical images, especially in LGE studies. The current state of the art is based in tedious along with error-prone manual procedures, and the main difficulty is the lack of contrast from atrial tissue including the surrounding background. Fully automated solutions are desirable to speed up the process and remove inter- and intra-observer variability. In this direction, a combination of multi-atlas registration within 3D level-set has been proposed, reporting a reasonable performance in the main atrial body and pulmonary vein regions [5]. The large computational burden of this multi-atlas approach can be alleviated by the use of convolutional neural networks (CNNs), as it has been illustrated in [6] for the analysis of 2D MRI slices.

The first idea followed in this work is the use of 3D CNNs as the effective solution for the segmentation of the 3D LGE datasets. The starting point is the v-net [7], which takes into consideration the spatial redundancy naturally present on the entire volumetric stack with 3D-kernels, showing good benefits in different cardiac segmentation problems [8],[9]. This architecture is modified to reduce the memory burden and speed-up its training. The last idea explored in this work is the choice of a sensible loss function, the joint combination of both the mean squared error (MSE) and Dice Loss, which has been reported to be beneficial as the MSE minimizes global image details while the Dice Loss reduces local over-segmentation errors [7].

## 2 Atrial Datasets

A population of 100 3D GE-MRIs, data and masks, were made available through the [2018 Atrial Segmentation Challenge] and used in this work. Images had a acquisition resolution of 0.625 x 0.625 x 0.625 .

## 3 Method

A volumetric fully convolutional network, V-FCNN, is designed with two main paths (see Fig.1): the volumetric down-sampling path as well as the volumetric up-sampling path. The volumetric down-sampling path has four 3-D convolutions blocks, each following by PreLU along with 3-D Batch Normalisation (BN) layers. This path takes as input the entire (XYZ) volume and progressively reduces the size of each slice (XY) together with the number of stack spatial slices (Z). In this phase, the volume is compressed and presents both (XY) and (Z) reduction. In a complementary fashion, the volumetric up-sampling path restores the compressed volume to its initial size, with every 3-D up-sampling convolution blocks being followed by PreLU and 3-D Batch Normalisation (BN). Each sampling paths of V-FCNN contains four blocks within 16,32,64,128 -3D kernels respectively (with the size fix to ).

Image down-sampling during a segmentation task presents the problem of feature map reduction, followed by a strong spatial information loss. This problem has been addressed in v-net [7], [8] by adding skips layers between down-sampling and up-sampling layers in order to fuse low-level features within high level features.

Our work explores an alternative approach to the skip path connections. In particular, we boost our model to capture fine details in two ways. First, the use of max-pooling operations is avoided in order to prevent the loss of spatial resolution (i.e if pools do not overlap well, pooling operation loses appreciable information where the objects are located in the image). The second idea is to use a loss layer that combines the

and metrics as an attempt to recover details in the image. The rationale is that the term searches for local details in the volume data, as a consequence, the can be seen as a regularization that instead focuses on global features on the MRI volume. This bimodal loss also prevents the V-FCNN to fall in a local minimum; especially within small atrial regions. Specifically, the loss is defined as:(1) |

where are the Ground Truth (GT) binary slices, as well as are the correspondent’s prediction masks. is an index through the spatial slices in the Z (z-axis). Then, the sums for in runs over all N pixels of the prediction masks and the GT masks. Finally, controls the amount of during the optimisation training process (set to ).

The size of each MRI slice is reduced to

pixels using down-sampling bi-cubic interpolation, allowing the network to be faster in training stage moreover enabling a solution without expensive GPU hardware (i.e. large images need more GPU global memory

GB). An up-sampling bi-cubic interpolation is finally used for restoring the mask to the size of pixels.### 3.1 Implementation details

For our experiments, we train the network with a number of epochs of 1000 up to convergence. The Stochastic Gradient Descent (SDG) was used with a learning rate of

, while the momentum and weight decay are , respectively. Furthermore, to increase the generalization of the network, data argumentation was used, finding particularly effective the random vertical and horizontal flip in close combination with plane translation. Input sequences were equalized in grayscale intensity with CLAHE (Contrast Limited Adaptive Histogram Equalization) [10], and the noise was minimized with a combination of High-Pass Filters and Gaussian blurring filters.## 4 Experiments

The experimentation phase is divided into two phases. In the first phase (preparation phase) 5 patients are used for validation ( slices in total), and the rest (90) is used for training. In the second phase (competition), our algorithm was evaluated in the test-set with 54 patients ( atria slices in total).

Segmentation accuracy was measured with the Dice Metric (DM) [11], which was subdivided into 3 regions of the atria: top (including the pulmonary veins), middle and bottom (including the valve plane that divides the atria and ventricle). Besides, DM and the surface Hausdorff Distance (HD) [12] are computed for the entire atrial anatomy.

Particularly, given a point and a surface , the distance is defined as:

(2) |

whereas the is the Euclidean distance between two points in a Euclidean space. Then, the HD between mesh surfaces and is given as:

(3) |

Proposed V-FCNN achieved in the first phase an average DM of , and in the top, middle and bottom regions, respectively (see Table 1 and an illustrative example in Fig. 2). The HD ranged from 0.31 to 0.86 mm. The average DM and HD were and respectively. The DM in the competition phase was .

Visual inspection of the contours reveals how a loss of accuracy is seen on the first top slices for two of the cases, creating an artificial flattening of the shape (see patients 1 and 2 of Fig. 3), and on the top of the atria with the more variable anatomy of the pulmonary veins.

TOP | MID | BOTTOM | 3D-MESH | |

Patient number | DM (%) | DM (%) | DM (%) | HD |

1 | 77.74 (8.46) | 84.62 (2.40) | 76.99 (10.89) | 0.86 |

2 | 72.26 (9.50) | 81.25 (1.57) | 54.18 (33.70) | 0.66 |

3 | 74.10 (8.92) | 77.75 (1.36) | 68.27 (14.75) | 0.31 |

4 | 81.60 (0.74) | 81.54 (1.20) | 72.32 (14.32) | 0.55 |

5 | 84.48 (2.58) | 85.36 (0.66) | 76.33 (7.05) | 0.58 |

Automatic segmentation results (Dice Metric and Hausdorff Distance) for all five test patients. Results report the mean and standard deviation, and are divided into three different atrium sections: top, middle and bottom.

## 5 Discussion and Conclusions

The exploitation of spatial coherence across different volumetric slices is an important resource for improving the accuracy of fully automated segmentation. Proposed V-FCNN achieves a good segmentation performance, mostly in the middle atrial section. Limitations occur in the top and bottom sections, caused by the presence of the pulmonary veins and the difficulty to identify where the atria and ventricle split. In the preparation phase the V-FCNN achieved a DM between (top case) and (worst case), along with a DM of at the competition phase.

A cardiac imaging study has a huge level of correlation in space, as well as many segmentation algorithms have exploited the spatial coherence within different deep-learning techniques [7], [8], [9], [13], [14]. Some of them have used 3D convolutional kernels [7], [8], [9] that have the advantage of directly capturing the spatial information in each convolutional layer without adding extra parameters (i.e. the addition of a recurring network). This is the main reason that has driven us to use them in our optimized V-FCNN.

The limitation of the 3D convolutional kernels is the large memory burden, and this is the reason why the original detail of the images was reduced (from 640 to 127 pixels). However, this image reduction is required due to the current hardware limitations available. In fact, while the 3D kernels have a good ability to extract the spatial correlation of the MRI sequence, on the other they need a high capacity in GPU global memory size. Thus, an excessive reduction of the MRI to adapt to the available hardware causes an inevitable loss in segmentation details (i.e probable motivation of

dice). Therefore, we recommend future tests with 3D kernel at more higher MRI resolution with the help of NVIDIA V100 GPUs [15] that presents GB of global memory.Furthermore, the concatenation of two CNNs, the second working at the full resolution cropping the region of interests containing the atria, is a good strategy that will minimize the impact of this choice. An alternative is to work with the information at two levels of detail, which has shown to achieve a better performance [16].

Proposed V-FCNN simplifies the v-net [7] by removing the skip-paths, in an attempt to achieve a much quicker training convergence. The constitutive advantage of the skip-paths is to increase the localization, but this comes at the cost of a considerable slow down of the speed of the network (i.e GPU global memory leak) while propagating the gradient, at every iteration, forward and back through those paths [17].

The choice of the loss function is an important consideration in any CNN, and the joint minimization of MSE and Dice Loss has been adopted in our solution. The interpretation is that the MSE looks at global volumetric features, while the Dice Loss (DL) regularize it trying to fit local details. This choice allowed the learning to avoid local minima and the corresponding slow convergence of using only MSE. The optimal weight between them, and the inclusion of further criteria such as an L1/L2 loss or a statistical distance to a library of existing cases, should be addressed in future extensions of this work.

An alternative strategy to 3D kernels is the use of recurrent units [13],[14], where recurrence is used to capture the redundancy between adjacent slices. The use of recurrence is a much more memory efficient approach (i.e fewer parameters to capture recurring partners), also reporting improvements at the challenging apical slices of the left ventricle [13]

. But these solutions are limited by the vanishing gradient that unfortunately occurs within long sequences, what can be partially avoided by imposing upper bound constraints on the backward gradient (ie. gradient clipping) or with a regularisation term

[18]. Future work is thus still needed to maximize the synergies between 3D kernels and recurrent units within a more higher input MRI resolution.## Acknowledgements

This work was supported by the Wellcome/EPSRC Centre for Medical Engineering at King’s College London [g.a. 203148/Z/16/Z]. PL holds a Wellcome Trust Senior Research Fellowship [g.a. 209450/Z/17/Z].

## References

- [1] EN Prystowsky, DW Benson, and V Fuster. Management of patients with atrial fibrillation?: A statement for healthcare professionals from the subcommittee on electrocardiography and electrophysiology. American Heart Association. Circulation, 93:15, 1996.
- [2] Marta Varela, Felipe Bisbal, Ernesto Zacur, Antonio Berruezo, Oleg V. Aslanidi, Lluis Mont, and Pablo Lamata. Novel computational analysis of left atrial anatomy improves prediction of atrial fibrillation recurrence after ablation. Frontiers in Physiology, 8:68, 2017.
- [3] RJ Kim, E Wu, A Rafael, and et al. The use of contrast‐enhanced magnetic resonance imaging to identify reversible myocardial dysfunction. N Engl J Med, 343:1445–1453, 2000.
- [4] Patrick M. Boyle, Sohail Zahid, and Natalia A. Trayanova. Towards personalized computational modelling of the fibrotic substrate for atrial arrhythmia. EP Europace, 18:136–145, 2016.
- [5] Qian Tao, Rahil Shahzad, Esra Gucuk Ipek, Floris F. Berendsen, Saman Nazarian, and Rob J. van der Geest. Fully automated segmentation of left atrium and pulmonary veins in late gadolinium enhanced mri. Journal of Cardiovascular Magnetic Resonance, 18(1):O84, Jan 2016.
- [6] Aliasghar Mortazi, Rashed Karim, Kawal S. Rhode, Jeremy Burt, and Ulas Bagci. Cardiacnet: Segmentation of left atrium and proximal pulmonary veins from MRI using multi-view CNN. CoRR, abs/1705.06333, 2017.
- [7] Fausto Milletari, Nassir Navab, and Seyed-Ahmad Ahmadi. V-net: Fully convolutional neural networks for volumetric medical image segmentation. CoRR, abs/1606.04797, 2016.
- [8] Fabian Isensee, Paul Jaeger, Peter M. Full, Ivo Wolf, Sandy Engelhardt, and Klaus H. Maier-Hein. Automatic cardiac disease assessment on cine-mri via time-series segmentation and domain specific features. CoRR, abs/1707.00587, 2017.
- [9] Hinrich B. Winther, Christian Hundt, Bertil Schmidt, Christoph Czerner, Johann Bauersachs, Frank K. Wacker, and Jens Vogel-Claussen. $$-net: Deep learning for generalized biventricular cardiac mass and function parameters. CoRR, abs/1706.04397, 2017.
- [10] Hardeep Kaur and J. Sharmila Rani. Mri brain image enhancement using histogram equalization techniques. 2016 International Conference on Wireless Communications, Signal Processing and Networking (WiSPNET), pages 770–773, 2016.
- [11] M.R. Avendi, Arash Kheradvar, and Hamid Jafarkhani. A combined deep-learning and deformable-model approach to fully automatic segmentation of the left ventricle in cardiac mri. Medical Image Analysis, 30:108 – 119, 2016.
- [12] Paolo Cignoni, Claudio Rocchini, and Roberto Scopigno. Metro: Measuring error on simplified surfaces. CoRR, 1996.
- [13] Rudra P. K. Poudel, Pablo Lamata, and Giovanni Montana. Recurrent fully convolutional neural networks for multi-slice mri cardiac segmentation. In Maria A. Zuluaga, Kanwal Bhatia, Bernhard Kainz, Mehdi H. Moghari, and Danielle F. Pace, editors, Reconstruction, Segmentation, and Analysis of Medical Images, pages 83–94, Cham, 2017. Springer International Publishing.
- [14] Jianxu Chen, Lin Yang, Yizhe Zhang, Mark S. Alber, and Danny Z. Chen. Combining fully convolutional and recurrent neural networks for 3d biomedical image segmentation. CoRR, abs/1609.01006, 2016.
- [15] Zhe Jia, Marco Maggioni, Benjamin Staiger, and Daniele Paolo Scarpazza. Dissecting the NVIDIA volta GPU architecture via microbenchmarking. CoRR, abs/1804.06826, 2018.
- [16] Z. Xiong, V. V. Fedorov, X. Fu, E. Cheng, R. Macleod, and J. Zhao. Fully automatic left atrium segmentation from late gadolinium enhanced magnetic resonance imaging using a dual fully convolutional neural network. IEEE Transactions on Medical Imaging, pages 1–1, 2018.
- [17] Adam Paszke, Abhishek Chaurasia, Sangpil Kim, and Eugenio Culurciello. Enet: A deep neural network architecture for real-time semantic segmentation. CoRR, abs/1606.02147, 2016.
- [18] Razvan Pascanu, Tomas Mikolov, and Yoshua Bengio. Understanding the exploding gradient problem. CoRR, abs/1211.5063, 2012.