Login / Signup

The Framing of machine learning risk prediction models illustrated by evaluation of sepsis in general wards.

Simon Meyer LauritsenBo ThiessonMarianne Johansson JørgensenAnders Hammerich RiisUlrick Skipper EspelundJesper Bo WeileJeppe Lange
Published in: NPJ digital medicine (2021)
Problem framing is critical to developing risk prediction models because all subsequent development work and evaluation takes place within the context of how a problem has been framed and explicit documentation of framing choices makes it easier to compare evaluation metrics between published studies. In this work, we introduce the basic concepts of framing, including prediction windows, observation windows, window shifts and event-triggers for a prediction that strongly affects the risk of clinician fatigue caused by false positives. Building on this, we apply four different framing structures to the same generic dataset, using a sepsis risk prediction model as an example, and evaluate how framing affects model performance and learning. Our results show that an apparently good model with strong evaluation results in both discrimination and calibration is not necessarily clinically usable. Therefore, it is important to assess the results of objective evaluations within the context of more subjective evaluations of how a model is framed.
Keyphrases
  • machine learning
  • intensive care unit
  • acute kidney injury
  • high resolution
  • randomized controlled trial
  • sleep quality
  • depressive symptoms
  • electronic health record
  • big data
  • deep learning
  • physical activity