Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Kolmogorov-Arnold causal generative models

About

Causal generative models provide a principled framework for answering observational, interventional, and counterfactual queries from observational data. However, many deep causal models rely on highly expressive architectures with opaque mechanisms, limiting auditability in high-stakes domains. We propose KaCGM, a causal generative model for mixed-type tabular data where each structural equation is parameterized by a Kolmogorov--Arnold Network (KAN). This decomposition enables direct inspection of learned causal mechanisms, including symbolic approximations and visualization of parent--child relationships, while preserving query-agnostic generative semantics. We introduce a validation pipeline based on distributional matching and independence diagnostics of inferred exogenous variables, allowing assessment using observational data alone. Experiments on synthetic and semi-synthetic benchmarks show competitive performance against state-of-the-art methods. A real-world cardiovascular case study further demonstrates the extraction of simplified structural equations and interpretable causal effects. These results suggest that expressive causal generative modeling and functional transparency can be achieved jointly, supporting trustworthy deployment in tabular decision-making settings. Code: https://github.com/aalmodovares/kacgm

Alejandro Almod\'ovar, Mar Elizo, Patricia A. Apell\'aniz, Santiago Zazo, Juan Parras• 2026

Related benchmarks

TaskDatasetResultRank
Causal Generative Modeling11 synthetic datasets Mixed type variables (test)
RF Observed Accuracy0.52
7
Causal InferenceSachs Non-additive Noise Semi-synthetic
Accuracy (RF, Observed)68
5
Causal InferenceSachs Additive Noise Semi-synthetic
RF Accuracy (Observed)54
5
Causal Generative Modeling11 synthetic datasets Continuous variables (test)
RF Accuracy (Observed)52
5
Showing 4 of 4 rows

Other info

Follow for update