提出高效金字塔注意力网络,精准分割DWI脑梗死图像。
EPRA U-Net: An Efficient Pyramid Residual Attention Framework for Accurate Infarct Segmentation in Diffusion-Weighted MRI
- 用EfficientNet编码器+残差循环块增强特征提取能力。
- 在167名患者4895张图上达0.8984的分割准确率,漏检率降低超25%。
- 适合临床脑卒中快速诊断,对小病灶敏感性高。
准确识别扩散加权磁共振成像(DWI)上的急性缺血性梗死,是实现可靠病变量化与有效临床决策支持的关键。本文提出EPRA U-Net(高效金字塔残差注意力U-Net),一种面向DWI梗死分割的任务专用架构。采用基于EfficientNet的编码器作为分层特征提取器,参数量最小化;集成残差循环(R2)块(循环展开步数t=2)和空洞空间金字塔池化(ASPP),强化空间依赖建模;引入双注意力机制,突出病灶激活并抑制背景干扰。为契合临床需求,采用侧重灵敏度的Tversky损失函数优化。在包含167例患者共4895张DWI切片的自建数据集上评估,结果表明EPRA U-Net表现最优:像素级Dice达0.8984,样本级Dice为0.9469,IoU为0.8155,召回率0.8887,病灶F1为0.9378,HD95为11.62像素。相较于UNet++、DeepLabV3+和TransUNet,漏检率分别降低16%、25%和29%。
原文摘要 · Abstract (English)
Objective: Accurate identification of acute ischemic infarcts on diffusion-weighted magnetic resonance imaging (DWI) is a critical prerequisite for reliable lesion quantification and effective clinical decision support in the management of cerebrovascular events. Methods: This study presents EPRA U-Net (Efficient Pyramid Residual Attention U-Net), a task-specific integrated architecture for efficient and accurate infarct segmentation of DWI images. In the proposed architecture, an EfficientNet-based encoder was used as a hierarchical feature extractor with a minimized parameterization. In addition, a Residual-Recurrent (R2) block (recurrent unrolling step t = 2, following the original formulation) and Atrous Spatial Pyramid Pooling (ASPP) were integrated to enhance the performance of spatial dependency modeling. Additionally, a dual attention mechanism was incorporated to highlight lesion-related activations while concurrently enabling the suppression of extraneous background responses. To prioritize lesion detection consistent with clinical imperative, a Tversky loss function was adopted, emphasizing the sensitivity of detection over its specificity during the optimization process. Results: Experimental evaluations were conducted utilizing an in-house dataset comprising 167 patients with 4,895 DWI slices; subsequently, the performance of the proposed EPRA U-Net was assessed in comparison with state-of-the-art models, specifically UNet++, DeepLabV3+, and TransUNet. The experimental results suggest that EPRA U-Net attained superior performance, evidenced by a pixel-aggregated Dice of 0.8984, a per-sample Dice of 0.9469, an IoU of 0.8155, a Recall of 0.8887, a Lesion F1 of 0.9378, and an HD95 of 11.62 px. Furthermore, a clear reduction in the rate of missed lesions, specifically by 16%, 25%, and 29%, was observed when compared with UNet++, DeepLabV3+, and TransUNet, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。