Semi-Supervised Medical Image Segmentation with Cross-View Consistency and Contrastive Learning
Medical image segmentation plays a crucial role in many clinical applications. To alleviate the dependency on massive annotations, semi-supervised learning has attracted increasing attention. However, these methods face significant intra-class and inter-class variation and do not fully utilize the critical multi-view information inherent in medical images. This study proposes a novel network, CV-Net, which integrates multi-view information for semi-supervised medical image segmentation. Concretely, the network is based on Mean-Teacher architecture which largely narrows the empirical distribution gap between labeled and unlabeled data. The proposed cross-view consistency regularization module incorporates a dual-branch attention architecture to integrate consistent semantics while focusing on details, enhancing feature extraction capabilities. The proposed bi-semantic contrastive learning module leverages limited labels and explore pseudo-labels to define semantically similar regions, enhancing the representation capacity. Experiments conducted on two datasets demonstrated the effectiveness of the proposed network. CV-Net showed significant improvements across four metrics, evident with both 5% and 10% labeled data. Specifically, with 5% labeled data, the mean Dice increased by 1.37%. Compared with previous state-of-the-art methods, CV-Net achieved the best results, notably reducing both intra-class and inter-class errors.
Authors
- Aimin Hao (ORCID: https://orcid.org/0000-0002-5774-6706)
- Xuehao Wang (ORCID: https://orcid.org/0000-0002-4416-693X)
- Kexuan Li (ORCID: https://orcid.org/0009-0006-0765-0032)
- Shuai Li (ORCID: https://orcid.org/0000-0002-8696-8594)
- Zhen Sun (ORCID: https://orcid.org/0000-0001-7430-8081)
- Yi Xiao (ORCID: https://orcid.org/0000-0002-4309-1227)
- Bo Sun (ORCID: https://orcid.org/0000-0003-1168-1051)
- Huadan Xue (ORCID: https://orcid.org/0000-0002-4278-2165)
- Jingjuan Liu
Institutions
- Chinese Academy of Medical Sciences & Peking Union Medical College (CN)
- Peking Union Medical College Hospital (CN)
- State Key Laboratory of Virtual Reality Technology and Systems
- Beihang University (CN)
Publication Details
- Published
- 2024-12-03
- DOI
- https://doi.org/10.1109/bibm62325.2024.10822203
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China