Modeling and Discovering Direct Causes for Predictive Models
Machine Learning
2025-05-20 v2 Artificial Intelligence
Methodology
Abstract
We introduce a causal modeling framework that captures the input-output behavior of predictive models (e.g., machine learning models). The framework enables us to identify features that directly cause the predictions, which has broad implications for data collection and model evaluation. We then present sound and complete algorithms for discovering direct causes (from data) under some assumptions. Furthermore, we propose a novel independence rule that can be integrated with the algorithms to accelerate the discovery process, as we demonstrate both theoretically and empirically.
Cite
@article{arxiv.2412.02878,
title = {Modeling and Discovering Direct Causes for Predictive Models},
author = {Yizuo Chen and Amit Bhatia},
journal= {arXiv preprint arXiv:2412.02878},
year = {2025}
}