Medical image segmentation plays a pivotal role in computer-aided diagnosis and treatment planning. Traditional segmentation approaches often struggle to balance global and local context, either capturing overall anatomical structures or focusing on minute details, but not both. This paper introduces the Swin-Hierarchical Attention Unet (Swin-HAUnet), which harmonizes this dichotomy by integrating global contextual insights with local feature enhancement. The proposed network architecture employs a hybrid approach, leveraging an advanced transformer-based encoder to process wide-ranging contextual information and an attention-enhanced decoder to refine the segmentation of nuanced and intricate anatomical features. We performed experiments on two publicly available datasets, the Synapse multi-organ segmentation CT dataset and the UW-Madison dataset. The Swin-HAUnet shows a marked improvement in performance, achieving a Dice similarity coefficient of 79.91%, a notable increase of 1.26% over the baseline model on Synapse datasets. These results underscore the model’s effectiveness in complex segmentation tasks and the importance of attention mechanisms in medical image analysis.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Swin-HAUnet: A Swin-Hierarchical Attention Unet For Enhanced Medical Image Segmentation

  • Jiarong Chen,
  • Xuyang Zhang,
  • Rongwen Li,
  • Peng Zhou

摘要

Medical image segmentation plays a pivotal role in computer-aided diagnosis and treatment planning. Traditional segmentation approaches often struggle to balance global and local context, either capturing overall anatomical structures or focusing on minute details, but not both. This paper introduces the Swin-Hierarchical Attention Unet (Swin-HAUnet), which harmonizes this dichotomy by integrating global contextual insights with local feature enhancement. The proposed network architecture employs a hybrid approach, leveraging an advanced transformer-based encoder to process wide-ranging contextual information and an attention-enhanced decoder to refine the segmentation of nuanced and intricate anatomical features. We performed experiments on two publicly available datasets, the Synapse multi-organ segmentation CT dataset and the UW-Madison dataset. The Swin-HAUnet shows a marked improvement in performance, achieving a Dice similarity coefficient of 79.91%, a notable increase of 1.26% over the baseline model on Synapse datasets. These results underscore the model’s effectiveness in complex segmentation tasks and the importance of attention mechanisms in medical image analysis.