用区域感知指导学习提升牙科CBCT图像分割精度
RAIL: Region-Aware Instructive Learning for Semi-Supervised Tooth Segmentation in CBCT
- 双组双学生架构,通过交替训练实现知识迁移与区域指导
- 在标注少的情况下,分割准确率超越现有最优方法
- 适合牙科影像分析、医疗图像分割等低标注场景使用
半监督学习已成为从CBCT扫描中进行3D牙齿分割的有力方法,尤其在标注数据极少的情况下。然而,现有方法仍面临两大挑战:监督训练中结构模糊或误标区域的修正监督不足,以及无监督阶段由不可靠伪标签导致的性能下降。为此,我们提出区域感知指导学习(RAIL),一种双组双学生半监督框架。每组包含两个由共享教师网络指导的学生模型。通过在两组间交替训练,RAIL促进组间知识传递和协同区域感知指导,同时减少对单一模型特征的过拟合。具体地,引入两种指导机制:分歧聚焦监督(DFS)控制器仅在学生预测与真实标签及最优学生结果均不一致的区域施加监督,集中修正结构模糊或误标区域;在无监督阶段,置信度感知学习(CAL)调制器强化高置信度区域的一致性,降低低置信度预测的影响,避免模型学习不稳定模式,提升伪标签整体可靠性。在四个CBCT牙齿分割数据集上的大量实验表明,RAIL在标注有限条件下优于当前最先进方法。
原文摘要 · Abstract (English)
Semi-supervised learning has become a compelling approach for 3D tooth segmentation from CBCT scans, where labeled data is minimal. However, existing methods still face two persistent challenges: limited corrective supervision in structurally ambiguous or mislabeled regions during supervised training and performance degradation caused by unreliable pseudo-labels on unlabeled data. To address these problems, we propose Region-Aware Instructive Learning (RAIL), a dual-group dual-student, semi-supervised framework. Each group contains two student models guided by a shared teacher network. By alternating training between the two groups, RAIL promotes intergroup knowledge transfer and collaborative region-aware instruction while reducing overfitting to the characteristics of any single model. Specifically, RAIL introduces two instructive mechanisms. Disagreement-Focused Supervision (DFS) Controller improves supervised learning by instructing predictions only within areas where student outputs diverge from both ground truth and the best student, thereby concentrating supervision on structurally ambiguous or mislabeled areas. In the unsupervised phase, Confidence-Aware Learning (CAL) Modulator reinforces agreement in regions with high model certainty while reducing the effect of low-confidence predictions during training. This helps prevent our model from learning unstable patterns and improves the overall reliability of pseudo-labels. Extensive experiments on four CBCT tooth segmentation datasets show that RAIL surpasses state-of-the-art methods under limited annotation. Our code will be available at https://github.com/Tournesol-Saturday/RAIL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。