Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Differentiable Causal Discovery from Interventional Data

About

Learning a causal directed acyclic graph from data is a challenging task that involves solving a combinatorial problem for which the solution is not always identifiable. A new line of work reformulates this problem as a continuous constrained optimization one, which is solved via the augmented Lagrangian method. However, most methods based on this idea do not make use of interventional data, which can significantly alleviate identifiability issues. This work constitutes a new step in this direction by proposing a theoretically-grounded method based on neural networks that can leverage interventional data. We illustrate the flexibility of the continuous-constrained framework by taking advantage of expressive neural architectures such as normalizing flows. We show that our approach compares favorably to the state of the art in a variety of settings, including perfect and imperfect interventions for which the targeted nodes may even be unknown.

Philippe Brouillard, S\'ebastien Lachapelle, Alexandre Lacoste, Simon Lacoste-Julien, Alexandre Drouin• 2020

Related benchmarks

TaskDatasetResultRank
Crash AvoidanceCrash Spuriousness S (test)
Success Rate53.4
10
Object StackingStack Spuriousness S (test)
Success Rate90.5
10
Box/Door UnlockingUnlock Spuriousness S (test)
Success Rate23.1
10
Box/Door UnlockingUnlock Composition C (test)
Success Rate3.62e+3
10
Object StackingStack Composition C (test)
Success Rate73.9
10
Box/Door UnlockingUnlock In-distribution I (test)
Success Rate4.49e+3
10
Crash AvoidanceCrash In-distribution I (test)
Success Rate42.3
10
Object StackingStack In-distribution I (test)
Success Rate92.7
10
Crash AvoidanceCrash Composition C (test)
Success Rate8.4
10
Causal DiscoveryChemistry environment Jungle (ID)
Success Rate6.30e+3
9
Showing 10 of 17 rows

Other info

Follow for update