arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2605.20536cs.CV

HADS-Net:一种融合注意力增强的双流网络,用于乳腺超声图像分类

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

NedumCares

Chinedu Emmanuel Mbonu, Blessing Nwamaka Iduh, Joseph Ikechukwu Odo, Doris Chinedu Asogwa

AI总结:

本文提出HADS-Net,一种融合注意力增强的双流网络,通过两个并行路径利用全局纹理和局部边界线索,结合物理信息增强,以提高乳腺超声图像分类的准确性。

摘要:

准确地将乳腺超声图像分类为良性、恶性和正常类别是至关重要的临床任务,但受到斑点噪声、声影效应和类间视觉模糊的阻碍。现有的深度学习方法依赖于单流架构,使用通用增强方法,忽略了超声成像的物理特性,并且没有先前的方法专门针对被确定为最诊断性视觉线索的病变边界特征进行处理。我们提出了HADS-Net,一种混合注意力增强的双流网络,通过两个并行路径利用全局纹理和局部边界线索。流1应用物理信息增强模拟斑点噪声、声影效应和增益变化,在提取特征前使用预训练的EfficientNet-B3投影到512维空间。流2提取Sobel边缘图,经过轻量级CNN处理后投影到相同的512维空间。交叉注意力融合模块允许纹理流选择性地查询边界特征,生成联合优化的表示,由通过自适应类别加权焦点损失训练的MLP进行分类。使用五折分层交叉验证,在50个周期中使用余弦退火,选择验证损失最低的全局最佳检查点。在BUSI数据集上,HADS-Net实现了96.58%的准确率,宏ROC-AUC为0.9978,宏F1为0.9654,以及良性、恶性和正常类别的F1分数分别为0.970、0.951和0.976。没有恶性病变被误分类为正常。这些结果证实,模态特定的增强与跨模态注意力融合是超声波乳腺癌诊断的有效策略。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑