arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.20184cs.LGstat.ML

Kolmogorov-Arnold 因果生成模型

Kolmogorov-Arnold causal generative models

Alejandro Almodóvar, Mar Elizo, Patricia A. Apellániz, Santiago Zazo, Juan Parras

AI总结:

本文提出KaCGM,一种基于Kolmogorov-Arnold网络的因果生成模型,用于混合类型表格数据,通过分解结构方程实现因果机制的直接检验和可视化,提升生成模型的可解释性和可信度。

摘要:

因果生成模型为从观测数据回答观测性、干预性和反事实查询提供了原则性框架。然而,许多深度因果模型依赖于高表达性架构,其机制不透明,限制了在高风险领域中的可审计性。我们提出KaCGM,一种用于混合类型表格数据的因果生成模型,其中每个结构方程由Kolmogorov--Arnold网络(KAN)参数化。这种分解使能够直接检查学习到的因果机制,包括符号近似和父-子关系的可视化,同时保持查询无关的生成语义。我们引入了基于分布匹配和推断外生变量独立性诊断的验证流程,允许仅使用观测数据进行评估。在合成和半合成基准上的实验显示,KaCGM在性能上与最先进的方法相竞争。一个现实世界的心血管案例研究进一步展示了简化结构方程和可解释的因果效应的提取。这些结果表明,表达性因果生成建模和功能性透明可以同时实现,支持在表格决策设置中的可信部署。代码:https://github.com/aalmodovares/kacgm

英文摘要:

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

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑