CLUE: Contrastive Learning with Uncertainty Estimation for Semi-Supervised Medical Image Segmentation
Deep learning based medical image segmentation requires high quality pixel-level labeled data, which demands significant time and cost. Most existing semi-supervised learning methods exclude pseudo-labels with high uncertainty from training. This causes class imbalance and limits deep representation learning. In this study, we propose a semi-supervised medical image segmentation method based on contrastive learning that leverages uncertainty information. The proposed method achieved DSC and Jaccard scores of 89.48 and 81.50, with only 10% labeled data, surpassing existing methods.
Authors
- Soojin Jang (ORCID: https://orcid.org/0000-0002-2719-7646)
- Youngbin Kim (ORCID: https://orcid.org/0000-0003-1919-1575)
- Yoonji Lee (ORCID: https://orcid.org/0000-0002-2494-5792)
- Seunguk Yu
Institutions
- Chung-Ang University (KR)
Publication Details
- Published
- 2024-11-03
- DOI
- https://doi.org/10.1109/icce-asia63397.2024.10773986
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Research Foundation of Korea