Wavelet-Enhanced CNN for Breast Ultrasound Classification Under Speckle Noise
Background/Objectives: Ultrasound is widely used for breast cancer screening and diagnosis, particularly in low- and middle-income settings, but its diagnostic reliability is often compromised by speckle noise that degrades lesion margins and tissue texture. This study proposes a compact convolutional neural network architecture that replaces standard max or average pooling layers with wavelet-based pooling using Symlet families, and optionally includes wavelet-domain preprocessing to suppress input noise. Methods: We conducted 108 experiments across six pooling configurations (avg, max, Sym2 ± preprocessing, Sym4 + preprocessing, Sym6 + preprocessing), two network depths, three batch sizes, and three simulated speckle levels (0%, 10%, 20%). Results: The proposed wavelet-based pooling framework showed consistently stronger in-domain performance than conventional pooling strategies across clean and speckle-corrupted settings, with the Sym2 + preprocessing configuration giving the best overall results. The model achieved 93.90% accuracy and 98.89% ROC AUC under clean internal test conditions and maintained stable performance under increased simulated noise levels. However, external validation on the independent BrEaST-Lesions-USG dataset revealed substantial performance degradation, with accuracy decreasing to 53.97% and ROC AUC to 0.4713, indicating limited cross-dataset generalization. Conclusions: These findings suggest that wavelet pooling is an effective architectural modification for improving in-domain robustness under controlled perturbation, although additional strategies are still required before reliable real-world deployment can be claimed.
Institutions
- Suranaree University of Technology (TH)
Publication Details
- Journal
- Biomedicines
- Published
- 2026-05-19
- DOI
- https://doi.org/10.3390/biomedicines14051151
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Suranaree University of Technology