High-Resolution Cranial Implant Prediction via Patch-Wise Training

Abstract

In this study, we proposed two methods for AutoImplant (https://autoimplant.grand-challenge.org/) - the cranial implant design challenge. The shape of the implant is predicted based on the inputted defective skull. This task can be accomplished either by directly predicting the implant with the defective skull, or indirectly rebuilding the complete skull and then taking the difference between the defective and complete skulls. In our work, a deep learning model is applied to automatically predict the implant. In order to solve the problem that high resolution images can often not be directly inputted to the deep learning model, two proposed methods of resize and patch-based are examined. On the test set, the proposed resize method achieves an average dice similarity score (DSC) of 0.7350 and a Hausdorff distance (HD) of 7.2425 mm, while the proposed patch-based method achieves an average DSC of 0.8887 and a HD of 5.5339 mm.