Realistic 3D infant head surfaces augmentation to improve AI-based diagnosis of cranial deformities

Abstract

Evaluation of the head shape of newborns is needed to detect cranial deformities, disturbances in head growth, and consequently, to predict short- and long-term neurodevelopment. Currently, there is a lack of automatic tools to provide a detailed evaluation of the head shape. Artificial intelligence (AI) methods, namely deep learning (DL), can be explored to develop fast and automatic approaches for shape evaluation. However, due to the clinical variability of patients’ head anatomy, generalization of AI networks to the clinical needs is paramount and extremely challenging. In this work, a new framework is proposed to augment the 3D data used for training DL networks for shape evaluation. The proposed augmentation strategy deforms head surfaces towards different deformities. For that, a point-based 3D morphable model (p3DMM) is developed to generate a statistical model representative of head shapes of different cranial deformities. Afterward, a constrained transformation approach (3DHT) is applied to warp a head surface towards a target deformity by estimating a dense motion field from a sparse one resulted from the p3DMM. Qualitative evaluation showed that the proposed method generates realistic head shapes indistinguishable from the real ones. Moreover, quantitative experiments demonstrated that DL networks training with the proposed augmented surfaces improves their performance in terms of head shape analysis. Overall, the introduced augmentation allows to effectively transform a given head surface towards different deformity shapes, potentiating the development of DL approaches for head shape analysis.