EAAI Journal 2025 Journal Article
A lightweight segmentation model based on dilated multi-scale residual attention U-Net for brain tumor segmentation
- Lihong Zhang
- Yuzhuo Li
- Yingbo Liang
- Tong Liu
- Wenwu Zhang
- Junding Sun
To address the limited computational capacity of current clinical medical devices, which hampers the effective use of complex high-performance segmentation algorithms, this paper proposes a new lightweight brain tumor segmentation algorithm: Lightweight Dilated Multi-Scale Residual Attention U-Net (LDMRA U-Net). This model is based on the previously proposed Dilated Multi-Scale Residual Attention U-Net (DMRA U-Net). The overall network architecture incorporates the proposed Lightweight Channel Aggregation (LCA) mechanism to reduce the number of channels, along with a multi-level decoder aggregation strategy to minimize key information loss. Additionally, to enhance more direct information transfer between different levels of the encoder-decoder, the proposed Cross Enhanced Attention (CEA) module is incorporated into the skip connection part, improving segmentation performance. We evaluated our proposed method on the Brain Tumor Segmentation Challenge (BraTS) 2019 dataset, and the experimental results show that the number of parameters in LDMRA U-Net is only about 7. 41 % of that in DMRA U-Net, with Dice Similarity Coefficients (DSC) of 0. 8882, 0. 8685, and 0. 8622 in the Whole Tumor (WT), Tumor Core (TC), and Enhancing Tumor (ET) regions, respectively. LDMRA U-Net demonstrates significant performance improvements compared to existing lightweight methods. Our method helps to improve brain tumor segmentation accuracy with very small parameter sizes, and has the potential to be applied in a clinical setting.