用深度学习自动划分牙冠边缘线,提升修复精度与效率
Mesh based segmentation for automated margin line generation on incisors receiving crown treatment
- 基于网格的神经网络分割牙体,定位边缘线边界面
- 集成模型在13组测试中7次达标(误差<200μm)
- 适合牙科数字化设计人员及研究者使用
牙冠修复是修复患者受损或缺失牙齿的重要手段。当前多依赖商业软件,需人工划定牙体预备表面的边缘线,过程重复性差且易出错。本文提出一种基于深度学习的自动边缘线生成框架。利用合作牙科实验室提供的切牙数据集训练分割模型,改进网格神经网络输入通道,将其用于将预备牙体分为两区域,边缘线位于两区域边界面上。采用5折交叉验证训练5个模型,并通过投票分类器融合结果以提升分割性能。随后使用图割法进行边界平滑与优化,选取两区域间的边界面表示边缘线,再以样条曲线拟合边界面中心点以预测最终边缘线。实验显示,集成模型结合最大概率预测在13组测试中成功7次(以200μm为人类误差阈值)。此外,预备质量越高,预测与真实边缘线偏差越小(斯皮尔曼等级相关系数-0.683)。研究已公开训练与测试数据集。
原文摘要 · Abstract (English)
Dental crowns are essential dental treatments for restoring damaged or missing teeth of patients. Recent design approaches of dental crowns are carried out using commercial dental design software. Once a scan of a preparation is uploaded to the software, a dental technician needs to manually define a precise margin line on the preparation surface, which constitutes a non-repeatable and inconsistent procedure. This work proposes a new framework to determine margin lines automatically and accurately using deep learning. A dataset of incisor teeth was provided by a collaborating dental laboratory to train a deep learning segmentation model. A mesh-based neural network was modified by changing its input channels and used to segment the prepared tooth into two regions such that the margin line is contained within the boundary faces separating the two regions. Next, k-fold cross-validation was used to train 5 models, and a voting classifier technique was used to combine their results to enhance the segmentation. After that, boundary smoothing and optimization using the graph cut method were applied to refine the segmentation results. Then, boundary faces separating the two regions were selected to represent the margin line faces. A spline was approximated to best fit the centers of the boundary faces to predict the margin line. Our results show that an ensemble model combined with maximum probability predicted the highest number of successful test cases (7 out of 13) based on a maximum distance threshold of 200 m (representing human error) between the predicted and ground truth point clouds. It was also demonstrated that the better the quality of the preparation, the smaller the divergence between the predicted and ground truth margin lines (Spearman's rank correlation coefficient of -0.683). We provide the train and test datasets for the community.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。