用注意力和空洞池化提升脑肿瘤分割精度
Enhancing Brain Tumor Segmentation Using Channel Attention and Transfer learning
- 融合EfficientNetB0、通道注意力与ASPP模块增强特征提取
- 在BraTS 2020上全肿瘤Dice达0.903,核心区域HD95为3.54
- 适合医学图像分割研究者及临床辅助诊断系统开发者
准确高效的脑肿瘤分割对临床诊断、治疗规划和病情监测至关重要。本文提出一种改进的ResUNet架构,采用EfficientNetB0作为编码器,引入通道注意力机制和空洞空间金字塔池化(ASPP)模块。EfficientNetB0利用预训练特征提升特征提取效率,通道注意力强化模型对肿瘤相关特征的关注,ASPP实现多尺度上下文学习,有效应对不同大小和形状的肿瘤。模型在TCGA LGG和BraTS 2020两个基准数据集上评估,结果表明,相比基线ResUNet及其EfficientNet变体,本方法在全肿瘤和肿瘤核心区域的分割性能持续更优,于BraTS 2020上分别取得Dice系数0.903和0.851,以及HD95分数9.43和3.54。与当前先进方法相比,该方法在全肿瘤与核心区域分割上表现具有竞争力。结果表明,结合强编码器、注意力机制与ASPP可显著提升脑肿瘤分割性能,具有进一步优化与推广至其他医学图像分割任务的潜力。
原文摘要 · Abstract (English)
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 ResUNet 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, crucial for handling tumors of varying sizes and shapes. The proposed model was evaluated on two benchmark datasets: TCGA LGG and BraTS 2020. Experimental results demonstrate that our method consistently outperforms the baseline ResUNet and its EfficientNet variant, achieving Dice coefficients 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 with 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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。