A diffusion-assisted attention U-Net++ framework for accurate breast tumor segmentation in ultrasound imaging
Abstract
Accurate segmentation of breast tumors in ultrasound (US) imaging is challenging due to speckle noise, low contrast, and irregular lesion morphology. This study proposes a diffusion-assisted Attention U-Net++ framework that integrates anisotropic diffusion filtering with nested attention skip pathways to enhance noise suppression and boundary precision. The anisotropic diffusion stage reduces speckle artifacts while preserving structural edges, and the Attention U-Net++ network refines multiscale feature fusion through attention gating. Experiments were conducted on the Breast Ultrasound Images (BUSI) dataset comprising 780 images from 600 subjects. The proposed framework achieved a Dice coefficient of 0.965, IoU of 0.932, precision of 0.97, recall of 0.98 for benign lesions, and precision of 0.965, recall of 0.94 for malignant lesions, with an overall accuracy of 0.95. Grad-CAM visualization confirmed that the model focused on clinically relevant tumor regions, enhancing interpretability. The results demonstrate that combining diffusion-based preprocessing with attention-guided segmentation yields robust, efficient, and explainable performance, offering a reliable foundation for future computer-aided breast-ultrasound diagnostic systems.
Fields of Study & Subjects
Medical Image Segmentation Techniques
AI in cancer detection
Brain Tumor Detection and Classification
Publication details
- Journal / source
- International Conference on Advances in Image Processing (ICAIP)
- Pages
- 47–47
- Publication date
- 6 February 2026