用几何先验优化伪标签噪声,实现仅20%标注数据的牙科点云精准分割
GeoT: Geometry-guided Instance-dependent Transition Matrix for Semi-supervised Tooth Point Cloud Segmentation
- 引入实例依赖的转移矩阵,结合几何先验建模伪标签噪声
- 在仅20%标注数据下达到全监督方法性能,提升未标注数据利用率
- 适用于牙科点云分割场景,尤其适合标注成本高的医疗领域
从口内扫描中精确分割牙科点云是多种正畸应用的必要前提。由于牙科标注耗时费力,大量数据缺乏标注,促使半监督方法受到关注。现有半监督医学分割方法的主要挑战在于未标注数据产生的噪声伪标签。为此,我们提出GeoT,首个采用实例依赖转移矩阵(IDTM)显式建模伪标签噪声的半监督牙科分割框架。为应对由数万牙科点引发的IDTM解空间过大问题,通过两点几何先验:点级几何正则化(PLGR)增强3D空间邻接关系与IDTM空间的一致性,类别级几何平滑(CLGS)利用牙齿类别的固定空间分布以最优估计IDTM。在公开的Teeth3DS数据集和私有数据集上的大量实验表明,该方法能充分挖掘未标注数据潜力,仅用20%标注数据即可达到全监督方法水平。
原文摘要 · Abstract (English)
Achieving meticulous segmentation of tooth point clouds from intra-oral scans stands as an indispensable prerequisite for various orthodontic applications. Given the labor-intensive nature of dental annotation, a significant amount of data remains unlabeled, driving increasing interest in semi-supervised approaches. One primary challenge of existing semi-supervised medical segmentation methods lies in noisy pseudo labels generated for unlabeled data. To address this challenge, we propose GeoT, the first framework that employs instance-dependent transition matrix (IDTM) to explicitly model noise in pseudo labels for semi-supervised dental segmentation. Specifically, to handle the extensive solution space of IDTM arising from tens of thousands of dental points, we introduce tooth geometric priors through two key components: point-level geometric regularization (PLGR) to enhance consistency between point adjacency relationships in 3D and IDTM spaces, and class-level geometric smoothing (CLGS) to leverage the fixed spatial distribution of tooth categories for optimal IDTM estimation. Extensive experiments performed on the public Teeth3DS dataset and private dataset demonstrate that our method can make full utilization of unlabeled data to facilitate segmentation, achieving performance comparable to fully supervised methods with only $20\%$ of the labeled data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。