用可学习对称性提升颅骨重建精度,加速个性化植入物设计。
Automatic Skull Reconstruction by Deep Learnable Symmetry Enforcement
- 引入可学习的对称性约束,改进深度网络对颅骨结构的重建能力。
- 在多个指标上优于基线(DSC 0.94 vs 0.84,HD95 1.31 vs 2.43)。
- 计算量极低(<500 GPU小时),适合临床快速部署。
每年有成千上万的人因颅骨损伤需定制植入物填充颅腔,但重建手术等待时间常长达数周甚至数月,尤其在医疗资源匮乏地区。当前个性化植入物建模依赖经验丰富的生物力学专家,成本高且耗时。近年来人工智能尤其是深度学习为自动化该过程带来希望,但面临三大挑战:训练数据集规模小、体数据分辨率高、数据异质性强。本文提出一种基于可学习对称性强化的新方法,通过训练神经网络自动计算颅骨对称性,既可在训练中作为目标函数,也可在重建后用于优化。在公开的SkullBreak和SkullFix数据集上定量评估,结果表明对称性保留的重建网络显著优于基线(DSC 0.94/0.94/1.31 vs 0.84/0.76/2.43,分别对应DSC、bDSC、HD95)。定性分析显示其在真实临床病例中表现优异。该方法性能接近顶尖模型,但仅需<500 GPU小时,远低于其他方法(>100,000 GPU小时),是迈向临床自动颅骨缺陷重建的重要一步。
原文摘要 · Abstract (English)
Every year, thousands of people suffer from skull damage and require personalized implants to fill the cranial cavity. Unfortunately, the waiting time for reconstruction surgery can extend to several weeks or even months, especially in less developed countries. One factor contributing to the extended waiting period is the intricate process of personalized implant modeling. Currently, the preparation of these implants by experienced biomechanical experts is both costly and time-consuming. Recent advances in artificial intelligence, especially in deep learning, offer promising potential for automating the process. However, deep learning-based cranial reconstruction faces several challenges: (i) the limited size of training datasets, (ii) the high resolution of the volumetric data, and (iii) significant data heterogeneity. In this work, we propose a novel approach to address these challenges by enhancing the reconstruction through learnable symmetry enforcement. We demonstrate that it is possible to train a neural network dedicated to calculating skull symmetry, which can be utilized either as an additional objective function during training or as a post-reconstruction objective during the refinement step. We quantitatively evaluate the proposed method using open SkullBreak and SkullFix datasets, and qualitatively using real clinical cases. The results indicate that the symmetry-preserving reconstruction network achieves considerably better outcomes compared to the baseline (0.94/0.94/1.31 vs 0.84/0.76/2.43 in terms of DSC, bDSC, and HD95). Moreover, the results are comparable to the best-performing methods while requiring significantly fewer computational resources (< 500 vs > 100,000 GPU hours). The proposed method is a considerable contribution to the field of applied artificial intelligence in medicine and is a step toward automatic cranial defect reconstruction in clinical practice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。