Brosula Raphael, Corbin Conor K, Chen Jonathan H
Genomic Center for Infectious Diseases, Broad Institute of MIT and Harvard, Cambridge, MA, USA.
Department of Computer Science, Stanford University, Stanford, CA, USA.
AMIA Jt Summits Transl Sci Proc. 2024 May 31;2024:95-104. eCollection 2024.
Access to real-world data streams like electronic medical records (EMRs) has accelerated the development of supervised machine learning (ML) models for clinical applications. However, few studies investigate the differential impact of particular features in the EMR on model performance under temporal dataset shift. To explain how features in the EMR impact models over time, this study aggregates features into by their source (e.g. medication orders, diagnosis codes and lab results) and based on their reflection of patient pathophysiology or healthcare processes. We adapt Shapley values to explain feature groups' and feature categories' marginal contribution to initial and sustained model performance. We investigate three standard clinical prediction tasks and find that while feature contributions to initial performance differ across tasks, pathophysiological features help mitigate temporal discrimination deterioration. These results provide interpretable insights on how specific feature groups contribute to model performance and robustness to temporal dataset shift.
获取电子病历(EMR)等真实世界数据流加速了用于临床应用的监督机器学习(ML)模型的开发。然而,很少有研究调查EMR中特定特征在时间数据集偏移下对模型性能的不同影响。为了解释EMR中的特征如何随时间影响模型,本研究根据特征的来源(如用药医嘱、诊断代码和实验室结果)以及它们对患者病理生理学或医疗过程的反映,将特征进行聚合。我们采用Shapley值来解释特征组和特征类别对初始和持续模型性能的边际贡献。我们研究了三项标准临床预测任务,发现虽然特征对初始性能的贡献因任务而异,但病理生理特征有助于减轻时间歧视恶化。这些结果为特定特征组如何对模型性能以及对时间数据集偏移的鲁棒性做出贡献提供了可解释的见解。