Attention-gated U-Net model for semantic segmentation of brain tumors and feature extraction for survival prognosis
Not on the map. Why?
Abstract (arXiv)
Gliomas, among the most common primary brain tumors, vary widely in aggressiveness, prognosis, and histology, making treatment challenging due to complex and time-intensive surgical interventions. This study presents an Attention-Gated Recurrent Residual U-Net (R2U-Net) based Triplanar (2.5D) model for improved brain tumor segmentation. The proposed model enhances feature representation and segmentation accuracy by integrating residual, recurrent, and triplanar architectures while maintaining computational efficiency, potentially aiding in better treatment planning. The proposed method achieves a Dice Similarity Score (DSC) of 0.900 for Whole Tumor (WT) segmentation on the BraTS2021 validation set, demonstrating performance comparable to leading models. Additionally, the triplanar network extracts 64 features per planar model for survival days prediction, which are reduced to 28 using an Artificial Neural Network (ANN). This approach achieves an accuracy of 45.71%, a Mean Squared Error (MSE) of 108,318.128, and a Spearman Rank Correlation Coefficient (SRC) of 0.338 on the test dataset.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 51 references that are in the map. First posted on arXiv 14 Feb 2026.
Builds on
25 of 51 references.
References · 25
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2017Squeeze-and-Excitation Networkscited by 4,789
- 2018Group Normalizationcited by 1,524
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
- 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,198
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
13 more
- 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
- 2018Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challengecited by 511
- 2018Recurrent Residual Convolutional Neural Network based on U-Net (R2U-Net) for Medical Image Segmentationcited by 180
- 2018Concurrent Spatial and Channel Squeeze & Excitation in Fully Convolutional Networkscited by 122
- 2020nnU-Net for Brain Tumor Segmentationcited by 103
- 2018No New-Netcited by 61
- 2020Brain tumor segmentation with self-ensembled, deeply-supervised 3D U-net neural networks: a BraTS 2020 challenge solutioncited by 20
- 2022Reciprocal Adversarial Learning for Brain Tumor Segmentation: A Solution to BraTS Challenge 2021 Segmentation Taskcited by 9
- 20203D Semantic Segmentation of Brain Tumor for Overall Survival Predictioncited by 6
- 2020Uncertainty-driven refinement of tumor-core segmentation using 3D-to-2D networks with label uncertaintycited by 6
- 2020A Review on End-To-End Methods for Brain Tumor Segmentation and Overall Survival Predictioncited by 3
- 2021Glioblastoma Multiforme Patient Survival Predictioncited by 2
- 2021Brain tumour segmentation using a triplanar ensemble of U-Netscited by 2
Built on by
Not cited yet.