Two-stage MR Image Segmentation Method for Brain Tumors based on Attention Mechanism
Not on the map. Why?
Abstract (arXiv)
Multimodal magnetic resonance imaging (MRI) can reveal different patterns of human tissue and is crucial for clinical diagnosis. However, limited by cost, noise and manual labeling, obtaining diverse and reliable multimodal MR images remains a challenge. For the same lesion, different MRI manifestations have great differences in background information, coarse positioning and fine structure. In order to obtain better generation and segmentation performance, a coordination-spatial attention generation adversarial network (CASP-GAN) based on the cycle-consistent generative adversarial network (CycleGAN) is proposed. The performance of the generator is optimized by introducing the Coordinate Attention (CA) module and the Spatial Attention (SA) module. The two modules can make full use of the captured location information, accurately locating the interested region, and enhancing the generator model network structure. The ability to extract the structure information and the detailed information of the original medical image can help generate the desired image with higher quality. There exist some problems in the original CycleGAN that the training time is long, the parameter amount is too large, and it is difficult to converge. In response to this problem, we introduce the Coordinate Attention (CA) module to replace the Res Block to reduce the number of parameters, and cooperate with the spatial information extraction network above to strengthen the information extraction ability. On the basis of CASP-GAN, an attentional generative cross-modality segmentation (AGCMS) method is further proposed. This method inputs the modalities generated by CASP-GAN and the real modalities into the segmentation network for brain tumor segmentation. Experimental results show that CASP-GAN outperforms CycleGAN and some state-of-the-art methods in PSNR, SSMI and RMSE in most tasks.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 27 of its 43 references that are in the map. First posted on arXiv 17 Apr 2023.
Builds on
27 of 43 references.
References · 27
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2016Image-to-Image Translation with Conditional Adversarial Networkscited by 4,510
- 2018CBAM: Convolutional Block Attention Modulecited by 2,463
- 2020pi-GAN: Periodic Implicit Generative Adversarial Networks for 3D-Aware Image Synthesiscited by 708
- 2017Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networkscited by 707
- 2017Unsupervised Image-to-Image Translation Networkscited by 602
- 2021The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classificationcited by 507
- 20183D MRI brain tumor segmentation using autoencoder regularizationcited by 324
- 2017Learning to Discover Cross-Domain Relations with Generative Adversarial Networkscited by 278
- 2018Generative Adversarial Network in Medical Imaging: A Reviewcited by 257
- 2021Coordinate Attention for Efficient Mobile Network Designcited by 239
15 more
- 2018MedGAN: Medical Image Translation using GANscited by 152
- 2018Image Synthesis in Multi-Contrast MRI with Conditional Generative Adversarial Networkscited by 131
- 2018Distribution Matching Losses Can Hallucinate Features in Medical Image Translationcited by 99
- 2017Deep MR to CT Synthesis using Unpaired Datacited by 86
- 2020Robust Multimodal Brain Tumor Segmentation via Feature Disentanglement and Gated Fusioncited by 67
- 2018Deep Residual Learning for Accelerated MRI using Magnitude and Phase Networkscited by 53
- 2021DU-GAN: Generative Adversarial Networks with Dual-Domain U-Net Based Discriminators for Low-Dose CT Denoisingcited by 46
- 2018Cross-modality image synthesis from unpaired data using CycleGAN: Effects of gradient consistency loss and training data sizecited by 42
- 2017Automatic Brain Tumor Segmentation using Cascaded Anisotropic Convolutional Neural Networkscited by 33
- 2022Cross-Modality Deep Feature Learning for Brain Tumor Segmentationcited by 22
- 2017Joint Sequence Learning and Cross-Modality Convolution for 3D Biomedical Segmentationcited by 15
- 2019Wasserstein GANs for MR Imaging: from Paired to Unpaired Trainingcited by 14
- 2019Infinite Brain MR Images: PGGAN-based Data Augmentation for Tumor Detectioncited by 10
- 2019Rethinking Medical Image Reconstruction via Shape Prior, Going Deeper and Faster: Deep Joint Indirect Registration and Reconstructioncited by 7
- 2020Generalized Wasserstein Dice Score, Distributionally Robust Deep Learning, and Ranger for brain tumor segmentation: BraTS 2020 challengecited by 7
Built on by
Not cited yet.