학술논문

Usformer: A Light Neural Network for Left Atrium Segmentation of 3D LGE MRI
Document Type
Conference
Source
2023 31st European Signal Processing Conference (EUSIPCO) European Signal Processing Conference (EUSIPCO), 2023 31st. :995-999 Sep, 2023
Subject
Signal Processing and Analysis
Training
Image segmentation
Three-dimensional displays
Magnetic resonance imaging
Computational modeling
Computed tomography
Manuals
Left atrium segmentation
Late gadolinium enhanced magnetic resonance image
Transformer
Language
ISSN
2076-1465
Abstract
Left atrial fibrosis is an important mediator of atrial fibrillation and atrial myopathy. Late gadolinium-enhancement (LGE) MRI is a proven non-invasive test for the evaluation of left atrial (LA) fibrosis. However, manual segmentation is labor-intensive. Automatic segmentation is challenging due to varying intensities of data acquired by different vendors, low contrast between the LA and surrounding tissues, and complex LA shapes. Current approaches based on 3D networks are computationally expensive and time-consuming due to the large size of 3D LGE MRIs and networks. To address this, most approaches use two-stage methods to first locate the LA center using a down-scaled version of the MRIs and then crop the full-resolution MRIs around the LA center for final segmentation. We propose a light transformer-based model to accurately segment LA volume in one stage, avoiding errors introduced by sub-optimal two-stage training. Transposed attention in transformer blocks can capture long-range dependencies among pixels in large 3D volumes without significant computation requirements. Our proposed model achieved a promising dice similarity coefficient of 92.6 % in the 2018 Atrial Segmentation Challenge, with only 611k parameters, which is about 1 % of the method ranked 3rd in the challenge but with similar performance.