Liu Jiale, Xu Yechuan, Tao Haojie, Mao Keming
Software College, Northeastern University, Shenyang, People's Republic of China.
China Mobile Virtual Reality Innovation Center, Nanchang, People's Republic of China.
Biomed Phys Eng Express. 2025 Aug 6;11(5). doi: 10.1088/2057-1976/adf3b7.
Due to the scarcity and high cost of pixel-level annotations for training data, semi-supervised learning has gradually become a key solution. Most existing methods rely on consistency regularization and pseudo-label generation, often adopting multi-branch structures to generate pseudo-labels for co-training. Such approaches, however, commonly yield low-confidence pseudo-labels from perturbed inputs, which can degrade model performance. To address these challenges, we propose a novel semi-supervised segmentation framework that leverages a multi-stage training strategy, distinguishing between the training processes for labeled and unlabeled data to enhance pseudo-label reliability. This framework effectively minimizes the negative impact of multi-branch gradient interference during co-training, reducing the adverse effects of input perturbations. Furthermore, we introduce a Balanced Uncertainty Adjustment Module (BUAM) to improve pseudo-label generation, thus maximizing data utilization efficiency. By enhancing model stability and producing more reliable pseudo-labels, the proposed multi-stage approach offers a clear advantage over existing methods. Extensive experiments on the ISIC and Cardiac MRI medical image datasets demonstrate the advantages and effectiveness of our framework, which outperforms the state-of-the-art methods.
由于训练数据的像素级标注稀缺且成本高昂,半监督学习逐渐成为关键解决方案。大多数现有方法依赖一致性正则化和伪标签生成,通常采用多分支结构进行协同训练以生成伪标签。然而,此类方法通常会从受干扰的输入中产生低置信度的伪标签,这可能会降低模型性能。为应对这些挑战,我们提出了一种新颖的半监督分割框架,该框架利用多阶段训练策略,区分标记数据和未标记数据的训练过程,以提高伪标签的可靠性。此框架有效减少了协同训练期间多分支梯度干扰的负面影响,降低了输入扰动的不利影响。此外,我们引入了平衡不确定性调整模块(BUAM)来改进伪标签生成,从而最大限度地提高数据利用效率。通过增强模型稳定性并生成更可靠的伪标签,所提出的多阶段方法相对于现有方法具有明显优势。在ISIC和心脏MRI医学图像数据集上进行的大量实验证明了我们框架的优势和有效性,其性能优于当前的先进方法。