Application of machine learning in screening for congenital heart diseases using fetal echocardiography

There is a growing body of literature supporting the utilization of machine learning (ML) to improve diagnosis and prognosis tools of cardiovascular disease. The current study was to investigate the impact that the ML framework may have on the sensitivity of predicting the presence or absence of congenital heart disease (CHD) using fetal echocardiography. A comprehensive fetal echocardiogram including 2D cardiac chamber quantification, valvar assessments, assessment of great vessel morphology, and Doppler-derived blood flow interrogation was recorded. The postnatal echocardiogram was used to ascertain the diagnosis of CHD. A random forest (RF) algorithm with a nested tenfold cross-validation was used to train models for assessing the presence of CHD. The study population was derived from a database of 3910 singleton fetuses with maternal age of 28.8 ± 5.2 years and gestational age at the time of fetal echocardiography of 22.0 weeks (IQR 21–24). The proportion of CHD was 14.1% for the studied cohort confirmed by post-natal echocardiograms. Our proposed RF-based framework provided a sensitivity of 0.85, a specificity of 0.88, a positive predictive value of 0.55 and a negative predictive value of 0.97 to detect the CHD with the mean of mean ROC curves of 0.94 and the mean of mean PR curves of 0.84. Additionally, six first features, including cardiac axis, peak velocity of blood flow across the pulmonic valve, cardiothoracic ratio, pulmonary valvar annulus diameter, right ventricular end-diastolic diameter, and aortic valvar annulus diameter, are essential features that play crucial roles in adding more predictive values to the model in detecting patients with CHD. ML using RF can provide increased sensitivity in prenatal CHD screening with very good performance. The incorporation of ML algorithms into fetal echocardiography may further standardize the assessment for CHD.