Distinguishing Cause from Effect with Causal Velocity Models
This work addresses a fundamental challenge in causal inference for researchers, offering a novel approach that extends beyond traditional model classes, though it is incremental in advancing bivariate causal discovery methods.
The paper tackled the problem of inferring causal direction in bivariate structural causal models by introducing a causal velocity parametrization, which allows for causal discovery without assumptions on noise distributions and shows positive results in simulations where existing methods fail.
Bivariate structural causal models (SCM) are often used to infer causal direction by examining their goodness-of-fit under restricted model classes. In this paper, we describe a parametrization of bivariate SCMs in terms of a causal velocity by viewing the cause variable as time in a dynamical system. The velocity implicitly defines counterfactual curves via the solution of initial value problems where the observation specifies the initial condition. Using tools from measure transport, we obtain a unique correspondence between SCMs and the score function of the generated distribution via its causal velocity. Based on this, we derive an objective function that directly regresses the velocity against the score function, the latter of which can be estimated non-parametrically from observational data. We use this to develop a method for bivariate causal discovery that extends beyond known model classes such as additive or location scale noise, and that requires no assumptions on the noise distributions. When the score is estimated well, the objective is also useful for detecting model non-identifiability and misspecification. We present positive results in simulation and benchmark experiments where many existing methods fail, and perform ablation studies to examine the method's sensitivity to accurate score estimation.