@inproceedings{653adb49e39248218fbcad08695ffbe0,
title = "Deep Learning Using Augmentation via Registration: 1st Place Solution to the AutoImplant 2020 Challenge",
abstract = "Automatic cranial implant design can save clinicians time and resources by computing the implant shape and size from a single image of a defective skull. We aimed to improve upon previously proposed deep learning methods by augmenting the training data set using transformations that warped the images into different shapes and orientations. The transformations were computed by non-linearly registering the complete skull images between the 100 subjects in the training data set. The transformations were then applied to warp each of the defective and complete skull images so that the shape and orientation resembled that of a different subject in the training set. One hundred ninety-seven of the registrations failed, resulting in an augmented training set of 9,803 defective and complete skull image pairs. The augmented training set was used to train an ensemble of four U-Net models to predict the complete skull shape from the defective skulls using cross-validation. The ensemble of models performed very well and predicted the implant shapes with a mean dice similarity coefficient of 0.942 and a mean Hausdorff distance of 3.598 mm for all 110 test cases. Our solution ranked first among all participants of the AutoImplant 2020 challenge. The code for this project is available at https://github.com/ellisdg/3DUnetCNN.",
keywords = "Augmentation, Deep learning, Shape completion",
author = "Ellis, {David G.} and Aizenberg, {Michele R.}",
note = "Funding Information: Acknowledgments. This work was completed utilizing the Holland Computing Center of the University of Nebraska, which receives support from the Nebraska Research Initiative. Publisher Copyright: {\textcopyright} 2020, Springer Nature Switzerland AG.; 1st Automatization of Cranial Implant Design in Cranioplasty Challenge, AutoImplant 2020 ; Conference date: 08-10-2020 Through 08-10-2020",
year = "2020",
doi = "10.1007/978-3-030-64327-0_6",
language = "English (US)",
isbn = "9783030643263",
series = "Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)",
publisher = "Springer Science and Business Media Deutschland GmbH",
pages = "47--55",
editor = "Jianning Li and Jan Egger",
booktitle = "Towards the Automatization of Cranial Implant Design in Cranioplasty - First Challenge, AutoImplant 2020, Held in Conjunction with MICCAI 2020, Proceedings",
address = "Germany",
}