Confidence-Enhanced Semi-supervised Learning for Mediastinal Neoplasm Segmentation

Automated segmentation of mediastinal neoplasms in preoperative computed tomography (CT) scans is critical for accurate diagnosis. Though convolutional neural networks (CNNs) have proven effective in medical imaging analysis, the segmentation of mediastinal neoplasms, which vary greatly in shape, size, and texture, presents a unique challenge due to the inherent local focus of convolution operations. To address this limitation, we propose a confidence-enhanced semi-supervised learning framework for mediastinal neoplasm segmentation. Specifically, we introduce a confidence-enhanced module that improves segmentation accuracy over indistinct tumor boundaries by assessing and excluding unreliable predictions simultaneously, which can greatly enhance the efficiency of exploiting unlabeled data. Additionally, we implement an iterative learning strategy designed to continuously refine the estimates of prediction reliability throughout the training process, ensuring more precise confidence assessments. Quantitative analysis on a real-world dataset demonstrates that our model significantly improves the performance by leveraging unlabeled data, surpassing existing semi-supervised segmentation benchmarks. Finally, to promote more efficient academic communication, the analysis code is publicly available athttps://github.com/fxiaotong432/CEDS.