The Causal Information Bottleneck and Optimal Causal Variable Abstractions

Francisco N. F. Q. Simoes, Mehdi Dastani, Thijs van Ommen

Keywords: causal abstractions, causal inference, information bottleneck

TL;DR: We propose the Causal Information Bottleneck (CIB), a method for learning variable abstractions suitable for causal tasks.

Abstract: To effectively study complex causal systems, it is often useful to construct abstractions of parts of the system by discarding irrelevant details while preserving key features. The Information Bottleneck (IB) method is a widely used approach to construct variable abstractions by compressing random variables while retaining predictive power over a target variable. Traditional methods like IB are purely statistical and ignore underlying causal structures, making them ill-suited for causal tasks. We propose the Causal Information Bottleneck (CIB), a causal extension of the IB, which compresses a set of chosen variables while maintaining causal control over a target variable. This method produces abstractions of (sets of) variables which are causally interpretable, give us insight about the interactions between the abstracted variables and the target variable, and can be used when reasoning about interventions. We present experimental results demonstrating that the learned abstractions accurately capture causal relations as intended.

Code Link: https://github.com/francisco-simoes/cib-optimization-psagd

Readers: auai.org/UAI/2025/Conference, auai.org/UAI/2025/Conference/Area_Chairs, auai.org/UAI/2025/Conference/Reviewers, auai.org/UAI/2025/Conference/Submission245/Authors, auai.org/UAI/2025/Conference/Submission245/Reproducibility_Reviewers

Submission Number: 245