用文本引导的自监督框架,自动设计牙种植体基台,省时且更准。
SSA3D: Text-Conditioned Assisted Self-Supervised Framework for Automatic Dental Abutment Design
- 双分支架构:重建分支学结构,回归分支直接预测参数
- 训练时间减半,准确率高于传统自监督方法
- 支持临床文本提示,适合牙科数字化设计场景
基台设计是牙种植修复的关键步骤。然而,人工设计需繁琐测量与试戴,且因缺乏大规模标注数据,基于AI的自动化研究受限。自监督学习(SSL)虽可缓解数据稀缺问题,但需预训练和微调,计算成本高、训练耗时长。本文提出一种自监督辅助的自动基台设计框架SSA3D,采用双分支结构:重建分支学习恢复掩码的口内扫描数据,并将学到的结构信息传递给回归分支;回归分支在监督学习下直接预测基台参数,避免了独立的预训练与微调流程。同时设计文本条件提示(TCP)模块,融入种植体位置、系统和系列等临床信息,引导网络关注相关区域并约束参数预测。在自建数据集上的大量实验表明,SSA3D训练时间减少一半,精度高于传统SSL方法,且达到当前最优性能,显著提升自动基台设计的准确率与效率。
原文摘要 · Abstract (English)
Abutment design is a critical step in dental implant restoration. However, manual design involves tedious measurement and fitting, and research on automating this process with AI is limited, due to the unavailability of large annotated datasets. Although self-supervised learning (SSL) can alleviate data scarcity, its need for pre-training and fine-tuning results in high computational costs and long training times. In this paper, we propose a Self-supervised assisted automatic abutment design framework (SS$A^3$D), which employs a dual-branch architecture with a reconstruction branch and a regression branch. The reconstruction branch learns to restore masked intraoral scan data and transfers the learned structural information to the regression branch. The regression branch then predicts the abutment parameters under supervised learning, which eliminates the separate pre-training and fine-tuning process. We also design a Text-Conditioned Prompt (TCP) module to incorporate clinical information (such as implant location, system, and series) into SS$A^3$D. This guides the network to focus on relevant regions and constrains the parameter predictions. Extensive experiments on a collected dataset show that SS$A^3$D saves half of the training time and achieves higher accuracy than traditional SSL methods. It also achieves state-of-the-art performance compared to other methods, significantly improving the accuracy and efficiency of automated abutment design.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。