提出双向协同学习框架,提升医学图像分割的半监督性能。
Fully Differentiable Bidirectional Dual-Task Synergistic Learning for Semi-Supervised 3D Medical Image Segmentation
- 设计可微双向交互机制,实现任务间在线协同
- 在两个数据集上达到当前最优分割精度
- 适用于需要多任务联合训练的医学影像场景
半监督学习通过利用未标注数据缓解图像分割对大规模像素级标注数据的需求。由于标注成本高且需专业临床知识,高质量标注数据稀缺仍是医学图像分析的主要挑战。半监督学习展现出显著潜力,伪标签和一致性正则化成为两大主流范式。双任务协作学习作为新兴的一致性感知范式,通过建立相关任务间的预测一致性获取额外监督。然而,现有方法受限于单向交互机制(通常为回归到分割),分割结果只能离线转化为回归输出,无法充分挖掘在线双向跨任务协同的潜力。为此,我们提出完全可微的双向协同学习(DBiSL)框架,无缝集成并增强四个关键半监督学习组件:监督学习、一致性正则化、伪监督学习和不确定性估计。在两个基准数据集上的实验表明,该方法性能达到当前最优。除技术贡献外,本工作为统一的半监督学习框架设计提供新思路,建立了双任务驱动半监督学习的新架构基础,并提供一个适用于更广泛计算机视觉应用的通用多任务学习框架。代码将在录用后开源。
原文摘要 · Abstract (English)
Semi-supervised learning relaxes the need of large pixel-wise labeled datasets for image segmentation by leveraging unlabeled data. The scarcity of high-quality labeled data remains a major challenge in medical image analysis due to the high annotation costs and the need for specialized clinical expertise. Semi-supervised learning has demonstrated significant potential in addressing this bottleneck, with pseudo-labeling and consistency regularization emerging as two predominant paradigms. Dual-task collaborative learning, an emerging consistency-aware paradigm, seeks to derive supplementary supervision by establishing prediction consistency between related tasks. However, current methodologies are limited to unidirectional interaction mechanisms (typically regression-to-segmentation), as segmentation results can only be transformed into regression outputs in an offline manner, thereby failing to fully exploit the potential benefits of online bidirectional cross-task collaboration. Thus, we propose a fully Differentiable Bidirectional Synergistic Learning (DBiSL) framework, which seamlessly integrates and enhances four critical SSL components: supervised learning, consistency regularization, pseudo-supervised learning, and uncertainty estimation. Experiments on two benchmark datasets demonstrate our method's state-of-the-art performance. Beyond technical contributions, this work provides new insights into unified SSL framework design and establishes a new architectural foundation for dual-task-driven SSL, while offering a generic multitask learning framework applicable to broader computer vision applications. The code will be released on github upon acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。