Modeling true world data-generating processes lies at the heart of empirical science. Structural Causal Models (SCMs) and their associated Directed Acyclic Graphs (DAGs) provide an increasingly popular answer to such problems by defining the causal generative process that transforms random noise into observations. However, learning them from observational data poses an ill-posed and NP-hard inverse problem in general. In this work, we propose a new and equivalent formalism that does not require DAGs to describe them, viewed as fixed-point problems on the causally ordered variables, and we show three important cases where they can be uniquely recovered given the topological ordering (TO). To the best of our knowledge, we obtain the weakest conditions for their recovery when TO is known. Based on this, we design a two-stage causal generative model that first infers the causal order from observations in a zero-shot manner, thus by-passing the search, and then learns the generative fixed-point SCM on the ordered variables. To infer TOs from observations, we propose to amortize the learning of TOs on generated datasets by sequentially predicting the leaves of graphs seen during training. To learn fixed-point SCMs, we design a transformer-based architecture that exploits a new attention mechanism enabling the modeling of causal structures, and show that this parameterization is consistent with our formalism. Finally, we conduct an extensive evaluation of each method individually, and show that when combined, our model outperforms various baselines on generated out-of-distribution problems.
翻译:摘要:对真实世界数据生成过程进行建模是实证科学的核心。结构因果模型(SCM)及其相关的有向无环图(DAG)通过定义将随机噪声转化为观测的因果生成过程,为这类问题提供了日益流行的解决方案。然而,从观测数据中学习这些模型通常构成一个不适定且NP难的逆问题。本研究提出一种无需DAG描述的新等价形式,将其视为因果有序变量上的定点问题,并展示了在已知拓扑排序(TO)条件下可唯一恢复的三个重要情形。据我们所知,在已知TO的前提下,我们获得了最弱的恢复条件。基于此,我们设计了一个两阶段因果生成模型:首先以零样本方式从观测中推断因果顺序(从而绕开搜索过程),然后在有序变量上学习生成式定点SCM。为从观测中推断TO,我们提出通过逐步预测训练中见过图的叶节点,在生成数据集上摊销TO的学习过程。为学习定点SCM,我们设计了基于Transformer的架构,其利用新型注意力机制实现因果结构建模,并证明该参数化方法与我们的形式体系一致。最后,我们对各方法分别进行广泛评估,并表明当组合使用时,我们的模型在生成的非分布外问题中优于多种基线方法。