Accurate Estimation of Feature Importance Faithfulness for Tree Models

Authors

  • Mateusz Gajewski Poznan University of Technology IDEAS NCBR
  • Adam Karczmarz Faculty of Mathematics, Informatics and Mechanics University of Warsaw, IDEAS NCBR
  • Mateusz Rapicki Faculty of Mathematics, Informatics and Mechanics University of Warsaw,
  • Piotr Sankowski Faculty of Mathematics, Informatics and Mechanics University of Warsaw, MIM Solutions

DOI:

https://doi.org/10.1609/aaai.v39i16.33834

Abstract

In this paper, we consider a perturbation-based metric of predictive faithfulness of feature rankings (or attributions) that we call PGI squared When applied to decision tree-based regression models, the metric can be computed exactly and efficiently for arbitrary independent feature perturbation distributions. In particular, the computation does not involve Monte Carlo sampling that has been typically used for computing similar metrics and which is inherently prone to inaccuracies. As a second contribution, we proposed a procedure for constructing feature ranking based on PGI squared. Our results indicate the proposed ranking method is comparable to the widely recognized SHAP explainer, offering a viable alternative for assessing feature importance in tree-based models.

Downloads

Published

2025-04-11

How to Cite

Gajewski, M., Karczmarz, A., Rapicki, M., & Sankowski, P. (2025). Accurate Estimation of Feature Importance Faithfulness for Tree Models. Proceedings of the AAAI Conference on Artificial Intelligence, 39(16), 16691-16698. https://doi.org/10.1609/aaai.v39i16.33834

Issue

Section

AAAI Technical Track on Machine Learning II