# Deep Learning-based Framework for Automatic Cranial Defect Reconstruction and Implant Modeling

arXiv: https://arxiv.org/abs/2204.06310 · HTML: https://mlsanity.com/paper/2204.06310/

## Key facts

- Cited by 8 papers in the map (data through 14 Oct 2026).
- Builds on 9 of its 47 references that are in the map.
- First posted on arXiv 13 Apr 2022.

## Builds on (most cited first, 9 of 9)

- [PyTorch: An Imperative Style, High-Performance Deep Learning Library](https://mlsanity.com/paper/1912.01703/) (2019)
- [U-Net: Convolutional Networks for Biomedical Image Segmentation](https://mlsanity.com/paper/1505.04597/) (2015)
- [4D Spatio-Temporal ConvNets: Minkowski Convolutional Neural Networks](https://mlsanity.com/paper/1904.08755/) (2019)
- [VoxelMorph: A Learning Framework for Deformable Medical Image Registration](https://mlsanity.com/paper/1809.05231/) (2018)
- [A Framework For Contrastive Self-Supervised Learning And Designing A New Approach](https://mlsanity.com/paper/2009.00104/) (2020)
- [A Baseline Approach for AutoImplant: the MICCAI 2020 Cranial Implant Design Challenge](https://mlsanity.com/paper/2006.12449/) (2020)
- [Cranial Implant Design via Virtual Craniectomy with Shape Priors](https://mlsanity.com/paper/2009.13704/) (2020)
- [Learning to Rearrange Voxels in Binary Segmentation Masks for Smooth Manifold Triangulation](https://mlsanity.com/paper/2108.05269/) (2021)
- [Segmentation of Defective Skulls from CT Data for Tissue Modelling](https://mlsanity.com/paper/1911.08805/) (2019)

## Built on by (most cited first, 8 of 8)

- [MedShapeNet -- A Large-Scale Dataset of 3D Medical Shapes for Computer Vision](https://mlsanity.com/paper/2308.16139/) (2023)
- [Deep Generative Networks for Heterogeneous Augmentation of Cranial Defects](https://mlsanity.com/paper/2308.04883/) (2023)
- [Anatomy Completor: A Multi-class Completion Framework for 3D Anatomy Reconstruction](https://mlsanity.com/paper/2309.04956/) (2023)
- [Automatic Cranial Defect Reconstruction with Self-Supervised Deep Deformable Masked Autoencoders](https://mlsanity.com/paper/2404.13106/) (2024)
- [Improving Deep Learning-based Automatic Cranial Defect Reconstruction by Heavy Data Augmentation: From Image Registration to Latent Diffusion Models](https://mlsanity.com/paper/2406.06372/) (2024)
- [RibCageImp: A Deep Learning Framework for 3D Ribcage Implant Generation](https://mlsanity.com/paper/2411.09204/) (2024)
- [Automatic Skull Reconstruction by Deep Learnable Symmetry Enforcement](https://mlsanity.com/paper/2411.17342/) (2024)
- [MedPCFM-TED: One-Step Point Cloud Flow Matching for Implant Generation via Teacher-Guided Endpoint Distillation](https://mlsanity.com/paper/2609.16934/) (2026)

## Abstract (arXiv)

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 architecture to perform the defect reconstruction, and a dedicated iterative procedure to improve the implant geometry, followed by automatic generation of models ready for 3-D printing. We propose a cross-case augmentation based on imperfect image registration combining cases from different datasets. We perform ablation studies regarding different augmentation strategies and compare them to other state-of-the-art methods. We evaluate the method on three datasets introduced during the AutoImplant 2021 challenge, organized jointly with the MICCAI conference. We perform the quantitative evaluation using the Dice and boundary Dice coefficients, and the Hausdorff distance. The average Dice coefficient, boundary Dice coefficient, and the 95th percentile of Hausdorff distance are 0.91, 0.94, and 1.53 mm respectively. We perform an additional qualitative evaluation by 3-D printing and visualization in mixed reality to confirm the implant's usefulness. We propose a complete pipeline that enables one to create the cranial implant model ready for 3-D printing. The described method is a greatly extended version of the method that scored 1st place in all AutoImplant 2021 challenge tasks. We freely release the source code, that together with the open datasets, makes the results fully reproducible. The automatic reconstruction of cranial defects may enable manufacturing personalized implants in a significantly shorter time, possibly allowing one to perform the 3-D printing process directly during a given intervention. Moreover, we show the usability of the defect reconstruction in mixed reality that may further reduce the surgery time.