融合注意力机制与多尺度池化,提升脑肿瘤分割精度。
Hybridization of Attention UNet with Repeated Atrous Spatial Pyramid Pooling for Improved Brain Tumour Segmentation
- 将注意力UNet与重复式空洞金字塔池化结合,增强多尺度特征提取。
- 在BraTS2018数据集上,Dice系数达0.872,优于现有模型。
- 适合医学图像分割研究者,尤其关注脑肿瘤自动识别的场景。
脑肿瘤在空间和尺度上具有高度异质性,使得医学图像中的肿瘤分割任务极具挑战,可能引发误诊和治疗偏差。自动化分割可提高客观性、可重复性并缩短处理时间。传统卷积神经网络因无法准确表征肿瘤大小和形态而表现不佳。尽管UNet通过下采样-上采样结构被广泛用于语义分割,但其对多尺度信息的捕捉仍有限。本文提出一种新架构,将Attention-UNet与重复式空洞空间金字塔池化(ASPP)相结合。ASPP通过不同膨胀率的并行空洞卷积有效捕获多尺度上下文信息,扩展感受野的同时保持细节。注意力机制则通过融合局部特征与全局依赖关系提供必要上下文。该集成显著提升了语义分割性能,在BraTS2018数据集上实现0.872的Dice系数,超越UNet、Attention UNet及带空间金字塔池化的Attention UNet,为肿瘤分割设立了新基准。
原文摘要 · Abstract (English)
Brain tumors are highly heterogeneous in terms of their spatial and scaling characteristics, making tumor segmentation in medical images a difficult task that might result in wrong diagnosis and therapy. Automation of a task like tumor segmentation is expected to enhance objectivity, repeatability and at the same time reducing turn around time. Conventional convolutional neural networks (CNNs) exhibit sub-par performance as a result of their inability to accurately represent the range of tumor sizes and forms. Developing on that, UNets have been a commonly used solution for semantic segmentation, and it uses a downsampling-upsampling approach to segment tumors. This paper proposes a novel architecture that integrates Attention-UNet with repeated Atrous Spatial Pyramid Pooling (ASPP). ASPP effectively captures multi-scale contextual information through parallel atrous convolutions with varying dilation rates. This allows for efficient expansion of the receptive field while maintaining fine details. The attention provides the necessary context by incorporating local characteristics with their corresponding global dependencies. This integration significantly enhances semantic segmentation performance. Our approach demonstrates significant improvements over UNet, Attention UNet and Attention UNet with Spatial Pyramid Pooling allowing to set a new benchmark for tumor segmentation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。