arXiv:2605.20536cs.CV2026-05

通过物理增强与双流注意力融合,提升乳腺超声图像分类准确率

HADS-Net:A Hybrid Attention-Augmented Dual-Stream Network with Physics-Informed Augmentation for Breast Ultrasound Image Classification

论文配图:HADS-Net:A Hybrid Attention-Augmented Dual-Stream Network with Physics-Informed Augmentation for Breast Ultrasound Image Classification
图 1 · 摘自论文原文
  • 双流结构分别提取纹理与边界特征,用跨模态注意力融合
  • 在BUSI数据集上达到96.58%准确率,恶性病变无误判
  • 专为超声成像物理特性设计增强,适合医学影像诊断研究者

乳腺超声图像的良恶性及正常分类是临床关键任务,但受斑点噪声、声影和类间视觉模糊影响。现有深度学习方法多采用单一通道架构,通用增强策略忽略超声采集物理特性,且未专门建模病灶边界这一最具诊断价值的视觉线索。本文提出HADS-Net:一种混合注意力增强的双流网络,通过两条并行路径分别捕捉全局纹理与局部边界特征。第一流使用模拟斑点噪声、声影和增益变化的物理感知增强,经预训练EfficientNet-B3投影至512维;第二流提取Sobel边缘图,由轻量CNN处理后同样映射到512维空间。交叉注意力融合模块使纹理流可选择性查询边界特征,生成联合优化表示,由带自适应类别加权焦点损失的MLP分类。采用五折分层交叉验证,50轮训练中以最低验证损失对应的最优检查点进行测试。在BUSI数据集上,模型取得96.58%准确率,宏平均ROC-AUC为0.9978,宏平均F1为0.9654,良性、恶性、正常类别的个体F1分别为0.970、0.951、0.976,且所有恶性病变均未被误判为正常。结果表明,结合模态特异性增强与跨模态注意力融合,是提升超声乳腺癌诊断效能的有效策略。

原文摘要 · Abstract (English)

Accurate classification of breast ultrasound images into benign, malignant, and normal categories is a critical clinical task complicated by speckle noise, acoustic shadowing, and inter-class visual ambiguity. Existing deep learning methods rely on single-stream architectures with generic augmentation that ignores ultrasound acquisition physics, and no prior method dedicates a stream to the lesion boundary features identified as the most diagnostically significant visual cue. We propose HADS-Net, a Hybrid Attention-Augmented Dual-Stream Network exploiting global texture and local boundary cues through two parallel pathways. Stream 1 applies physics-informed augmentation simulating speckle noise, acoustic shadowing, and gain variation before extracting features via pretrained EfficientNet-B3 projected to 512 dimensions. Stream 2 extracts Sobel edge maps processed by a lightweight CNN projected to the same 512-dimensional space. A cross-attention fusion module allows the texture stream to selectively query boundary features, producing a jointly optimised representation classified by an MLP trained with adaptive class-weighted focal loss. Five-fold stratified cross-validation with cosine annealing over 50 epochs is used, with the globally best checkpoint selected by lowest validation loss evaluated on a held-out test set. On the BUSI dataset, HADS-Net achieves 96.58% accuracy, macro ROC-AUC of 0.9978, macro F1 of 0.9654, and per-class F1-scores of 0.970, 0.951, and 0.976 for benign, malignant, and normal. No malignant lesion is misclassified as normal. These results confirm that modality-specific augmentation with cross-modal attention fusion is an effective strategy for ultrasound-based breast cancer diagnosis.

医学影像超声诊断双流网络注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。