From Local to Cluster: A Unified Framework for Causal Discovery with Latent Variables

TL;DR

L2C is a unified framework that automatically discovers clusters and infers macro causal structures in the presence of latent variables, improving causal discovery accuracy.

Contribution

It introduces L2C, which bridges local and cluster-level causal discovery without assuming known clusters or causal sufficiency, using a novel cluster reduction theorem.

Findings

L2C accurately recovers ground truth clusters.

L2C achieves superior macro causal effect identification.

L2C is computationally efficient and theoretically sound.

Abstract

Latent variables pose a fundamental challenge to causal discovery and inference. Conventional local methods focus on direct neighbors but fail to provide macro level insights. Cluster level methods enable macro causal reasoning but either assume clusters are known a priori or require causal sufficiency. Moreover, directly applying single variable causal discovery methods to cluster level problems violates causal sufficiency and leads to incorrect results. To overcome these limitations, this paper proposes L2C (Local to Cluster Causal Abstraction), a unified framework that bridges local structure learning and cluster level causal discovery. Unlike prior work that requires a complete manual assignment of micro variables to clusters, L2C discovers the partition automatically from local causal patterns. Our solution leverages a cluster reduction theorem to reduce any cluster to at most three nodes without loss of causal information, applies local causal discovery to identify direct causes, effects, and V structures in the presence of latent variables, and performs macro level causal inference via cluster level calculus on the learned cluster graph. L2C does not assume causal sufficiency, as latent variables are handled through local discovery. Theoretical analysis shows that L2C ensures soundness, atomic completeness, and computational efficiency. Extensive experiments on synthetic and real world data demonstrate that L2C accurately recovers ground truth clusters and achieves superior macro causal effect identification compared to existing baselines.