用小波与注意力融合提升超声乳腺肿瘤分割精度
WDFFU-Mamba: A Wavelet-guided Dual-attention Feature Fusion Mamba for Breast Tumor Segmentation in Ultrasound Images
- 小波去噪增强高频特征,改善低层表示
- 双注意力融合模块提升语义一致性,Dice达0.917
- 适合临床超声图像分割,泛化性强
乳腺超声(BUS)图像分割在辅助临床诊断和早期肿瘤筛查中至关重要。然而,斑点噪声、成像伪影、病灶形态不规则及边界模糊等问题严重阻碍了精准分割。为此,本文提出一种新型分割网络WDFFU-Mamba,将小波引导增强与双注意力特征融合结合于U型Mamba架构中。设计了波段去噪高频引导特征(WHF)模块,通过抑制噪声的高频线索增强低层特征表达;引入双注意力特征融合(DAFF)模块,有效融合跳跃连接与语义特征,提升上下文一致性。在两个公开的BUS数据集上的大量实验表明,WDFFU-Mamba在分割精度上显著优于现有方法,Dice系数达到0.917,HD95降低至4.86像素。小波域增强与注意力融合的结合显著提升了分割准确率与鲁棒性,同时保持计算高效。该模型不仅性能优越,且跨数据集泛化能力良好,适用于真实临床环境中的乳腺肿瘤超声分析。
原文摘要 · Abstract (English)
Breast ultrasound (BUS) image segmentation plays a vital role in assisting clinical diagnosis and early tumor screening. However, challenges such as speckle noise, imaging artifacts, irregular lesion morphology, and blurred boundaries severely hinder accurate segmentation. To address these challenges, this work aims to design a robust and efficient model capable of automatically segmenting breast tumors in BUS images.We propose a novel segmentation network named WDFFU-Mamba, which integrates wavelet-guided enhancement and dual-attention feature fusion within a U-shaped Mamba architecture. A Wavelet-denoised High-Frequency-guided Feature (WHF) module is employed to enhance low-level representations through noise-suppressed high-frequency cues. A Dual Attention Feature Fusion (DAFF) module is also introduced to effectively merge skip-connected and semantic features, improving contextual consistency.Extensive experiments on two public BUS datasets demonstrate that WDFFU-Mamba achieves superior segmentation accuracy, significantly outperforming existing methods in terms of Dice coefficient and 95th percentile Hausdorff Distance (HD95).The combination of wavelet-domain enhancement and attention-based fusion greatly improves both the accuracy and robustness of BUS image segmentation, while maintaining computational efficiency.The proposed WDFFU-Mamba model not only delivers strong segmentation performance but also exhibits desirable generalization ability across datasets, making it a promising solution for real-world clinical applications in breast tumor ultrasound analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。