We developed an inherently interpretable multilevel Bayesian framework for representing variation in regression coefficients that mimics the piecewise linearity of ReLU-activated deep neural networks. We used the framework to formulate a survival model for using medical claims to predict hospital readmission and death that focuses on discharge placement, adjusting for confounding in estimating causal local average treatment effects. We trained the model on a 5\% sample of Medicare beneficiaries from 2008 and 2011, based on their 2009--2011 inpatient episodes, and then tested the model on 2012 episodes. The model scored an AUROC of approximately 0.76 on predicting all-cause readmissions (defined using official CMS methodology) or death within 30-days of discharge, being competitive against XGBoost and a Bayesian deep neural network, demonstrating that one need-not sacrifice interpretability for accuracy. Crucially, we provide what blackboxes cannot -- the exact gold-standard global interpretation of the model, identifying relative risk factors and quantifying the effect of discharge placement. We also show that the posthoc explainer SHAP fails to provide accurate explanations.
翻译:我们开发了一个内在的、可解释的多层次巴伊西亚框架,以代表回归系数的变化,它模仿了RELU激活的深神经网络的细线性。我们利用这个框架开发了一个生存模型,用于使用医疗索赔来预测医院重新接纳和死亡,重点是排放安置,根据对当地因果平均治疗效应的估计进行调整。我们根据2009-2011年的住院病例,对2008年和2011年5个百分点的医疗保险受益人样本进行了培训,然后对2012年的病例进行了测试。该模型在预测所有原因的重新接纳(使用正式的CMS方法定义)或排放30天内死亡方面获得了大约0.76的AUROC。我们还显示,后HAMP解释器未能提供准确的解释。