Why did AI get this one wrong? - Tree-based explanations of machine learning model predictions

被引:24
|
作者
Parimbelli, Enea [1 ,3 ,5 ]
Buonocore, Tommaso Mario [1 ]
Nicora, Giovanna [1 ,2 ]
Michalowski, Wojtek [3 ]
Wilk, Szymon [4 ]
Bellazzi, Riccardo [1 ]
机构
[1] Univ Pavia, Dept Elect Comp & Biomed Engn, Pavia, Italy
[2] enGenome srl, Pavia, Italy
[3] Univ Ottawa, Telfer Sch Management, Ottawa, ON, Canada
[4] Poznan Univ Tech, Inst Comp Sci, Div Intelligent Decis Support Syst, Poznan, Poland
[5] Dept Elect Comp & Biomed Engn, Via Ferrata 5, I-27100 Pavia, Italy
基金
欧盟地平线“2020”;
关键词
XAI; Black-box; Explanation; Local explanation; Interpretable; Explainable; Fidelity; Reliability; Post-hoc; Model agnostic; Surrogate model; DATASET SHIFT;
D O I
10.1016/j.artmed.2022.102471
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Increasingly complex learning methods such as boosting, bagging and deep learning have made ML models more accurate, but harder to interpret and explain, culminating in black-box machine learning models. Model developers and users alike are often presented with a trade-off between performance and intelligibility, especially in high-stakes applications like medicine. In the present article we propose a novel methodological approach for generating explanations for the predictions of a generic machine learning model, given a specific instance for which the prediction has been made. The method, named AraucanaXAI, is based on surrogate, locally-fitted classification and regression trees that are used to provide post-hoc explanations of the prediction of a generic machine learning model. Advantages of the proposed XAI approach include superior fidelity to the original model, ability to deal with non-linear decision boundaries, and native support to both classification and regression problems. We provide a packaged, open-source implementation of the AraucanaXAI method and evaluate its behaviour in a number of different settings that are commonly encountered in medical applications of AI. These include potential disagreement between the model prediction and physician's expert opinion and low reliability of the prediction due to data scarcity.
引用
收藏
页数:9
相关论文
共 50 条
  • [21] A tree-based machine learning model to approach morphologic assessment of malignant salivary gland tumors
    Lopez-Janeiro, Alvaro
    Cabanuz, Clara
    Blasco-Santana, Luis
    Ruiz-Bravo, Elena
    ANNALS OF DIAGNOSTIC PATHOLOGY, 2022, 56
  • [22] An ensemble tree-based machine learning model for predicting the uniaxial compressive strength of travertine rocks
    Barzegar, Rahim
    Sattarpour, Masoud
    Deo, Ravinesh
    Fijani, Elham
    Adamowski, Jan
    NEURAL COMPUTING & APPLICATIONS, 2020, 32 (13): : 9065 - 9080
  • [23] An interpretable framework for modeling global solar radiation using tree-based ensemble machine learning and Shapley additive explanations methods
    Song, Zhe
    Cao, Sunliang
    Yang, Hongxing
    APPLIED ENERGY, 2024, 364
  • [24] Tree-based ensemble machine learning model for nitrate reduction by zero-valent iron
    Istiqomah, Nurul Alvia
    Jung, Donghwi
    Khim, Jeehyeong
    JOURNAL OF WATER PROCESS ENGINEERING, 2023, 56
  • [25] Tree-Based and Machine Learning Algorithm Analysis for Breast Cancer Classification
    Bhardwaj, Arpit
    Bhardwaj, Harshit
    Sakalle, Aditi
    Uddin, Ziya
    Sakalle, Maneesha
    Ibrahim, Wubshet
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
  • [26] Uncovering Sociological Effect Heterogeneity Using Tree-Based Machine Learning
    Brand, Jennie E.
    Xu, Jiahui
    Koch, Bernard
    Geraldo, Pablo
    SOCIOLOGICAL METHODOLOGY, VOL 51, ISSUE 2, 2021, 51 (02): : 189 - 223
  • [27] The predictability of tree-based machine learning algorithms in the big data context
    Qolipour F.
    Ghasemzadeh M.
    Mohammad-Karimi N.
    International Journal of Engineering, Transactions A: Basics, 2021, 34 (01): : 82 - 89
  • [28] A general tree-based machine learning accelerator with memristive analog CAM
    Pedretti, Giacomo
    Serebryakov, Sergey
    Strachan, John Paul
    Graves, Catherine E.
    2022 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS 22), 2022, : 220 - 224
  • [29] Determining the Happiness Class of Countries with Tree-Based Algorithms in Machine Learning
    Dogruel, Merve
    Kara, Selin Soner
    ACTA INFOLOGICA, 2023, 7 (02): : 243 - 252
  • [30] Predicting the pathological invasiveness in patients with a solitary pulmonary nodule via Shapley additive explanations interpretation of a tree-based machine learning radiomics model: a multicenter study
    Zhang, Rong
    Hong, Minping
    Cai, Hongjie
    Liang, Yanting
    Chen, Xinjie
    Liu, Ziwei
    Wu, Meilian
    Zhou, Cuiru
    Bao, Chenzhengren
    Wang, Huafeng
    Yang, Shaomin
    Hu, Qiugen
    QUANTITATIVE IMAGING IN MEDICINE AND SURGERY, 2023, 13 (12) : 7828 - +