Automatic tooth arrangement with joint features of point and mesh representations via diffusion probabilistic models

• Published in: Computer aided geometric design 2024-06, Vol.111, p.102293, Article 102293
• Main Authors: Lei, Changsong, Xia, Mengfei, Wang, Shaofeng, Liang, Yaqian, Yi, Ran, Wen, Yu-Hui, Liu, Yong-Jin
• Format: Article
• Language: English
• Subjects: Automatic tooth arrangement; Diffusion probabilistic models; Transformation matrices prediction
• ISSN: 0167-8396; 1879-2332
• DOI: 10.1016/j.cagd.2024.102293

Tooth arrangement is a crucial step in orthodontics treatment, in which aligning teeth could improve overall well-being, enhance facial aesthetics, and boost self-confidence. To improve the efficiency of tooth arrangement and minimize errors associated with unreasonable designs by inexperienced practitioners, some deep learning-based tooth arrangement methods have been proposed. Currently, most existing approaches employ MLPs to model the nonlinear relationship between tooth features and transformation matrices to achieve tooth arrangement automatically. However, the limited datasets (which to our knowledge, have not been made public) collected from clinical practice constrain the applicability of existing methods, making them inadequate for addressing diverse malocclusion issues. To address this challenge, we propose a general tooth arrangement neural network based on the diffusion probabilistic model. Conditioned on the features extracted from the dental model, the diffusion probabilistic model can learn the distribution of teeth transformation matrices from malocclusion to normal occlusion by gradually denoising from a random variable, thus more adeptly managing real orthodontic data. To take full advantage of effective features, we exploit both mesh and point cloud representations by designing different encoding networks to extract the tooth (local) and jaw (global) features, respectively. In addition to traditional metrics ADD, PA-ADD, CSA, and MErot, we propose a new evaluation metric based on dental arch curves to judge whether the generated teeth meet the individual normal occlusion. Experimental results demonstrate that our proposed method achieves state-of-the-art tooth alignment results and satisfactory occlusal relationships between dental arches. We will publish the code and dataset. •We propose TADPM, an automatic Tooth Arrangement neural network via the Diffusion Probabilistic Model.•To extract effective dental model features, we design different encoder networks at local and global levels.•A new metric is proposed based on dental arch curves to precisely evaluate the occlusal relationship between dental arches.