Enhanced ResU-Net for brain tumor segmentation using EfficientNetB0, Channel Attention, and ASPP.
Authors
Affiliations (2)
Affiliations (2)
- Department of Electrical and Computer Engineering, Texas Tech University, 910 Boston Ave, Lubbock, Texas, 79409, United States.
- Department of Rehabilitation and Movement Sciences, Rutgers Health, 65 Bergen St, Newark, New Jersey, 07107-3001, United States.
Abstract
Accurate and efficient segmentation of brain tumors is critical for diagnosis, treatment planning, and monitoring in clinical practice. In this study, we present an enhanced ResU-Net architecture for automatic brain tumor segmentation, integrating an EfficientNetB0 encoder, a channel attention mechanism, and an Atrous Spatial Pyramid Pooling (ASPP) module. The EfficientNetB0 encoder leverages pre-trained features to improve feature extraction efficiency, while the channel attention mechanism enhances the model's focus on tumor-relevant features. ASPP enables multiscale contextual learning, which is crucial for handling tumors of varying sizes and shapes. The proposed model was evaluated on two benchmark datasets: The Cancer Genome Atlas Low Grade Glioma (TCGA-LGG) and Brain Tumor Segmentation (BraTS-2020). Experimental results demonstrate that our method consistently outperforms the baseline ResU-Net and its EfficientNet variant, achieving DSC of 0.903 and 0.851, and HD95 scores of 9.43 and 3.54 for whole tumor and tumor core regions on the BraTS 2020 dataset, respectively. Compared to state-of-the-art methods, our approach shows competitive performance, particularly in whole tumor and tumor core segmentation. These results indicate that combining a powerful encoder with attention mechanisms and ASPP can significantly enhance brain tumor segmentation performance. The proposed approach holds promise for further optimization and application in other medical image segmentation tasks.