Local Feature Selection without Label or Feature Leakage for Interpretable Machine Learning Predictions

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Oosterhuis, Harrie, Lyu, Lijun, Anand, Avishek
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911957196472320
author Oosterhuis, Harrie
Lyu, Lijun
Anand, Avishek
author_facet Oosterhuis, Harrie
Lyu, Lijun
Anand, Avishek
contents Local feature selection in machine learning provides instance-specific explanations by focusing on the most relevant features for each prediction, enhancing the interpretability of complex models. However, such methods tend to produce misleading explanations by encoding additional information in their selections. In this work, we attribute the problem of misleading selections by formalizing the concepts of label and feature leakage. We rigorously derive the necessary and sufficient conditions under which we can guarantee no leakage, and show existing methods do not meet these conditions. Furthermore, we propose the first local feature selection method that is proven to have no leakage called SUWR. Our experimental results indicate that SUWR is less prone to overfitting and combines state-of-the-art predictive performance with high feature-selection sparsity. Our generic and easily extendable formal approach provides a strong theoretical basis for future work on interpretability with reliable explanations.
format Preprint
id arxiv_https___arxiv_org_abs_2407_11778
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Local Feature Selection without Label or Feature Leakage for Interpretable Machine Learning Predictions
Oosterhuis, Harrie
Lyu, Lijun
Anand, Avishek
Machine Learning
Local feature selection in machine learning provides instance-specific explanations by focusing on the most relevant features for each prediction, enhancing the interpretability of complex models. However, such methods tend to produce misleading explanations by encoding additional information in their selections. In this work, we attribute the problem of misleading selections by formalizing the concepts of label and feature leakage. We rigorously derive the necessary and sufficient conditions under which we can guarantee no leakage, and show existing methods do not meet these conditions. Furthermore, we propose the first local feature selection method that is proven to have no leakage called SUWR. Our experimental results indicate that SUWR is less prone to overfitting and combines state-of-the-art predictive performance with high feature-selection sparsity. Our generic and easily extendable formal approach provides a strong theoretical basis for future work on interpretability with reliable explanations.
title Local Feature Selection without Label or Feature Leakage for Interpretable Machine Learning Predictions
topic Machine Learning
url https://arxiv.org/abs/2407.11778