Statistically Valid Variable Importance Assessment through Conditional Permutations - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

Statistically Valid Variable Importance Assessment through Conditional Permutations

Résumé

Variable importance assessment has become a crucial step in machine-learning applications when using complex learners, such as deep neural networks, on large-scale data. Removal-based importance assessment is currently the reference approach, particularly when statistical guarantees are sought to justify variable inclusion. It is often implemented with variable permutation schemes. On the flip side, these approaches risk misidentifying unimportant variables as important in the presence of correlations among covariates. Here we develop a systematic approach for studying Conditional Permutation Importance (CPI) that is model agnostic and computationally lean, as well as reusable benchmarks of state-of-the-art variable importance estimators. We show theoretically and empirically that CPI overcomes the limitations of standard permutation importance by providing accurate type-I error control. When used with a deep neural network, CPI consistently showed top accuracy across benchmarks. An experiment on real-world data analysis in a largescale medical dataset showed that CPI provides a more parsimonious selection of statistically significant variables. Our results suggest that CPI can be readily used as drop-in replacement for permutation-based methods.
Fichier principal
Vignette du fichier
NeurIPS_submission_supp.pdf (3.98 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04334161 , version 1 (10-12-2023)

Licence

Identifiants

Citer

Ahmad Chamma, Denis Engemann, Bertrand Thirion. Statistically Valid Variable Importance Assessment through Conditional Permutations. NeurIPS 2023 - 37th Conference on Neural Information Processing Systems, Dec 2023, New Orleans (Louisiana), United States. ⟨hal-04334161⟩
126 Consultations
106 Téléchargements

Altmetric

Partager

More