UGB-SegNet: Uncertainty-Guided Multi-Scale Attention and Boundary-Aware Learning for Breast Ultrasound Lesion Segmentation
Alam, T.; Hsu, F. R.
Show abstract
Accurate breast lesion segmentation in ultrasound is difficult because speckle noise, heterogeneous lesion appearance, and poorly defined margins can obscure clinically relevant boundaries. This work presents UGB-SegNet, an uncertainty-guided breast ultrasound segmentation network that combines four components: multi-scale convolutional block attention (CBAM) at three EfficientNet-B0 encoder stages, a learnable feature pyramid network for adaptive multi-scale fusion, an entropy-guided decoder that routes fine-grained encoder information toward uncertain regions, and a boundary-aware composite loss with deep supervision. The model was evaluated on the public BUSI dataset (647 annotated lesion images) using a stratified 454/64/129 train/validation/test split. On the held-out test set, UGB-SegNet achieved 81.30% Dice, 68.49% IoU, 81.71% sensitivity, 98.14% specificity, and 15.63-pixel HD95. Under identical training conditions it outperformed five directly trained baselines: U-Net, Attention U-Net, SegNet, DeepLabV3+, and TransUNet. A five-fold resampling experiment over the development set with the held-out test set fixed yielded a mean Dice of 76.36% +/- 1.66% (95% CI 74.90-77.81). Reported ablations attribute gains of +2.06 Dice points to multi-scale CBAM, +1.52 to learnable FPN fusion, +1.22 to the uncertainty-guided decoder, and +1.24 to boundary-aware training. UGB-SegNet contains 14.00M parameters and runs at 62.4 FPS at 224 x 224 resolution on the tested NVIDIA GPU, providing a compact accuracy-efficiency trade-off for future prospective clinical validation.
Matching journals
The top 9 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Model uncertainty estimates for deep learning mammographic density prediction using ordinal and classification approaches 93%
- Breast density prediction from low and standard dose mammograms using deep learning: effect of image resolution and model training approach on prediction quality 92%
- Mammographic density assessed using deep learning in women at high risk of developing breast cancer: the effect of weight change on density 89%
Similar papers in this journal
- SCU-Net: A deep learning method for segmentation and quantification of breast arterial calcifications on mammograms 96%
- Phase Recognition in Contrast-Enhanced CT Scans based on Deep Learning and Random Sampling 93%
- Necessity and Impact of Specialization of Large Foundation Model for Medical Segmentation Tasks 92%
Similar papers in this journal
- Automated detection of the HER2 gene amplification status in Fluorescence in situ hybridization images for the diagnostics of cancer tissues 94%
- Incremental Learning Approach for Semantic Segmentation of Skin Histology Images 93%
- Reproducible And Clinically Translatable Deep Neural Networks For Cervical Screening 92%
Similar papers in this journal
- A Machine Learning Ensemble Based on Radiomics to Predict BI-RADS Category and Reduce the Biopsy Rate of Ultrasound-Detected Suspicious Breast Masses 94%
- An Explainable Web-Based Diagnostic System for Alzheimer's Disease Using XRAI and Deep Learning on Brain MRI 89%
- Deep neural frameworks improve the accuracy of general practitioners in the classification of pigmented skin lesions 89%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.