用深度强化学习自动规划牙齿排列路径,更安全高效。
3D Geometric Tooth Alignment Planning via Deep Reinforcement Learning

- 将排牙规划建模为马尔可夫决策过程,用Transformer处理复杂牙齿交互。
- 在1万份临床数据上验证,路径安全性与几何效率优于现有方法。
- 引入动态遮蔽和分阶段训练,贴近临床逻辑,适合牙科数字化场景。
3D几何牙齿排列规划是现代数字正畸的核心,旨在从初始错颌状态生成到目标对齐的序列路径。本文提出一种新的深度强化学习(DRL)框架,自动生成此类排列路径。将规划过程建模为马尔可夫决策过程(MDP),以捕捉其序列决策特性,重点优化几何轨迹并集成关键空间约束,如牙间碰撞避免与路径效率。所提方法采用深度确定性策略梯度(DDPG)算法,结合三项创新:(1)基于Transformer的智能体,用于建模牙齿间的复杂空间交互,并管理高维状态-动作空间;(2)动态遮蔽机制,每步仅允许少数牙齿移动,更贴合临床逐次排牙逻辑;(3)两阶段课程学习策略,逐步提升任务难度,确保训练稳定与高效路径发现。在包含10,000个专家设计治疗方案的临床数据集上评估,实验结果表明,该方法在路径安全性与几何效率方面均优于现有基线,提供了一种稳健且自动化的3D几何正畸排列规划解决方案。
原文摘要 · Abstract (English)
3D geometric tooth alignment planning, which determines sequential trajectories from initial malocclusion to the final target alignment, is a cornerstone of modern digital orthodontics. This paper presents a novel deep reinforcement learning (DRL) framework to automate the generation of these alignment paths. We formulate the planning process as a Markov Decision Process (MDP) to capture its sequential decision-making nature, focusing on optimizing geometric trajectories while integrating essential spatial constraints, such as inter-dental collision avoidance and path efficiency. The proposed method leverages the Deep Deterministic Policy Gradient (DDPG) algorithm, enhanced by three key innovations: (1) a Transformer-based agent to model complex spatial interactions between teeth and manage high-dimensional state-action spaces; (2) a dynamic masking scheme that restricts movement to a sparse subset of teeth per step, better reflecting the clinical logic of sequential alignment; and (3) a two-stage curriculum learning strategy that gradually increases task difficulty to ensure training stability and efficient path discovery. We evaluate our approach on a dataset of 10K expert-designed treatment plans based on clinical data. Experimental results demonstrate that our method outperforms existing baselines in terms of path safety and geometric efficiency, providing a robust and automated solution for 3D geometric orthodontic alignment planning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。