A theory of independent mechanisms for extrapolation in generative models
This work addresses the challenge of making generative models useful for predictions in new contexts, which is an incremental step in improving model robustness and applicability.
The paper tackles the problem of enabling generative models to extrapolate to unseen environments by proposing a theoretical framework based on the principle of independence of mechanisms, and demonstrates through experiments that enforcing this principle improves extrapolation capabilities.
Generative models can be trained to emulate complex empirical data, but are they useful to make predictions in the context of previously unobserved environments? An intuitive idea to promote such extrapolation capabilities is to have the architecture of such model reflect a causal graph of the true data generating process, such that one can intervene on each node independently of the others. However, the nodes of this graph are usually unobserved, leading to overparameterization and lack of identifiability of the causal structure. We develop a theoretical framework to address this challenging situation by defining a weaker form of identifiability, based on the principle of independence of mechanisms. We demonstrate on toy examples that classical stochastic gradient descent can hinder the model's extrapolation capabilities, suggesting independence of mechanisms should be enforced explicitly during training. Experiments on deep generative models trained on real world data support these insights and illustrate how the extrapolation capabilities of such models can be leveraged.