用少量2D标注实现高精度3D眼底OCT分割,省时省力。
PC-Seg: Progressive Cross-View Consistency for 3D OCT Segmentation from Sparse 2D Annotations

- 通过跨视角一致性学习,从稀疏2D标注生成可靠的3D伪标签。
- 仅用0.7%标注数据,达到全监督模型的分割精度。
- 适合医疗图像标注成本高的场景,如眼科疾病诊断。
OCT图像的体积分割对眼病诊断至关重要,但需耗时的体素级标注。半监督学习可降低标注成本,但现有方法多逐切片处理,忽略3D空间上下文。本文提出PC-Seg,一种渐进式跨视角一致性框架,仅用稀疏2D标注即可训练高精度3D分割模型。不同于传统多视图方法,PC-Seg使用单一2D模型从标准B-scan与正交切片中学习跨视角一致性,生成可靠体积分割伪标签。这些伪标签被蒸馏至3D模型,并在后续协同训练阶段,由2D与3D模型通过集成伪标签相互优化。在MSHC与Duke DME数据集上的实验表明,PC-Seg仅使用约0.7%训练数据标签,即达到全监督学习的性能,优于当前最先进的半监督及视网膜层分割方法。代码已开源:https://github.com/gsisaoki/pc-seg-official。
原文摘要 · Abstract (English)
Volumetric segmentation of optical coherence tomography (OCT) images is essential for diagnosing ocular diseases but requires labor-intensive voxel-wise annotations. While semi-supervised learning (SSL) can reduce annotation costs, most existing methods process data slice by slice and fail to exploit the inherent 3D spatial context. We propose PC-Seg, a progressive cross-view consistency framework that learns high-accuracy 3D segmentation models from sparse 2D annotations. Unlike conventional multi-view approaches, PC-Seg uses a single 2D model to learn cross-view consistency from standard B-scans and orthogonal slices, thereby generating reliable volumetric pseudo-labels. These pseudo-labels are then distilled into a 3D model, followed by a co-training stage in which the 2D and 3D models mutually refine each other through ensemble pseudo-labeling. Experiments on the MSHC and Duke DME datasets demonstrate that PC-Seg achieves accuracy comparable to fully supervised learning while using labels for only about 0.7% of the training data, outperforming state-of-the-art semi-supervised and retinal layer segmentation methods. Our code is publicly available at https://github.com/gsisaoki/pc-seg-official.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。