SwinTextUNet: Integrating CLIP-Based Text Guidance into Swin Transformer U-Nets for Medical Image Segmentation
Precise medical image segmentation is fundamental for enabling computer-aided diagnosis and effective treatment planning. Traditional models that rely solely on visual features often struggle when confronted with ambiguous or low-contrast patterns. To overcome these limitations, we introduce SwinTextUNet, a multimodal segmentation framework that incorporates Contrastive Language-Image Pre-training (CLIP), derived textual embeddings into a Swin Transformer U-Net backbone. By integrating cross-attention and convolutional fusion, the model effectively aligns semantic text guidance with hierarchical visual representations, enhancing robustness and accuracy. We evaluate our approach on the QaTa-COV19 dataset, where the proposed four-stage variant achieves an optimal balance between performance and complexity, yielding Dice and IoU scores of 86.47% and 78.2%, respectively. Ablation studies further validate the importance of text guidance and multimodal fusion. These findings underscore the promise of vision-language integration in advancing medical image segmentation and supporting clinically meaningful diagnostic tools.
Paper
References (25)
Scroll for more · 13 remaining