English

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.

Keywords

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}
}
R2 v1 2026-06-28T20:22:11.891Z