Explainability of Machine Learning Models under Missing Data
Authors: Tuan L. Vo, Thu Nguyen, Hugo L. Hammer, Michael A. Riegler, Pal Halvorsen
AI-generated Key Points
- Study by Tuan L. Vo et al. focuses on impact of imputation methods on calculating Shapley values in Explainable Artificial Intelligence
- Comparison of different strategies and evaluation of effects on feature importance and interaction determined by Shapley values
- Potential biases introduced by imputation methods affecting interpretability of machine learning models
- Lower test prediction mean square error (MSE) does not necessarily imply lower MSE in Shapley values and vice versa
- Xgboost can handle missing data directly, but using it on incomplete data can significantly impact interpretability compared to imputing before training
- Emphasis on considering imputation effects when interpreting models for robust insights and offering practical guidance for selecting appropriate techniques based on dataset characteristics and analysis objectives
- Different approaches leading to varying interpretations, highlighting the need to consider both accuracy and interpretability preservation when dealing with missing data in machine learning models within Explainable Artificial Intelligence frameworks
Abstract: Missing data is a prevalent issue that can significantly impair model performance and interpretability. This paper briefly summarizes the development of the field of missing data with respect to Explainable Artificial Intelligence and experimentally investigates the effects of various imputation methods on the calculation of Shapley values, a popular technique for interpreting complex machine learning models. We compare different imputation strategies and assess their impact on feature importance and interaction as determined by Shapley values. Moreover, we also theoretically analyze the effects of missing values on Shapley values. Importantly, our findings reveal that the choice of imputation method can introduce biases that could lead to changes in the Shapley values, thereby affecting the interpretability of the model. Moreover, and that a lower test prediction mean square error (MSE) may not imply a lower MSE in Shapley values and vice versa. Also, while Xgboost is a method that could handle missing data directly, using Xgboost directly on missing data can seriously affect interpretability compared to imputing the data before training Xgboost. This study provides a comprehensive evaluation of imputation methods in the context of model interpretation, offering practical guidance for selecting appropriate techniques based on dataset characteristics and analysis objectives. The results underscore the importance of considering imputation effects to ensure robust and reliable insights from machine learning models.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.