arXiv:2512.10152cs.LG2025-12

用可交换性重新思考因果发现,提升真实数据建模能力

Rethinking Bivariate Causal Discovery Through the Lens of Exchangeability

  • 基于可交换性构建层次模型,更灵活刻画因果机制与潜在变量
  • 新合成数据集更贴近图宾根真实数据的统计与因果结构
  • 神经网络方法SynthNN在真实数据上表现优异,验证模型合理性

传统因果发现方法通常基于独立同分布(i.i.d.)或时间序列假设。本文聚焦i.i.d.场景,主张以更广义的可交换性作为建模基础。为此,我们提出一种基于最新因果德菲内定理的可交换层次模型,证明在可交换性下,因果机制与潜变量分布的不确定性能被更准确捕捉。通过深入分析图宾根数据集,我们支持这一假设在真实数据中更为普遍。据此,我们构建了一个模拟该可交换生成过程的新合成数据集,其统计与因果结构比其他i.i.d.合成数据更接近真实数据。进一步提出神经网络方法SynthNN,仅在该合成数据上训练,却在真实图宾根数据上达到先进水平,有力验证了所提可交换生成模型的现实性。

原文摘要 · Abstract (English)

Causal discovery methods have traditionally been developed under two different modeling assumptions: independent and identically distributed (i.i.d.) data and time series data. In this paper, we focus on the i.i.d. setting, arguing that it should be reframed in terms of exchangeability, a strictly more general symmetry principle. For that goal, we propose an exchangeable hierarchical model that builds upon the recent Causal de Finetti theorem. Using this model, we show that both the uncertainty regarding the causal mechanism and the uncertainty in the distribution of latent variables are better captured under the broader assumption of exchangeability. In fact, we argue that this is most often the case with real data, as supported by an in-depth analysis of the Tübingen dataset. Exploiting this insight, we introduce a novel synthetic dataset that mimics the generation process induced by the proposed exchangeable hierarchical model. We show that our exchangeable synthetic dataset mirrors the statistical and causal structure of the Tübingen dataset more closely than other i.i.d. synthetic datasets. Furthermore, we introduce SynthNN, a neural-network-based causal-discovery method trained exclusively on the proposed synthetic dataset. The fact that SynthNN performs competitively with other state-of-the-art methods on the real-world Tübingen dataset provides strong evidence for the realism of the underlying exchangeable generative model.

因果发现可交换性生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。