Enhancing Semi-Supervised Brain Tumor Segmentation via Dual-Uncertainty Guided 2.5-D CNN–Transformer and Voxel-Wise Contrastive Learning
Data annotation in medical image segmentation is time-consuming and expensive. Semi-supervised learning presents a viable solution. However, unlike organ segmentation, current semi-supervised methods often produce highly uncertain and erroneous predictions in the boundary regions of brain tumors due to their variable shapes, uneven distribution, and complex boundaries. In this paper, we propose a 2.5D CNN-Transformer semi-supervised segmentation model (SemiCTrans) that incorporates 3D spatial information while leveraging the strengths of different network architectures to capture local and global image features. Additionally, we design a dual-uncertainty guided module (DUGM) that automatically retains reliable prediction regions in unlabeled data. Within these reliable regions, we enforce a consistency constraint on the predictions of the CNN-Transformer network and conduct voxel-level intra-and inter-class contrastive learning for tumor segmentation. Extensive experiments on two datasets, including one publicly available dataset (BraTS2020 dataset) and one private dataset, demonstrate that the proposed SemiCTrans achieves higher performance and superior robustness than state-of-the-art brain tumor image segmentation methods. Remarkably, even with just 1% labeled data, our method achieves a Dice similarity coefficient (DSC) score of 74.7% on the BraTS2020 dataset. Code is available at https://github.com/zydlsd/SemiCTrans.
Authors
- Dairong Cao (ORCID: https://orcid.org/0000-0002-0051-3143)
- Shuhui Cai (ORCID: https://orcid.org/0000-0003-2767-9490)
- Congbo Cai (ORCID: https://orcid.org/0000-0002-0600-8594)
- Yudan Zhou (ORCID: https://orcid.org/0000-0002-3305-2072)
- Jianfeng Bao (ORCID: https://orcid.org/0000-0002-8281-9401)
- Zhong Chen (ORCID: https://orcid.org/0000-0002-1473-2224)
- Zhen Xing (ORCID: https://orcid.org/0000-0002-2104-4681)
Institutions
- Fujian Medical University (CN)
- Xiamen University (CN)
- First Affiliated Hospital of Fujian Medical University (CN)
- First Affiliated Hospital of Zhengzhou University (CN)
Publication Details
- Journal
- IEEE Transactions on Instrumentation and Measurement
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1109/tim.2025.3569939
- Citations
- 1
- Primary Topic
- Brain Tumor Detection and Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.84
Funders
- National Natural Science Foundation of China
- National Key Research and Development Program of China