提出半监督框架SemiTooth,提升多源牙体分割准确率
SemiTooth: a Generalizable Semi-supervised Framework for Multi-Source Tooth Segmentation
- 多教师-多学生架构,分源学习未标注数据
- 在三源数据集上实现当前最优性能
- 适合临床牙科影像分析与跨机构数据应用
随着人工智能发展,智能牙科在临床诊断治疗中日益重要。牙体结构分割作为核心任务,在锥形束计算机断层扫描(CBCT)中取得显著进展。然而,全标注数据获取困难,以及不同机构间多源数据采集差异,导致数据质量低、体素级不一致和域间偏差。因此,高效利用多源未标注数据成为关键问题。本文提出SemiTooth,一种通用的半监督多源牙体分割框架。首先构建了包含三个来源、不同标注级别数据的MS3Toothset数据集;随后设计多教师-多学生框架,各学生网络分别从对应来源的未标注数据中学习,由其对应教师监督。此外,引入更严格的加权置信度约束,增强多源准确性。在MS3Toothset上的大量实验验证了SemiTooth的有效性与优越性,在半监督与多源牙体分割场景下达到当前最佳表现。
原文摘要 · Abstract (English)
With the rapid advancement of artificial intelligence, intelligent dentistry for clinical diagnosis and treatment has become increasingly promising. As the primary clinical dentistry task, tooth structure segmentation for Cone-Beam Computed Tomography (CBCT) has made significant progress in recent years. However, challenges arise from the obtainment difficulty of full-annotated data, and the acquisition variability of multi-source data across different institutions, which have caused low-quality utilization, voxel-level inconsistency, and domain-specific disparity in CBCT slices. Thus, the rational and efficient utilization of multi-source and unlabeled data represents a pivotal problem. In this paper, we propose SemiTooth, a generalizable semi-supervised framework for multi-source tooth segmentation. Specifically, we first compile MS3Toothset, Multi-Source Semi-Supervised Tooth DataSet for clinical dental CBCT, which contains data from three sources with different-level annotations. Then, we design a multi-teacher and multi-student framework, i.e., SemiTooth, which promotes semi-supervised learning for multi-source data. SemiTooth employs distinct student networks that learn from unlabeled data with different sources, supervised by its respective teachers. Furthermore, a Stricter Weighted-Confidence Constraint is introduced for multiple teachers to improve the multi-source accuracy.Extensive experiments are conducted on MS3Toothset to verify the feasibility and superiority of the SemiTooth framework, which achieves SOTA performance on the semi-supervised and multi-source tooth segmentation scenario.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。