Multi-view Cardiac Image Segmentation via Trans-Dimensional Priors

Abbas Khan,Muhammad Asad,Martin Benning,Caroline Roney,Gregory Slabaugh
2024-04-26
Abstract:We propose a novel multi-stage trans-dimensional architecture for multi-view cardiac image segmentation. Our method exploits the relationship between long-axis (2D) and short-axis (3D) magnetic resonance (MR) images to perform a sequential 3D-to-2D-to-3D segmentation, segmenting the long-axis and short-axis images. In the first stage, 3D segmentation is performed using the short-axis image, and the prediction is transformed to the long-axis view and used as a segmentation prior in the next stage. In the second step, the heart region is localized and cropped around the segmentation prior using a Heart Localization and Cropping (HLC) module, focusing the subsequent model on the heart region of the image, where a 2D segmentation is performed. Similarly, we transform the long-axis prediction to the short-axis view, localize and crop the heart region and again perform a 3D segmentation to refine the initial short-axis segmentation. We evaluate our proposed method on the Multi-Disease, Multi-View & Multi-Center Right Ventricular Segmentation in Cardiac MRI (M&Ms-2) dataset, where our method outperforms state-of-the-art methods in segmenting cardiac regions of interest in both short-axis and long-axis images. The pre-trained models, source code, and implementation details will be publicly available.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper proposes a solution to the problem of multi-view cardiac image segmentation by using a cross-dimension segmentation prior (TDSP) to achieve a sequence segmentation method from 3D to 2D and then back to 3D. This approach enhances the anatomical reference of the network input and encourages the generation of anatomically reasonable segmentation maps. In addition, the paper introduces a heart localization and cropping module (HLC) that focuses on the segmentation of the heart region, reducing computational complexity and eliminating false positive predictions. Experiments show that this method outperforms the current state-of-the-art methods in the M&Ms-2 challenge.