Interpretability in healthcare: A comparative study of local machine learning interpretability techniques

被引:97
作者
ElShawi, Radwa [1 ]
Sherif, Youssef [1 ]
Al-Mallah, Mouaz [2 ]
Sakr, Sherif [1 ]
机构
[1] Tartu Univ, Tartu, Estonia
[2] Houston Methodist Ctr, Houston, TX USA
关键词
big data; data science; interpretability; machine learning;
D O I
10.1111/coin.12410
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Although complex machine learning models (eg, random forest, neural networks) are commonly outperforming the traditional and simple interpretable models (eg, linear regression, decision tree), in the healthcare domain, clinicians find it hard to understand and trust these complex models due to the lack of intuition and explanation of their predictions. With the new general data protection regulation (GDPR), the importance for plausibility and verifiability of the predictions made by machine learning models has become essential. Hence, interpretability techniques for machine learning models are an area focus of research. In general, the main aim of these interpretability techniques is to shed light and provide insights into the prediction process of the machine learning models and to be able to explain how the results from the prediction was generated. A major problem in this context is that both the quality of the interpretability techniques and trust of the machine learning model predictions are challenging to measure. In this article, we propose four fundamental quantitative measures for assessing the quality of interpretability techniques-similarity, bias detection, execution time, and trust. We present a comprehensive experimental evaluation of six recent and popular local model agnostic interpretability techniques, namely, LIME, SHAP, Anchors, LORE, ILIME" and MAPLE on different types of real-world healthcare data. Building on previous work, our experimental evaluation covers different aspects for its comparison including identity, stability, separability, similarity, execution time, bias detection, and trust. The results of our experiments show that MAPLE achieves the highest performance for the identity across all data sets included in this study, while LIME achieves the lowest performance for the identity metric. LIME achieves the highest performance for the separability metric across all data sets. On average, SHAP has the smallest average time to output explanation across all data sets included in this study. For detecting the bias, SHAP and MAPLE enable the participants to better detect the bias. For the trust metric, Anchors achieves the highest performance on all data sets included in this work.
引用
收藏
页码:1633 / 1650
页数:18
相关论文
共 42 条
[1]   Rationale and Design of the Henry Ford ExercIse Testing Project (The FIT Project) [J].
Al-Mallah, Mouaz H. ;
Keteyian, Steven J. ;
Brawner, Clinton A. ;
Whelton, Seamus ;
Blaha, Michael J. .
CLINICAL CARDIOLOGY, 2014, 37 (08) :456-461
[2]   Predicting diabetes mellitus using SMOTE and ensemble machine learning approach: The Henry Ford ExercIse Testing (FIT) project [J].
Alghamdi, Manal ;
Al-Mallah, Mouaz ;
Keteylan, Steven ;
Brawner, Clinton ;
Ehrman, Jonathan ;
Sakr, Sherif .
PLOS ONE, 2017, 12 (07)
[3]  
[Anonymous], 2018, arXiv preprint arXiv:1808.05054
[4]  
Bloniarz A, 2016, JMLR WORKSH CONF PRO, V51, P1450
[5]   Intelligible Models for HealthCare: Predicting Pneumonia Risk and Hospital 30-day Readmission [J].
Caruana, Rich ;
Lou, Yin ;
Gehrke, Johannes ;
Koch, Paul ;
Sturm, Marc ;
Elhadad, Noemie .
KDD'15: PROCEEDINGS OF THE 21ST ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2015, :1721-1730
[6]   Machine Learning Interpretability: A Survey on Methods and Metrics [J].
Carvalho, Diogo, V ;
Pereira, Eduardo M. ;
Cardoso, Jaime S. .
ELECTRONICS, 2019, 8 (08)
[7]   DETECTION OF INFLUENTIAL OBSERVATION IN LINEAR-REGRESSION [J].
COOK, RD .
TECHNOMETRICS, 1977, 19 (01) :15-18
[8]  
Cover Thomas M, 1999, Elements of Information Theory
[9]   Machine Learning and the Profession of Medicine [J].
Darcy, Alison M. ;
Louie, Alan K. ;
Roberts, Laura Weiss .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2016, 315 (06) :551-552
[10]  
Domhan T, 2015, PROCEEDINGS OF THE TWENTY-FOURTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE (IJCAI), P3460