Artificial Intelligence-Based Multimodal Ultrasound Model for Prediction of Spontaneous Preterm Birth: Development
Han Bai1, Hui Shen1, Lihe Zhang1
1Department of Ultrasonic Medicine, The First Affiliated Hospital of Sun Yat-Sen University, Guangzhou, China.
Objectives:
The primary aim of this study was to develop a multimodal artificial intelligence model for predicting spontaneous preterm birth using cervical 2-dimensional ultrasound and elastography images, as well as ultrasound parameters.
Methods:
This prospective cohort study recruited pregnant women undergoing transvaginal ultrasound screening at 20-24 weeks of gestation. Participants were divided into spontaneous preterm birth group and full-term group. Three deep learning models were trained on images to compute scores via the best-performing model. Five supervised machine learning classifiers were developed and validated to predict spontaneous preterm birth based on ultrasound parameters and deep learning scores, and feature importance analysis was employed to visualize the importance of each feature in the model.
Results:
A total of 721 pregnant women (44 with spontaneous preterm birth) were included. In deep learning models based on ultrasound images, DenseNet121 (AUC: 0.92/0.93) model exhibited excellent performance, and corresponding scores were obtained. Elasticity contrast index, hardness ratio, external os stiffness, Ratio, 2-dimension-score, and elastography-score were included to construct the 5 machine learning models. Among the multimodal models, the neural network model showed the best predictive performance in the validation set (AUC = 0.85; 95% CI, 0.73-0.96). Feature importance analysis revealed that both elastography-score and 2-dimension-score made significant contributions in a majority of the models.
Conclusions:
The multimodal artificial intelligence models using ultrasound parameters and cervical 2-dimensional ultrasound and elastography images demonstrated promising predictive capabilities in internal validation for spontaneous preterm birth in the second trimester. This offers assistance for intervention in patients at high risk of spontaneous preterm birth.