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

https://doi.org/10.1016/j.artmed.2022.102471Get rights and content
Under a Creative Commons license
open access

Highlights

  • A novel XAI method for local, model-agnostic, post-hoc explanations is presented.

  • An open-source implementation of the described algorithm is also provided.

  • Comparative evaluation of the new method versus state-of-the-art XAI is performed.

  • Experimental results and structural properties show a good fit for XAI in medicine.

Abstract

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.

Keywords

XAI
Black-box
Explanation
Local explanation
Interpretable
Explainable
Fidelity
Reliability
Post-hoc
Model agnostic
Surrogate model

Cited by (0)

1

Equal contribution.