Semantic Segmentation of Brain Tumors Using a Local–Global Attention Model
Source abstract
Abstract
The distinctions between tumor areas and surrounding tissues are often subtle. Additionally, the morphology and size of tumors can vary significantly among different patients. These factors pose considerable challenges for the precise segmentation of tumors and subsequent diagnosis. Therefore, researchers are trying to develop an automated and accurate segmentation model. Currently, many segmentation models in deep learning rely on Convolutional Neural Network or Vision Transformer. However, Convolution-based models often fail to deliver precise segmentation results, while Transformer-based models often require more computational resources. To address these challenges, we propose a novel hybrid model named Local–Global UNet Transformer. In our model, we introduce: (1) a semantic-oriented masked attention to enhance the feature extraction capability of the decoder; and (2) network-in-network blocks to increase channel modeling complexity in the encoder while reducing the parameter consumption associated with residual blocks. We evaluate our model on two public brain tumor segmentation datasets, and the experimental results demonstrate that our model achieves the highest average Dice score on the BraTS2024-GLI dataset and ranks second on the BraTS2023-GLI dataset. In terms of HD95, our model attains the lowest values on both datasets. Furthermore, the ablation study proves the effectiveness of our model design.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Brain Tumor Detection and Classification · Machine Learning in Materials Science · Medical Image Segmentation Techniques
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
College of Computer Science and Mathematics, Fujian University of Technology, Fuzhou 350118, China
Fujian Provincial Key Laboratory of Big Data Mining and Applications, Fujian University of Technology, Fuzhou 350118, China
Funding
Fujian Provincial Natural Science Foundation of China
References (57)
Show source-provided references
1. Hallmarks of cancer: The next generation10.1016/j.cell.2011.02.013 ↗
2. Annotation-efficient deep learning for automatic medical image segmentation10.1038/s41467-021-26216-9 ↗
3. Whole-cell segmentation of tissue images with human-level performance using large-scale data annotation and deep learning10.1038/s41587-021-01094-0 ↗
4. Segment anything in medical images10.1038/s41467-024-44824-z ↗
5. Cinarer, G., and Emiroglu, B.G. (2019, January 11–13). Classificatin of Brain Tumors by Machine Learning Algorithms. Proceedings of the 2019 3rd International Symposium on Multidisciplinary Studies and Innovative Technologies (ISMSIT), Ankara, Turkey.10.1109/ISMSIT.2019.8932878 ↗
6. Padlia, M., and Sharma, J. (2018). Fractional Sobel Filter Based Brain Tumor Detection and Segmentation Using Statistical Features and SVM. Nanoelectronics, Circuits and Communication Systems, Springer.10.1007/978-981-13-0776-8_15 ↗
7. and Amarapur, B. (2017, January 8–9). An Automated Approach for Brain Tumor Identification using ANN Classifier. Proceedings of the 2017 International Conference on Current Trends in Computer, Electrical, Electronics and Communication (CTCEEC), Mysore, India.
8. Brain tumor segmentation with Deep Neural Networks10.1016/j.media.2016.05.004 ↗
9. Brain Tumor Segmentation Using Convolutional Neural Networks in MRI Images10.1109/TMI.2016.2538465 ↗
10. Simonyan, K., and Zisserman, A. (2014). Very Deep Convolutional Networks for Large-Scale Image Recognition. arXiv.
11. New variants of a method of MRI scale standardization10.1109/42.836373 ↗
12. MM-BiFPN: Multi-Modality Fusion Network With Bi-FPN for MRI Brain Tumor Segmentation10.1109/ACCESS.2021.3132050 ↗
13. Sagar, A. (2022). ViTBIS: Vision Transformer for Biomedical Image Segmentation. arXiv.10.1007/978-3-031-08999-2_3 ↗
14. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2020). An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale. arXiv.
15. D-former: A U-shaped Dilated Transformer for 3D medical image segmentation10.1007/s00521-022-07859-1 ↗
16. Wei, C., Ren, S., Guo, K., Hu, H., and Liang, J. (2023). High-Resolution Swin Transformer for Automatic Medical Image Segmentation. Sensors, 23.10.3390/s23073420 ↗
17. Sun, K., Xiao, B., Liu, D., and Wang, J. (2019, January 15–20). Deep High-Resolution Representation Learning for Human Pose Estimation. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.10.1109/CVPR.2019.00584 ↗
18. Wang, L., Dong, X., Wang, Y., Liu, L., An, W., and Guo, Y.K. (2022, January 18–24). Learnable Lookup Table for Neural Network Quantization. Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), New Orleans, LA, USA.10.1109/CVPR52688.2022.01210 ↗
19. TransConver: Transformer and convolution parallel network for developing automatic brain tumor segmentation in MRI images10.21037/qims-21-919 ↗
20. Wang, W., Chen, C., Ding, M., Li, J., Yu, H., and Zha, S. (2021). TransBTS: Multimodal Brain Tumor Segmentation Using Transformer. arXiv.10.1007/978-3-030-87193-2_11 ↗
21. Hatamizadeh, A., Nath, V., Tang, Y., Yang, D., Roth, H.R., and Xu, D. (2022). Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Images. arXiv.10.1007/978-3-031-08999-2_22 ↗
22. Federated benchmarking of medical artificial intelligence with MedPerf10.1038/s42256-023-00652-2 ↗
23. de Verdier, M.C., Saluja, R., Gagnon, L., Labella, D., Baid, U., Tahon, N.E.H.M., Foltyn-Dumitru, M., Zhang, J., Alafif, M.M., and Baig, S. (2024). The 2024 Brain Tumor Segmentation (BraTS) Challenge: Glioma Segmentation on Post-treatment MRI. arXiv.
24. Baid, U., Ghodasara, S., Bilello, M., Mohan, S., Calabrese, E., Colak, E., Farahani, K., Kalpathy-Cramer, J., Kitamura, F.C., and Pati, S. (2021). The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classification. arXiv.
25. The Multimodal Brain Tumor Image Segmentation Benchmark (BRATS)10.1109/TMI.2014.2377694 ↗
26. Advancing The Cancer Genome Atlas glioma MRI collections with expert segmentation labels and radiomic features10.1038/sdata.2017.117 ↗
27. Ronneberger, O., Fischer, P., and Brox, T. (2015). U-Net: Convolutional Networks for Biomedical Image Segmentation. arXiv.10.1007/978-3-319-24574-4_28 ↗
28. Xiao, X., Lian, S., Luo, Z., and Li, S. (2018, January 19–21). Weighted Res-UNet for High-Quality Retina Vessel Segmentation. Proceedings of the 2018 9th International Conference on Information Technology in Medicine and Education (ITME), Hangzhou, China.10.1109/ITME.2018.00080 ↗
29. Fully Dense UNet for 2D Sparse Photoacoustic Tomography Artifact Removal10.1109/JBHI.2019.2912935 ↗
30. Zhou, Z., Siddiquee, M.M.R., Tajbakhsh, N., and Liang, J. (2018). UNet++: A Nested U-Net Architecture for Medical Image Segmentation. arXiv.10.1007/978-3-030-00889-5_1 ↗
31. Milletarì, F., Navab, N., and Ahmadi, S.A. (2016, January 25–28). V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation. Proceedings of the 2016 Fourth International Conference on 3D Vision (3DV), Stanford, CA, USA.10.1109/3DV.2016.79 ↗
32. Efficient multi-scale 3D CNN with fully connected CRF for accurate brain lesion segmentation10.1016/j.media.2016.10.004 ↗
33. Chen, C., Liu, X., Ding, M., Zheng, J., and Li, J. (2019, January 13–17). 3D Dilated Multi-Fiber Network for Real-time Brain Tumor Segmentation in MRI. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Shenzhen, China.10.1007/978-3-030-32248-9_21 ↗
34. Çiçek, Ö., Abdulkadir, A., Lienkamp, S.S., Brox, T., and Ronneberger, O. (2016, January 17–21). 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Athens, Greece.10.1007/978-3-319-46723-8_49 ↗
35. nnU-Net: A self-configuring method for deep learning-based biomedical image segmentation10.1038/s41592-020-01008-z ↗
36. Huang, H., Lin, L., Tong, R., Hu, H., Zhang, Q., Iwamoto, Y., Han, X., Chen, Y., and Wu, J. (2020, January 4–9). UNet 3+: A Full-Scale Connected UNet for Medical Image Segmentation. Proceedings of the ICASSP 2020—2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Barcelona, Spain.10.1109/ICASSP40776.2020.9053405 ↗
37. Peiris, H., Hayat, M., Chen, Z., Egan, G.F., and Harandi, M. (October, January 27). A Robust Volumetric Transformer for Accurate 3D Tumor Segmentation. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Strasbourg, France.
38. Xing, Z., Yu, L., Wan, L., Han, T., and Zhu, L. (2022). NestedFormer: Nested Modality-Aware Transformer for Brain Tumor Segmentation. arXiv.10.1007/978-3-031-16443-9_14 ↗
39. Unsupervised brain imaging 3D anomaly detection and segmentation with transformers10.1016/j.media.2022.102475 ↗
40. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., and Guo, B. (2021, January 11–17). Swin Transformer: Hierarchical Vision Transformer using Shifted Windows. Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision (ICCV), Montreal, QC, Canada.10.1109/ICCV48922.2021.00986 ↗
41. Cao, H., Wang, Y., Chen, J., Jiang, D., Zhang, X., Tian, Q., and Wang, M. (2021). Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentation. arXiv.
42. BTSwin-Unet: 3D U-shaped Symmetrical Swin Transformer-based Network for Brain Tumor Segmentation with Self-supervised Pre-training10.1007/s11063-022-10919-1 ↗
43. Bakas, S., Reyes, M., Jakab, A., Bauer, S., Rempfler, M., Crimi, A., Shinohara, R.T., Berger, C., Ha, S.M., and Rozycki, M. (2018). Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challenge. arXiv.
44. Chen, J., Lu, Y., Yu, Q., Luo, X., Adeli, E., Wang, Y., Lu, L., Yuille, A.L., and Zhou, Y. (2021). TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation. arXiv.
45. (2025, March 25). Multi-Atlas Labeling Beyond the Cranial Vault. Available online: https://www.synapse.org/Synapse:syn3193805/wiki/.
46. Hatamizadeh, A., Yang, D., Roth, H.R., and Xu, D. (2022, January 3–8). UNETR: Transformers for 3D Medical Image Segmentation. Proceedings of the 2022 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), Waikoloa, HI, USA.10.1109/WACV51458.2022.00181 ↗
47. Roy, S., Koehler, G., Ulrich, C., Baumgartner, M., Petersen, J., Isensee, F., Jaeger, P.F., and Maier-Hein, K.H. (2023, January 8–12). MedNeXt: Transformer-driven Scaling of ConvNets for Medical Image Segmentation. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Vancouver, BC, Canada.10.1007/978-3-031-43901-8_39 ↗
48. UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentation10.1109/TMI.2024.3398728 ↗
49. SSCFormer: Revisiting ConvNet-Transformer Hybrid Framework From Scale-Wise and Spatial-Channel-Aware Perspectives for Volumetric Medical Image Segmentation10.1109/JBHI.2024.3392488 ↗
50. MM-UNet: A novel cross-attention mechanism between modules and scales for brain tumor segmentation10.1016/j.engappai.2024.108591 ↗
51. Yu, F., Cao, J., Liu, L., and Jiang, M. (2025, January 19–25). SuperLightNet: Lightweight Parameter Aggregation Network for Multimodal Brain Tumor Segmentation. Proceedings of the 2025 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA.
52. Pancreas segmentation with multi-channel convolution and combined deep supervision
53. Lee, H.H., Bao, S., Huo, Y., and Landman, B.A. (2022). 3D UX-Net: A Large Kernel Volumetric ConvNet Modernizing Hierarchical Transformer for Medical Image Segmentation. arXiv.
54. Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., and Antiga, L. (2019). PyTorch: An Imperative Style, High-Performance Deep Learning Library. arXiv.
55. Cardoso, M.J., Li, W., Brown, R., Ma, N., Kerfoot, E., Wang, Y., Murrey, B., Myronenko, A., Zhao, C., and Yang, D. (2022). MONAI: An open-source framework for deep learning in healthcare. arXiv.
56. Loshchilov, I., and Hutter, F. (2017). Decoupled Weight Decay Regularization. arXiv.
57. Liu, Z., Mao, H., Wu, C., Feichtenhofer, C., Darrell, T., and Xie, S. (2022, January 18–24). A ConvNet for the 2020s. Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), New Orleans, LA, USA.10.1109/CVPR52688.2022.01167 ↗