» Articles » PMID: 38435997

Enhancing Medical Image Segmentation with a Multi-transformer U-Net

Overview
Journal PeerJ
Date 2024 Mar 4
PMID 38435997
Authors
Affiliations
Soon will be listed here.
Abstract

Various segmentation networks based on Swin Transformer have shown promise in medical segmentation tasks. Nonetheless, challenges such as lower accuracy and slower training convergence have persisted. To tackle these issues, we introduce a novel approach that combines the Swin Transformer and Deformable Transformer to enhance overall model performance. We leverage the Swin Transformer's window attention mechanism to capture local feature information and employ the Deformable Transformer to adjust sampling positions dynamically, accelerating model convergence and aligning it more closely with object shapes and sizes. By amalgamating both Transformer modules and incorporating additional skip connections to minimize information loss, our proposed model excels at rapidly and accurately segmenting CT or X-ray lung images. Experimental results demonstrate the remarkable, showcasing the significant prowess of our model. It surpasses the performance of the standalone Swin Transformer's Swin Unet and converges more rapidly under identical conditions, yielding accuracy improvements of 0.7% (resulting in 88.18%) and 2.7% (resulting in 98.01%) on the COVID-19 CT scan lesion segmentation dataset and Chest X-ray Masks and Labels dataset, respectively. This advancement has the potential to aid medical practitioners in early diagnosis and treatment decision-making.

Citing Articles

IDCC-SAM: A Zero-Shot Approach for Cell Counting in Immunocytochemistry Dataset Using the Segment Anything Model.

Fanijo S, Jannesari A, Dickerson J Bioengineering (Basel). 2025; 12(2).

PMID: 40001703 PMC: 11851800. DOI: 10.3390/bioengineering12020184.


Flood change detection model based on an improved U-net network and multi-head attention mechanism.

Wang F, Feng X Sci Rep. 2025; 15(1):3295.

PMID: 39865097 PMC: 11770074. DOI: 10.1038/s41598-025-87851-6.


Joint segmentation of sternocleidomastoid and skeletal muscles in computed tomography images using a multiclass learning approach.

Ashino K, Kamiya N, Zhou X, Kato H, Hara T, Fujita H Radiol Phys Technol. 2024; 17(4):854-861.

PMID: 39242477 PMC: 11579203. DOI: 10.1007/s12194-024-00839-1.

References
1.
Wang C, Zhao Z, Ren Q, Xu Y, Yu Y . Dense U-net Based on Patch-Based Learning for Retinal Vessel Segmentation. Entropy (Basel). 2020; 21(2). PMC: 7514650. DOI: 10.3390/e21020168. View

2.
Torre V, Poggio T . On edge detection. IEEE Trans Pattern Anal Mach Intell. 2011; 8(2):147-63. DOI: 10.1109/tpami.1986.4767769. View

3.
Wadhwa A, Bhardwaj A, Singh Verma V . A review on brain tumor segmentation of MRI images. Magn Reson Imaging. 2019; 61:247-259. DOI: 10.1016/j.mri.2019.05.043. View

4.
Jiang J, Hu Y, Liu C, Halpenny D, Hellmann M, Deasy J . Multiple Resolution Residually Connected Feature Streams for Automatic Lung Tumor Segmentation From CT Images. IEEE Trans Med Imaging. 2018; 38(1):134-144. PMC: 6402577. DOI: 10.1109/TMI.2018.2857800. View

5.
Jaeger S, Karargyris A, Candemir S, Folio L, Siegelman J, Callaghan F . Automatic tuberculosis screening using chest radiographs. IEEE Trans Med Imaging. 2013; 33(2):233-45. DOI: 10.1109/TMI.2013.2284099. View