用双注意力机制提升新生儿脑超声图像的脑室分割精度
Dual Attention Residual U-Net for Accurate Brain Ultrasound Segmentation in IVH Detection
- 引入CBAM与稀疏注意力层,兼顾局部细节与全局上下文
- 在脑超声数据集上达到89.04%的骰子系数和81.84%的交并比
- 适合做新生儿脑出血早期诊断的医学图像分析研究者
颅内出血(IVH)是早产儿严重的神经并发症,需通过脑超声(US)图像实现早期精准检测以改善临床预后。尽管深度学习方法在辅助诊断中展现出潜力,但在捕捉局部空间细节与全局上下文依赖方面仍面临挑战。本文提出一种改进的残差U-Net架构,融合两种互补注意力机制:卷积块注意力模块(CBAM)与稀疏注意力层(SAL)。CBAM增强空间与通道特征的精炼能力,SAL采用双分支设计,稀疏注意力剔除低置信度查询-键对以抑制噪声,密集注意力确保信息充分传播。在脑超声数据集上的大量实验表明,该方法在脑室分割任务中达到89.04%的骰子系数与81.84%的交并比,显著优于现有方法,验证了空间精炼与注意力稀疏性结合的有效性。代码已开源。
原文摘要 · Abstract (English)
Intraventricular hemorrhage (IVH) is a severe neurological complication among premature infants, necessitating early and accurate detection from brain ultrasound (US) images to improve clinical outcomes. While recent deep learning methods offer promise for computer-aided diagnosis, challenges remain in capturing both local spatial details and global contextual dependencies critical for segmenting brain anatomies. In this work, we propose an enhanced Residual U-Net architecture incorporating two complementary attention mechanisms: the Convolutional Block Attention Module (CBAM) and a Sparse Attention Layer (SAL). The CBAM improves the model's ability to refine spatial and channel-wise features, while the SAL introduces a dual-branch design, sparse attention filters out low-confidence query-key pairs to suppress noise, and dense attention ensures comprehensive information propagation. Extensive experiments on the Brain US dataset demonstrate that our method achieves state-of-the-art segmentation performance, with a Dice score of 89.04% and IoU of 81.84% for ventricle region segmentation. These results highlight the effectiveness of integrating spatial refinement and attention sparsity for robust brain anatomy detection. Code is available at: https://github.com/DanYuan001/BrainImgSegment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。