用原型匹配解析牙齿3D点云,让咬合分析可解释且定位准确。
ProtoPointNet: Prototype-Based Interpretable Classification of 3D Dental Point Clouds with Verifiable Spatial Activations

- 以14维特征编码点云,融合几何、曲率与上下牙弓位移信息
- 在Bits2Bites数据集上达0.724宏F1和0.825 AUROC,垂直向表现最优
- 原型激活区域对应真实解剖位置,适合临床医生理解模型决策
基于原型的网络通过将预测关联到学习到的典型样本实现内在可解释性,但在3D点云和临床表面配对推理中的应用仍有限。我们提出ProtoPointNet,一种用于从注册的上下颌口内牙弓对中进行咬合分类的原型模型。每个点由14维描述符编码,包含局部表面几何、曲率以及显式的上下牙弓位移和间隙信息,从而暴露咬合关系以供原型匹配。采用共享多任务点云主干网络,分别学习矢状左、矢状右、垂直、横向和中线方向的原型头。为应对临床数据有限的问题,通过辅助监督和编码器冻结过渡从头训练原型。在Bits2Bites数据集上,ProtoPointNet达到平均测试宏F1为0.724,AUROC为0.825,其中垂直方向(F1 0.828)和矢状左方向(F1 0.807)表现最佳。投影的原型激活区域定位至解剖学合理的区域,如后磨牙和前磨牙显示反咬证据,前切牙显示咬合深度证据。结果表明,原型推理是一种透明且空间接地的替代方案,适用于牙齿表面配对分析的黑盒3D分类器。
原文摘要 · Abstract (English)
Prototype-based networks provide inherently interpretable classification by linking predictions to learned exemplars, but their use in 3D point clouds and clinical surface-pair reasoning remains limited. We introduce ProtoPointNet, a prototype-based model for dental occlusion classification from registered upper--lower intraoral arch pairs. Each point is encoded by a 14-dimensional descriptor combining local surface geometry, curvature, and explicit inter-arch displacement and clearance, exposing occlusal relationships to prototype matching. A shared multi-task point-cloud backbone learns axis-specific prototype heads for sagittal-left, sagittal-right, vertical, transverse, and midline classification. To support limited clinical data, we train prototypes from scratch using auxiliary supervision and encoder-freeze hand-off. On Bits2Bites, ProtoPointNet achieves mean test macro-F1 of 0.724 and AUROC of 0.825, with strongest performance on vertical (F1 0.828) and sagittal-left classification (F1 0.807). Projected prototype activations localise to anatomically plausible regions, including posterior molars and premolars for cross-bite evidence and anterior incisors for bite-depth evidence. These results support prototype-based reasoning as a transparent, spatially grounded alternative to black-box 3D classifiers for dental surface-pair analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。