DA-TransUNet: integrating spatial and channel dual attention with transformer U-net for medical image segmentation
DA-TransUNet: интеграция пространственного и канального двойного внимания с Transformer U-Net для сегментации медицинских изображений
2024-05-16
SCID: 54.1/y9capp2a
Discuss with AI
DA-TransUNetTransformer U-Netdual attention (position and channel)medical image segmentationskip-connection dual attention
Figures from the paper
Abstract (AI)
Accurate medical image segmentation is critical for disease quantification and treatment evaluation. While traditional U-Net architectures and their transformer-integrated variants excel in automated segmentation tasks. Existing models also struggle with parameter efficiency and computational complexity, often due to the extensive use of Transformers. However, they lack the ability to harness the image's intrinsic position and channel features. Research employing Dual Attention mechanisms of position and channel have not been specifically optimized for the high-detail demands of medical images. To address these issues, this study proposes a novel deep medical image segmentation framework, called DA-TransUNet, aiming to integrate the Transformer and dual attention block (DA-Block) into the traditional U-shaped architecture. Also, DA-TransUNet tailored for the high-detail requirements of medical images, optimizes the intermittent channels of Dual Attention (DA) and employs DA in each skip-connection to effectively filter out irrelevant information. This integration significantly enhances the model's capability to extract features, thereby improving the performance of medical image segmentation. DA-TransUNet is validated in medical image segmentation tasks, consistently outperforming state-of-the-art techniques across 5 datasets. In summary, DA-TransUNet has made significant strides in medical image segmentation, offering new insights into existing techniques. It strengthens model performance from the perspective of image features, thereby advancing the development of high-precision automated medical image diagnosis. The codes and parameters of our model will be publicly available at https://github.com/SUN-1024/DA-TransUnet.
Key Findings
1
DA-TransUNet consistently outperforms state-of-the-art techniques across five medical image segmentation datasets.
2
DA-TransUNet integrates a Transformer with a Dual Attention block (position and channel) into a U-shaped architecture for medical image segmentation.
3
DA-TransUNet is tailored for high-detail medical images, addressing limitations of prior dual-attention methods not optimized for medical imaging.
4
The authors will publicly release the model code and parameters at the provided GitHub repository.
5
The model applies optimized intermittent channel attention and inserts DA-Blocks in each skip-connection to filter irrelevant information and enhance feature extraction.
Research Object
DA-TransUNet deep medical image segmentation framework (Transformer + dual attention U-Net)
Research Subject
Integration and optimization of spatial (position) and channel dual-attention mechanisms with Transformer within a U-shaped architecture to improve feature extraction, parameter efficiency, and segmentation performance on medical images
Publication Details
Publication Date
2024-05-16
Journal
Publisher
ISSN
Cited by
324
Open access PDF
Access Type
Author Information
Download PDF
Subscribe to digest