Mesh U-Nets for 3D Cardiac Deformation Modeling

Abstract

During a cardiac cycle, the heart anatomy undergoes a series of complex 3D deformations, which can be analyzed to diagnose various cardiovascular pathologies including myocardial infarction. While volume-based metrics such as ejection fraction are commonly used in clinical practice to assess these deformations globally, they only provide limited information about localized changes in the 3D cardiac structures. The objective of this work is to develop a novel geometric deep learning approach to capture the mechanical deformation of complete 3D ventricular shapes, offering potential to discover new image-based biomarkers for cardiac disease diagnosis. To this end, we propose the mesh U-Net, which combines mesh-based convolution and pooling operations with U-Net-inspired skip connections in a hierarchical step-wise encoder-decoder architecture, in order to enable accurate and efficient learning directly on 3D anatomical meshes. The proposed network is trained to model both cardiac contraction and relaxation, that is, to predict the 3D cardiac anatomy at the end-systolic phase of the cardiac cycle based on the corresponding anatomy at end-diastole and vice versa. We evaluate our method on a multi-center cardiac magnetic resonance imaging (MRI) dataset of 1021 patients with acute myocardial infarction. We find mean surface distances between the predicted and gold standard anatomical meshes close to the pixel resolution of the underlying images and high similarity in multiple commonly used clinical metrics for both prediction directions. In addition, we show that the mesh U-Net compares favorably to a 3D U-Net benchmark by using 66% fewer network parameters and drastically smaller data sizes, while at the same time improving predictive performance by 14%. We also observe that the mesh U-Net is able to capture subpopulation-specific differences in mechanical deformation patterns between patients with different myocardial infarction types and clinical outcomes.