Machine Learning Model Interpretability for Precision Medicine

Abstract

Interpretability of machine learning models is critical for data-driven precision medicine efforts. However, highly predictive models are generally complex and are difficult to interpret. Here using Model-Agnostic Explanations algorithm, we show that complex models such as random forest can be made interpretable. Using MIMIC-II dataset, we successfully predicted ICU mortality with 80% balanced accuracy and were also were able to interpret the relative effect of the features on prediction at individual level.

References in corpus (1)

• "Why Should I Trust You?": Explaining the Predictions of Any Classifier

Cited by in corpus (8)

• The role of explainability in creating trustworthy artificial intelligence for health care: a comprehensive survey of the terminology, design choices, and evaluation strategies
• Charting the Sociotechnical Gap in Explainable AI: A Framework to Address the Gap in XAI
• Understanding Heart-Failure Patients EHR Clinical Features via SHAP Interpretation of Tree-Based Machine Learning Model Predictions
• OptiLIME: Optimized LIME Explanations for Diagnostic Computer Algorithms
• Explainability Pitfalls: Beyond Dark Patterns in Explainable AI
• Health improvement framework for planning actionable treatment process using surrogate Bayesian model
• Why model why? Assessing the strengths and limitations of LIME
• Technical Report: Partial Dependence through Stratification