通过双向建模背景提升医学图像分割精度,仅用20%标注数据超越全监督方法。
Background Matters: A Cross-view Bidirectional Modeling Framework for Semi-supervised Medical Image Segmentation
- 引入背景建模作为辅助视角,与前景模型双向协同优化。
- 在胰腺分割上仅用20%标签即达84.57% Dice 分数,超全监督方法。
- 适合追求低标注成本、高精度的医学图像分割研究者使用。
半监督医学图像分割(SSMIS)利用未标注数据减少对人工标注的依赖。然而,当前主流方法多聚焦于前景建模(仅分割目标区域),忽视了显式建模背景的潜力。本研究从理论和实证层面证明:背景预测的高置信度可增强对应前景建模的置信度。基于此,我们提出跨视图双向建模框架(CVBM),首次将背景建模引入分割流程,作为辅助视角提供互补监督信号,提升前景模型信心。此外,CVBM设计了双向一致性机制,确保前景预测与背景引导预测相互对齐。大量实验表明,该方法在LA、Pancreas、ACDC、HRF等数据集上均达到领先性能。尤其在胰腺数据集上,仅使用20%标注数据即实现84.57%的Dice分数,超越全监督方法(83.89%)。代码已开源。
原文摘要 · Abstract (English)
Semi-supervised medical image segmentation (SSMIS) leverages unlabeled data to reduce reliance on manually annotated images. However, current SOTA approaches predominantly focus on foreground-oriented modeling (i.e., segmenting only the foreground region) and have largely overlooked the potential benefits of explicitly modeling the background region. Our study theoretically and empirically demonstrates that highly certain predictions in background modeling enhance the confidence of corresponding foreground modeling. Building on this insight, we propose the Cross-view Bidirectional Modeling (CVBM) framework, which introduces a novel perspective by incorporating background modeling to improve foreground modeling performance. Within CVBM, background modeling serves as an auxiliary perspective, providing complementary supervisory signals to enhance the confidence of the foreground model. Additionally, CVBM introduces an innovative bidirectional consistency mechanism, which ensures mutual alignment between foreground predictions and background-guided predictions. Extensive experiments demonstrate that our approach achieves SOTA performance on the LA, Pancreas, ACDC, and HRF datasets. Notably, on the Pancreas dataset, CVBM outperforms fully supervised methods (i.e., DSC: 84.57% vs. 83.89%) while utilizing only 20% of the labeled data. Our code is publicly available at https://github.com/caoluyang0830/CVBM.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。