407
v1v2v3v4 (latest)

Partial Information Decomposition for Data Interpretability and Feature Selection

International Conference on Artificial Intelligence and Statistics (AISTATS), 2024
Main:8 Pages
14 Figures
Bibliography:3 Pages
1 Tables
Appendix:18 Pages
Abstract

In this paper, we introduce Partial Information Decomposition of Features (PIDF), a new paradigm for simultaneous data interpretability and feature selection. Contrary to traditional methods that assign a single importance value, our approach is based on three metrics per feature: the mutual information shared with the target variable, the feature's contribution to synergistic information, and the amount of this information that is redundant. In particular, we develop a novel procedure based on these three metrics, which reveals not only how features are correlated with the target but also the additional and overlapping information provided by considering them in combination with other features. We extensively evaluate PIDF using both synthetic and real-world data, demonstrating its potential applications and effectiveness, by considering case studies from genetics and neuroscience.

View on arXiv
Comments on this paper