English

High-Resolution Swin Transformer for Automatic Medical Image Segmentation

Computer Vision and Pattern Recognition 2022-07-26 v1 Artificial Intelligence

Abstract

The Resolution of feature maps is critical for medical image segmentation. Most of the existing Transformer-based networks for medical image segmentation are U-Net-like architecture that contains an encoder that utilizes a sequence of Transformer blocks to convert the input medical image from high-resolution representation into low-resolution feature maps and a decoder that gradually recovers the high-resolution representation from low-resolution feature maps. Unlike previous studies, in this paper, we utilize the network design style from the High-Resolution Network (HRNet), replace the convolutional layers with Transformer blocks, and continuously exchange information from the different resolution feature maps that are generated by Transformer blocks. The newly Transformer-based network presented in this paper is denoted as High-Resolution Swin Transformer Network (HRSTNet). Extensive experiments illustrate that HRSTNet can achieve comparable performance with the state-of-the-art Transformer-based U-Net-like architecture on Brain Tumor Segmentation(BraTS) 2021 and the liver dataset from Medical Segmentation Decathlon. The code of HRSTNet will be publicly available at https://github.com/auroua/HRSTNet.

Keywords

Cite

@article{arxiv.2207.11553,
  title  = {High-Resolution Swin Transformer for Automatic Medical Image Segmentation},
  author = {Chen Wei and Shenghan Ren and Kaitai Guo and Haihong Hu and Jimin Liang},
  journal= {arXiv preprint arXiv:2207.11553},
  year   = {2022}
}

Comments

14 pages, 3 figures, 2 tables

R2 v1 2026-06-25T01:10:19.909Z