Generative-Adversarial-Network-Based Data Augmentation for the Classification of Craniosynostosis

Abstract

Craniosynostosis is a congenital disease characterized by the premature closure of one or multiple sutures of the infant’s skull. For diagnosis, 3D photogrammetric scans are a radiation-free alternative to computed tomography. However, data is only sparsely available and the role of data augmentation for the classification of craniosynostosis has not yet been analyzed. In this work, we use a 2D distance map representation of the infants’ heads with a convolutional-neural-network-based classifier and employ a generative adversarial network (GAN) for data augmentation. We simulate two data scarcity scenarios with 15% and 10% training data and test the influence of different degrees of added synthetic data and balancing underrepresented classes. We used total accuracy and F1-score as a metric to evaluate the final classifiers. For 15% training data, the GAN-augmented dataset showed an increased F1-score up to 0.1 and classification accuracy up to 3 %. For 10% training data, both metrics decreased. We present a deep convolutional GAN capable of creating synthetic data for the classification of craniosynostosis. Using a moderate amount of synthetic data using a GAN showed slightly better performance, but had little effect overall. The simulated scarcity scenario of 10% training data may have limited the model’s ability to learn the underlying data distribution.

Keywords

• generative adversarial network
• classification
• craniosynostosis
• data augmentation