Swin Unet3D: a three-dimensional medical image segmentation network combining vision transformer and convolution

Yimin Cai,Yuqing Long,Zhenggong Han,Mingkun Liu,Yuchen Zheng,Wei Yang,Liming Chen
DOI: https://doi.org/10.1186/s12911-023-02129-z
IF: 3.298
2023-02-17
BMC Medical Informatics and Decision Making
Abstract:Semantic segmentation of brain tumors plays a critical role in clinical treatment, especially for three-dimensional (3D) magnetic resonance imaging, which is often used in clinical practice. Automatic segmentation of the 3D structure of brain tumors can quickly help physicians understand the properties of tumors, such as the shape and size, thus improving the efficiency of preoperative planning and the odds of successful surgery. In past decades, 3D convolutional neural networks (CNNs) have dominated automatic segmentation methods for 3D medical images, and these network structures have achieved good results. However, to reduce the number of neural network parameters, practitioners ensure that the size of convolutional kernels in 3D convolutional operations generally does not exceed , which also leads to CNNs showing limitations in learning long-distance dependent information. Vision Transformer (ViT) is very good at learning long-distance dependent information in images, but it suffers from the problems of many parameters. What's worse, the ViT cannot learn local dependency information in the previous layers under the condition of insufficient data. However, in the image segmentation task, being able to learn this local dependency information in the previous layers makes a big impact on the performance of the model.
medical informatics
What problem does this paper attempt to address?