用注意力增强的U-net实现乳腺病变精准分割,提升筛查可靠性。
DSU-Net: An Attention-Enhanced Dense Skip U-Net for Breast Lesion Segmentation in Mammographic Images
- 融合密集跳跃连接与注意力机制,强化特征传播与边界定位
- 在CBIS-DDSM数据集上达Dice 0.9421、IoU 0.8905,性能优异
- 适合医学影像分析、放射科辅助诊断等场景使用
乳腺癌是全球女性癌症致死的主要原因,早期检测对治疗至关重要。乳腺摄影是主要筛查手段,但可疑病灶的准确勾画仍具挑战性且存在阅片人差异。自动化分割方法可为放射科医生提供一致高效的病灶定位支持。本文提出DSU-Net,一种用于乳腺摄影图像中乳腺病变自动分割的注意力增强型密集跳跃U-Net架构。该框架结合密集跳跃连接与注意力机制,以改善特征传播、保留空间信息并增强病灶边界分割。实验基于数字乳腺筛查数据库的精选子集(CBIS-DDSM)。为缓解严重前景-背景不平衡问题,训练时采用包含Dice损失、焦点损失和二元交叉熵损失的复合损失函数。所提模型在验证集上取得0.9421的Dice相似系数、0.8905的交并比、0.9711的准确率以及0.9878的AUC-ROC值。定性评估显示对不同大小与形态的病灶均能精确勾画,定量结果证实了对病灶与背景区域的鲁棒区分能力。这些发现表明,DSU-Net可实现乳腺摄影图像中准确可靠的病灶分割,并凸显了注意力引导深度学习在计算机辅助乳腺癌筛查与诊断中的潜力。
原文摘要 · Abstract (English)
Breast cancer remains one of the leading causes of cancer-related mortality among women worldwide, making early detection essential for effective treatment. Mammography is the primary screening modality; however, accurate delineation of suspicious lesions remains challenging and subject to inter-observer variability. Automated segmentation methods can assist radiologists by providing consistent and efficient lesion localization. This study presents DSU-Net, an attention-enhanced Dense Skip U-Net architecture for automated breast lesion segmentation in mammographic images. The proposed framework integrates dense skip connections and attention mechanisms to improve feature propagation, preserve spatial information, and enhance lesion boundary delineation. Experiments were conducted using the Curated Breast Imaging Subset of the Digital Database for Screening Mammography (CBIS-DDSM). To address severe foreground-background imbalance, a composite loss function combining Dice loss, focal loss, and binary cross-entropy loss was employed during training. The proposed model achieved a Dice Similarity Coefficient of 0.9421, an Intersection over Union of 0.8905, an accuracy of 0.9711, and an AUC-ROC of 0.9878 on the validation dataset. Qualitative evaluation demonstrated accurate delineation of lesions with varying sizes and morphologies, while quantitative results confirmed robust discrimination between lesion and background regions. These findings demonstrate that DSU-Net provides accurate and reliable breast lesion segmentation in mammographic images and highlights the potential of attention-guided deep learning for computer-aided breast cancer screening and diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。