ZOZI-Seg: A transformer and UNet cascade network with Zoom-Out and Zoom-In scheme for aortic dissection segmentation in enhanced CT images

Abstract

Background & objective: Aortic dissection (AD) is a serious condition requiring rapid and accurate diagnosis. In this study, we aimed to improve the diagnostic accuracy of AD by presenting a novel method for aortic segmentation in computed tomography images that uses a combination of a transformer and a UNet cascade network with a Zoom-Out and Zoom-In scheme (ZOZI-seg).

Methods: The proposed method segments each compartment of the aorta, comprising the true lumen (TL), false lumen (FL), and thrombosis (TH) using a cascade strategy that captures both the global context (anatomical structure) and the local detail texture based on the dynamic patch size with ZOZI schemes. The ZOZI-seg model has a two-stage architecture using both a "3D transformer for panoptic context-awareness" and a "3D UNet for localized texture refinement." The unique ZOZI strategies for patching were demonstrated in an ablation study. The performance of our proposed ZOZI-seg model was tested using a dataset from Asan Medical Center and compared with those of existing models such as nnUNet and nnFormer.

Results: In terms of segmentation accuracy, our method yielded better results, with Dice similarity coefficients (DSCs) of 0.917, 0.882, and 0.630 for TL, FL, and TH, respectively. Furthermore, we indirectly compared our model with those in previous studies using an external dataset to evaluate its robustness and generalizability.

Conclusions: This approach may help in the diagnosis and treatment of AD in different clinical situations and provide a strong basis for further research and clinical applications.

Keywords: Aorta segmentation; Aortic dissection; CT; Transformer; U-Net; nnUNet.