Semi-Supervised Contrastive Learning for Breast Cancer Diagnosis in Ultrasound Imaging: a Unified Framework with Auxiliary System
To address the domain gap between natural and medical images, this study proposes a semi-supervised contrastive learning framework based on SimCLR for model pre-training. By fine-tuning the pre-trained ResNet-18 model on the BUSI dataset (780 breast ultrasound images), we achieve an average IoU of 70.27 % for segmentation and 73.13% classification accuracy. Additionally, a clinical auxiliary diagnosis system integrating tumor prediction, segmentation, batch processing, and confidence visualization is developed to enhance diagnostic efficiency.
Authors
- Zhihong Gong (ORCID: https://orcid.org/0000-0003-1129-1774)
- Zi Xiong (ORCID: https://orcid.org/0000-0002-8528-9207)
- Rong Liang (ORCID: https://orcid.org/0000-0001-5209-9426)
- Huan Lei (ORCID: https://orcid.org/0000-0002-2945-8934)
- Minjuan Zeng
Institutions
- Jiangxi University of Finance and Economics (CN)
- Southeast University (CN)
Publication Details
- Published
- 2025-05-23
- DOI
- https://doi.org/10.1109/cvidl65390.2025.11085621
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00