Deep Learning Using Augmentation via Registration: 1st Place Solution to the AutoImplant 2020 Challenge

David G. Ellis, Michele R. Aizenberg

Research output: Chapter in Book/Report/Conference proceedingConference contribution

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.

Original languageEnglish (US)
Title of host publicationTowards the Automatization of Cranial Implant Design in Cranioplasty - First Challenge, AutoImplant 2020, Held in Conjunction with MICCAI 2020, Proceedings
EditorsJianning Li, Jan Egger
PublisherSpringer Science and Business Media Deutschland GmbH
Pages47-55
Number of pages9
ISBN (Print)9783030643263
DOIs
StatePublished - 2020
Event1st Automatization of Cranial Implant Design in Cranioplasty Challenge, AutoImplant 2020 - Lima, Peru
Duration: Oct 8 2020Oct 8 2020

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume12439 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference1st Automatization of Cranial Implant Design in Cranioplasty Challenge, AutoImplant 2020
Country/TerritoryPeru
CityLima
Period10/8/2010/8/20

Keywords

  • Augmentation
  • Deep learning
  • Shape completion

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)

Fingerprint

Dive into the research topics of 'Deep Learning Using Augmentation via Registration: 1st Place Solution to the AutoImplant 2020 Challenge'. Together they form a unique fingerprint.

Cite this