Interpretable machine learning models for hospital readmission prediction: a two-step extracted regression tree approach.

Journal: BMC medical informatics and decision making
Published Date:

Abstract

BACKGROUND: Advanced machine learning models have received wide attention in assisting medical decision making due to the greater accuracy they can achieve. However, their limited interpretability imposes barriers for practitioners to adopt them. Recent advancements in interpretable machine learning tools allow us to look inside the black box of advanced prediction methods to extract interpretable models while maintaining similar prediction accuracy, but few studies have investigated the specific hospital readmission prediction problem with this spirit.

Authors

  • Xiaoquan Gao
    School of Industrial Engineering, Purdue University, West Lafayette, USA.
  • Sabriya Alam
    Department of Electrical Engineering and Computer Sciences, UC Berkeley, Berkeley, USA.
  • Pengyi Shi
    Krannert School of Management, Purdue University, West Lafayette, USA. shi178@purdue.edu.
  • Franklin Dexter
    The University of Iowa, Iowa City.
  • Nan Kong
    Weldon School of Biomedical Engineering, Purdue University, West Lafayette, USA.