Interpretable Predictive Models to Understand Risk Factors for Maternal and Fetal Outcomes.

Journal: Journal of healthcare informatics research
Published Date:

Abstract

Although most pregnancies result in a good outcome, complications are not uncommon and can be associated with serious implications for mothers and babies. Predictive modeling has the potential to improve outcomes through a better understanding of risk factors, heightened surveillance for high-risk patients, and more timely and appropriate interventions, thereby helping obstetricians deliver better care. We identify and study the most important risk factors for four types of pregnancy complications: (i) severe maternal morbidity, (ii) shoulder dystocia, (iii) preterm preeclampsia, and (iv) antepartum stillbirth. We use an Explainable Boosting Machine (EBM), a high-accuracy glass-box learning method, for the prediction and identification of important risk factors. We undertake external validation and perform an extensive robustness analysis of the EBM models. EBMs match the accuracy of other black-box ML methods, such as deep neural networks and random forests, and outperform logistic regression, while being more interpretable. EBMs prove to be robust. The interpretability of the EBM models reveal surprising insights into the features contributing to risk (e.g., maternal height is the second most important feature for shoulder dystocia) and may have potential for clinical application in the prediction and prevention of serious complications in pregnancy.

Authors

  • Tomas M Bosschieter
    Stanford University, Stanford, CA USA.
  • Zifei Xu
    Stanford University, Stanford, CA USA.
  • Hui Lan
    Stanford University, Stanford, CA USA.
  • Benjamin J Lengerich
    Massachusetts Institute of Technology, Cambridge, MA USA.
  • Harsha Nori
    Microsoft Research, Redmond, Washington, USA.
  • Ian Painter
    Foundation for Healthcare Quality, Seattle, WA USA.
  • Vivienne Souter
    Foundation for Healthcare Quality, Seattle, WA USA.
  • Rich Caruana
    Microsoft Research, Redmond, WA USA.

Keywords

No keywords available for this article.