UltraSegNet: A Hybrid Deep Learning Framework for Enhanced Breast Cancer Segmentation and Classification on Ultrasound Images
Source abstract
Abstract
Segmenting a breast ultrasound image is still challenging due to the presence of speckle noise, dependency on the operator, and the variation of image quality. This paper presents the UltraSegNet architecture that addresses these challenges through three key technical innovations: This work adds three things: (1) a changed ResNet-50 backbone with sequential 3 convolutions to keep fine anatomical details that are needed for finding lesion boundaries; (2) a computationally efficient regional attention mechanism that works on high-resolution features without using a transformer’s extra memory; and (3) an adaptive feature fusion strategy that changes local and global features based on how the image is being used. Extensive evaluation on two distinct datasets demonstrates UltraSegNet’s superior performance: On the BUSI dataset, it obtains a precision of 0.915, a recall of 0.908, and an F1 score of 0.911. In the UDAIT dataset, it achieves robust performance across the board, with a precision of 0.901 and recall of 0.894. Importantly, these improvements are achieved at clinically feasible computation times, taking 235 ms per image on standard GPU hardware. Notably, UltraSegNet does amazingly well on difficult small lesions (less than 10 mm), achieving a detection accuracy of 0.891. This is a huge improvement over traditional methods that have a hard time with small-scale features, as standard models can only achieve 0.63–0.71 accuracy. This improvement in small lesion detection is particularly crucial for early-stage breast cancer identification. Results from this work demonstrate that UltraSegNet can be practically deployable in clinical workflows to improve breast cancer screening accuracy.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: AI in cancer detection · Radiomics and Machine Learning in Medical Imaging · Brain Tumor Detection and Classification
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (60)
Show source-provided references
1. World Health Organization. Breast cancer [Internet]. 2021 [cited 2024 Oct 21]. Available from: https://www.who.int/news-room/fact-sheets/detail/breast-cancer.
2. Global cancer statistics 2020: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries
3. Community pharmacists’ readiness for breast cancer mammogram promotion: a national survey from Jordan10.2147/JMDH.S471151 ↗
4. Breast-cancer tumor size, overdiagnosis, and mammography screening effectiveness10.1056/NEJMoa1600249 ↗
5. Comparison of the performance of screening mammography, physical examination, and breast US and evaluation of factors that influence them: an analysis of 27,825 patient evaluations10.1148/radiol.2251011667 ↗
6. Combined screening with ultrasound and mammography vs mammography alone in women at elevated risk of breast cancer10.1001/jama.299.18.2151 ↗
7. Assessing improvement in detection of breast cancer with three-dimensional automated breast US in women with dense breast tissue: the SomoInsight Study10.1148/radiol.14132832 ↗
8. Solid breast nodules: use of sonography to distinguish between benign and malignant lesions10.1148/radiology.196.1.7784555 ↗
9. Variability in radiologists’ interpretations of mammograms10.1056/NEJM199412013312206 ↗
10. A survey on deep learning in medical image analysis10.1016/j.media.2017.07.005 ↗
11. A secure internet of medical things framework for breast cancer detection in sustainable smart cities10.3390/electronics12040858 ↗
12. Multi-task deep learning for breast ultrasound image analysis: lesion detection, classification, and segmentation
13. Attention-guided network for breast ultrasound image segmentation with multi-scale feature enhancement
14. A survey on deep learning-based medical image analysis
15. Hybrid CNN-Transformer architecture for medical image segmentation: application to breast ultrasound
16. Dual-path network with adaptive attention for robust breast lesion segmentation in ultrasound images
17. A metaheuristic approach based on coronavirus herd immunity optimiser for breast cancer diagnosis10.1007/s10586-024-04360-3 ↗
18. Expedited colorectal cancer detection through a dexterous hybrid CADx system with enhanced image processing and augmented polyp visualization10.1109/ACCESS.2025.3532807 ↗
19. Current methods in medical image segmentation10.1146/annurev.bioeng.2.1.315 ↗
20. CNN-based deep learning model for early identification and categorization of melanoma skin cancer using medical imaging10.1007/s42979-024-03270-w ↗
21. Ultrasound image segmentation: a survey10.1109/TMI.2006.877092 ↗
22. A comparative analysis of convolutional neural networks for breast cancer prediction
23. Marine predators algorithm with deep learning-based leukemia cancer classification on medical images
24. A computational approach to edge detection10.1109/TPAMI.1986.4767851 ↗
25. Watersheds in digital spaces: an efficient algorithm based on immersion simulations10.1109/34.87344 ↗
26. Snakes: active contour models10.1007/BF00133570 ↗
27. Theory of edge detection
28. A deep learning based architecture for malaria parasite detection10.11591/eei.v13i1.5485 ↗
29. Seeded region growing10.1109/34.295913 ↗
30. Stochastic relaxation, Gibbs distributions, and the Bayesian restoration of images10.1109/TPAMI.1984.4767596 ↗
31. Unsupervised texture segmentation using Gabor filters10.1016/0031-3203(91)90143-S ↗
32. A theory for multiresolution signal decomposition: the wavelet representation10.1109/34.192463 ↗
33. Vapnik · The nature of statistical learning theory · 199510.1007/978-1-4757-2440-0 ↗
34. Random forests10.1023/A:1010933404324 ↗
35. Performance-based classifier combination in atlas-based image segmentation using expectation-maximization parameter estimation10.1109/TMI.2004.830803 ↗
36. Deep learning in medical image analysis10.1146/annurev-bioeng-071516-044442 ↗
37. A dual-branch network for ultrasound image segmentation10.1016/j.bspc.2024.107368 ↗
38. Brain tumor segmentation in MRI with multi-modality spatial information enhancement and boundary shape correction10.1016/j.patcog.2024.110553 ↗
39. Hatamizadeh A, Tang H, Roth HR, Xu D. UNETR: transformers for 3D medical image segmentation. arXiv:2103.10504. 2022.10.1109/WACV51458.2022.00181 ↗
40. Chen J, Lu Y, Yu Q, Luo X, Adeli E, Wang Y, et al. TransUNet: transformers make strong encoders for medical image segmentation. arXiv:2102.04306. 2021.
41. A review on deep learning in medical image analysis
42. Med-Former: a transformer based architecture for medical image classification
43. Transformers in medical imaging: a survey
44. Transformers are RNNs: fast autoregressive transformers with linear attention
45. CCNet: criss-cross attention for semantic segmentation
46. Axial-DeepLab: stand-alone axial-attention for panoptic segmentation
47. Efficient transformer architectures for medical image analysis
48. TransUNet: rethinking the U-Net architecture design for medical image segmentation through the lens of transformers10.1016/j.media.2024.103280 ↗
49. Dataset of breast ultrasound images10.1016/j.dib.2019.104863 ↗
50. Automated breast ultrasound lesions detection using convolutional neural networks10.1109/JBHI.2017.2731873 ↗
51. ResNet and its application to medical image processing: research progress and challenges10.1016/j.cmpb.2023.107660 ↗
52. Squeeze-and-excitation networks
53. Oktay O, Schlemper J, Folgoc LL, Lee M, Heinrich M, Misawa K, et al. Attention U-Net: learning where to look for the pancreas. arXiv:1804.03999. 2018.
54. Imagenet classification with deep convolutional neural networks
55. Xception: deep learning with depthwise separable convolutions
56. Deep residual learning for imagerecognition
57. Deep learning architectures for breast cancer detection and classification
58. Deep learning applications in breast cancer diagnosis
59. Breast cancer diagnosis using deep neural networks
60. Zhou HY, Yu L, Wang L, Yang X. A review of vision transformer for medical image analysis: challenges and solutions. arXiv:2202.12165. 2022.