Improving Deep Learning-based Automatic Cranial Defect Reconstruction by Heavy Data Augmentation: From Image Registration to Latent Diffusion Models
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.
Code (0)
등록된 구현이 없습니다.
Tasks
Data AugmentationImage RegistrationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
High-Resolution Cranial Defect Reconstruction by Iterative, Low-Resolution, Point Cloud Completion Transformers
Each year thousands of people suffer from various types of cranial injuries and require personalized implants whose manual design is expensive and time-consuming. Therefore, an automatic, dedicated system to increase the…
GPUImage SegmentationPoint Cloud CompletionSemantic SegmentationDeep Learning-based Framework for Automatic Cranial Defect Reconstruction and Implant Modeling
The goal of this work is to propose a robust, fast, and fully automatic method for personalized cranial defect reconstruction and implant modeling. We propose a two-step deep learning-based method using a modified U-Net …
Image RegistrationMixed RealityAutomatic Cranial Defect Reconstruction with Self-Supervised Deep Deformable Masked Autoencoders
Thousands of people suffer from cranial injuries every year. They require personalized implants that need to be designed and manufactured before the reconstruction surgery. The manual design is expensive and time-consumi…
Data AugmentationImage SegmentationSemantic SegmentationCraNeXt: Automatic Reconstruction of Skull Implants With Skull Categorization Technique
Automatic cranial implant design aims to design a patient-specific implant where various machine-learning-based skull reconstruction techniques have been introduced to predict the implant. Despite the significant progres…
Deep Generative Networks for Heterogeneous Augmentation of Cranial Defects
The design of personalized cranial implants is a challenging and tremendous task that has become a hot topic in terms of process automation with the use of deep learning techniques. The main challenge is associated with …
Generative Adversarial Network