A diffusion-assisted attention U-Net++ framework for accurate breast tumor segmentation in ultrasound imaging

View at publisher
Authors: Ali Hamza, Martin Mézl
Cited by: 0

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