CardiacSegFormer: Transformer for Semantic Segmentation of Cardiac Images.

Yuexiao Feng,Shuaicong Hu,Yuhang Deng,Cuiwei Yang
DOI: https://doi.org/10.1145/3633637.3633692
2023-01-01
Abstract:Cardiac image segmentation is a sub task of medical image segmentation, which has very important clinical significance in the prevention and diagnosis of cardiac diseases. Recently, segmentation models based on the Transformer architecture have achieved remarkable performance due to their advantages in handling long-range dependencies and capturing global contextual information. However, the Transformer based segmentation model only uses simple unprocessed skip connections to compensate for information loss during the down-sampling process, resulting in its potential in cardiac image segmentation remains largely unexplored. In this work, we propose CardiacSegFormer, which consists of various improvements, such as the application of self-attention modules in skip connections to facilitate better information fusion. At the same time, low-entropy constraint is introduced, cosine annealing warm restart algorithms are used to help training a more effective model. Experiment on Automated Cardiac Diagnosis Challenge (ACDC) datasets demonstrate that the proposed method has a Dice Similarity Coefficient (DSC) of 91.84%, which is superior to the current the-state-of-art model. To conclusion, this work provides a high-performing baseline model for cardiac image segmentation tasks, showcasing its value in the improvement and enhancement of related models.
What problem does this paper attempt to address?