通过多专家注意力机制提升乳腺超声癌变识别准确率
Cross-Stage Attention Multi-Expert Network for Radiologist-Inspired Breast Ultrasound Diagnosis

- 设计跨阶段注意力模块增强特征提取,融合肿瘤整体、核心与边界信息
- 在2129张图像上达到96.33%准确率,远超基线模型
- 无需修改网络结构,可适配多种主干网络,适合临床辅助诊断
乳腺超声是早期乳腺癌筛查的重要无创手段,但因肿瘤异质性、边界模糊及数据不平衡,自动良恶性分类仍具挑战。本文提出交叉阶段注意力混合专家网络(CSA-MoE-Net),以改进特征表示与分类精度。采用增强型跨阶段注意力的ResNet-18作为主干网络,其中跨阶段注意力模块自适应重校准多层次特征,突出关键肿瘤特征并抑制冗余。三分支混合专家(MoE)模块分别从全肿瘤图像、肿瘤核心和边界学习互补特征,自适应门控网络融合生成融合专家特征(FEF)。在包含2,129张乳腺超声图像的平衡数据集上,经20次独立运行平均测试,模型准确率达96.33%,精确率为94.09%,召回率为98.53%,F1分数为96.25%,AUC达99.50%。相较基线ResNet-18,各项指标分别提升3.01、0.70、5.37、2.98和5.42个百分点。该机制无需侵入式改造,可无缝嵌入VGG-16、DenseNet-121等网络,实现稳定性能提升,为计算机辅助诊断提供可靠支持。
原文摘要 · Abstract (English)
Breast ultrasound imaging is an important noninvasive method for early breast cancer diagnosis, but automatic benign/malignant classification remains challenging due to tumor heterogeneity, blurred boundaries, and data imbalance. To improve feature representation and classification accuracy, this paper proposes the Cross-Stage Attention Mixture-of-Experts Network (CSA-MoE-Net). It adopts a Cross-Stage Attention-enhanced ResNet-18 as the backbone, in which the Cross-Stage Attention module adaptively recalibrates multi-level features, thereby enhancing key tumor features and suppressing redundancy. A three-branch Mixture of Experts (MoE) Block learns complementary features from the Whole Tumor Image, Tumor Core, and Boundary, and an Adaptive Gating Network fuses them to capture morphological, textural, and contextual information. The fused features are denoted as Fused Expert Feature (FEF) in the architecture. Experiments on a balanced dataset of 2,129 breast ultrasound images show that, averaged over 20 independent runs, the model achieves an accuracy of 96.33\%, precision of 94.09\%, recall of 98.53\%, F1-score of 96.25\%, and AUC of 99.50\%. Compared to the baseline ResNet-18, these metrics improve by 3.01, 0.70, 5.37, 2.98, and 5.42 percentage points, respectively. The proposed mechanism requires no invasive modification and can be seamlessly embedded into VGG-16, DenseNet-121, etc., yielding stable performance gains, thus providing reliable support for computer-aided diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。