mlsanityData to 7 Oct 2026

Interpretable Clustering with Adaptive Heterogeneous Causal Structure Learning in Mixed Observational Data

arXiv 2509.04415 ↗First posted 4 Sep 2025cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

Understanding causal heterogeneity is essential for scientific discovery in domains such as biology and medicine. However, existing methods lack causal awareness, with insufficient modeling of heterogeneity, confounding, and observational constraints, leading to poor interpretability and difficulty distinguishing true causal heterogeneity from spurious associations. We propose an unsupervised framework, HCL (Interpretable Causal Mechanism-Aware Clustering with Adaptive Heterogeneous Causal Structure Learning), that jointly infers latent clusters and their associated causal structures from mixed-type observational data without requiring temporal ordering, environment labels, interventions or other prior knowledge. HCL relaxes the homogeneity and sufficiency assumptions by introducing an equivalent representation that encodes both structural heterogeneity and confounding. It further develops a bi-directional iterative strategy to alternately refine causal clustering and structure learning, along with a self-supervised regularization that balance cross-cluster universality and specificity. Together, these components enable convergence toward interpretable, heterogeneous causal patterns. Theoretically, we show identifiability of heterogeneous causal structures under mild conditions. Empirically, HCL achieves superior performance in both clustering and structure learning tasks, and recovers biologically meaningful mechanisms in real-world single-cell perturbation data, demonstrating its utility for discovering interpretable, mechanism-level causal heterogeneity.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 19 of its 51 references that are in the map. First posted on arXiv 4 Sep 2025.

Builds on

19 of 51 references.

References · 19

  1. 2019Learning Sparse Nonparametric DAGscited by 207
  2. 2019Causal Discovery from Heterogeneous/Nonstationary Data with Independent Changescited by 166
  3. 2020Amortized Causal Discovery: Learning to Infer Causal Graphs from Time-Series Datacited by 99
  4. 2024Causal machine learning for predicting treatment outcomescited by 73
  5. 2023A Survey on Causal Discovery: Theory and Practicecited by 63
  6. 2022Causal Discovery in Heterogeneous Environments Under the Sparse Mechanism Shift Hypothesiscited by 54
  7. 2019Causal Discovery and Forecasting in Nonstationary Environments with State-Space Modelscited by 45
  8. 2023Causal machine learning for single-cell genomicscited by 22
  9. 2021Optimization-based Causal Estimation from Heterogenous Environmentscited by 16
  10. 2022Causal Discovery with Heterogeneous Observational Datacited by 13
  11. 2022Detecting hidden confounding in observational data using multiple environmentscited by 13
  12. 2023Federated Causality Learning with Explainable Adaptive Optimizationcited by 11
7 more
  1. 2021A Distance Covariance-based Kernel for Nonlinear Causal Clustering in Heterogeneous Populationscited by 10
  2. 2023Discovering Mixtures of Structural Causal Models from Time Series Datacited by 10
  3. 2019Leveraging directed causal discovery to detect latent common causescited by 9
  4. 2024Hierarchical and Density-based Causal Clusteringcited by 6
  5. 2024Interventional Causal Discovery in a Mixture of DAGscited by 3
  6. 2024Causal discovery with endogenous context variablescited by 3
  7. 2024Weakly-supervised causal discovery based on fuzzy knowledge and complex data complementaritycited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.