arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.06372cs.CVcs.AI
收藏

通过重度数据增强改进基于深度学习的颅骨缺损自动重建:从图像配准到潜在扩散模型

Improving Deep Learning-based Automatic Cranial Defect Reconstruction by Heavy Data Augmentation: From Image Registration to Latent Diffusion Models

AGH University of Kraków(克拉科夫AGH科技大学);University of Applied Sciences Western Switzerland (HES-SO Valais)(瑞士西部应用科学大学(瓦莱州))

Marek Wodzinski, Kamil Kwarciak, Mateusz Daniol, Daria Hemmerling

AI总结:

针对颅骨缺损重建模型跨分布泛化与标注不足问题,本文系统比较从图像配准到潜在扩散模型的重度数据增强方法,显著提升 Dice Score 并验证真实临床缺损重建效果。

摘要:

个性化颅骨植入物的建模与制造是重要的研究领域,可能缩短颅骨损伤患者的等待时间。个性化植入物的建模可通过基于深度学习的方法实现部分自动化。然而,该任务在泛化至来自此前未见分布的数据方面存在困难,导致研究成果难以用于真实临床环境。由于获取真值标注存在困难,必须考虑并引入不同技术,以提升用于训练深度网络的数据集异质性。在本研究中,我们对多种增强技术进行了大规模研究,范围从经典几何变换、图像配验、变分自编码器和生成对抗网络,直到潜在扩散模型的最新进展。我们表明,使用重度数据增强可显著提升定量和定性结果,使 SkullBreak 数据集的平均 Dice Score 高于 0.94,SkullFix 数据集高于 0.96。此外,我们证明经过合成增强的网络能够成功重建真实临床缺损。本工作对人工智能在个性化颅骨植入物自动建模领域作出了重要贡献。

英文摘要:

Modeling and manufacturing of personalized cranial implants are important research areas that may decrease the waiting time for patients suffering from cranial damage. The modeling of personalized implants may be partially automated by the use of deep learning-based methods. However, this task suffers from difficulties with generalizability into data from previously unseen distributions that make it difficult to use the research outcomes in real clinical settings. Due to difficulties with acquiring ground-truth annotations, different techniques to improve the heterogeneity of datasets used for training the deep networks have to be considered and introduced. In this work, we present a large-scale study of several augmentation techniques, varying from classical geometric transformations, image registration, variational autoencoders, and generative adversarial networks, to the most recent advances in latent diffusion models. We show that the use of heavy data augmentation significantly increases both the quantitative and qualitative outcomes, resulting in an average Dice Score above 0.94 for the SkullBreak and above 0.96 for the SkullFix datasets. Moreover, we show that the synthetically augmented network successfully reconstructs real clinical defects. The work is a considerable contribution to the field of artificial intelligence in the automatic modeling of personalized cranial implants.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑