arXiv:2608.18667cs.CV2026-08

用分阶段点云方法提升牙科CBCT图像中缺牙/错位牙的精准分割。

Teeth2Point: A Two-Stage Dental CBCT ROI-to-Point Segmentation Framework

论文配图:Teeth2Point: A Two-Stage Dental CBCT ROI-to-Point Segmentation Framework
图 1 · 摘自论文原文
  • 先用卷积网络定位牙齿区域,再转为点令牌输入Transformer。
  • 在4个数据集上异常情况平均提升1.44点DSC,优于最强基线。
  • 自监督预训练增强对复杂解剖结构的鲁棒性,适合临床疑难病例。

现代深度学习在牙科CBCT分割中表现优异,但缺牙或错位牙的准确标注仍是关键挑战,与临床实践高度相关。理论上,基于Transformer的模型可利用全局解剖上下文解决此类模糊性。然而,由于CBCT体积分辨率高且牙齿空间分布广泛,密集的块状体处理面临固有权衡:计算成本限制了自注意力中可用块数,导致难以同时捕捉大范围上下文和保留细粒度结构细节。本文提出Teeth2Point,一种高效的基于点的Transformer框架,可规避此权衡。该方法首先使用卷积模型定位包含牙齿的体积分区(ROIs),再通过自适应采样将ROIs转换为点令牌;随后,基于点令牌的Transformer模型在保持高分辨率的同时捕获全局上下文进行精准分割。模型先采用领域特定增强策略的自监督学习(SSL,类似DINO)进行预训练,包含随机令牌掩码,提升对复杂解剖变异的鲁棒性,再进行监督微调。相比最强两阶段基线,Teeth2Point在四个数据集上异常情况下的平均性能提升1.44 DSC点;相对于第一阶段nnU-Net,提升达1.9点。

原文摘要 · Abstract (English)

Modern deep learning architectures have demonstrated strong performance in dental CBCT segmentation. One remaining crucial challenge is accurate tooth labeling in cases with missing or malpositioned teeth, which are highly relevant for dental practice. Transformer-based architectures should in theory be able to resolve such ambiguities using global anatomical context. However, due to the high resolution of CBCT volumes and the wide spatial distribution of teeth within volumes, dense patch-based volumetric processing faces an inherent trade-off. Computational costs limit the number of patches that can be used in self-attention and thus, one can either increase the extent of the context captured in self-attention or capture fine-grained structural details by using small patches, but not both. In this work, we present Teeth2Point, an efficient point-based transformer framework for dental CBCT semantic segmentation that can avoid this trade-off. Teeth2Point first localizes volumetric regions of interest (ROIs) surrounding teeth using a convolutional model, then converts ROIs into point tokens using adaptive sampling. A transformer model predicts accurate segmentations using the point tokens, which allow capturing global context while retaining high resolution. The transformer is first pretrained using self-supervised learning (SSL), in the style of DINO but using domain-specific augmentation strategies, followed by supervised finetuning. The SSL pretraining, which includes random token masking, provides robustness to complex anatomical variations. Compared with the strongest two-stage baseline, Teeth2Point improves abnormal-case performance by 1.44 DSC points on average across four datasets; relative to the first-stage nnU-Net, the gain is 1.9 points.

牙科影像分割点云Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。