70

Data Augmentation via Causal-Residual Bootstrapping

Mateusz Gajewski
Sophia Xiao
Bijan Mazaheri
Main:8 Pages
10 Figures
Bibliography:3 Pages
19 Tables
Appendix:21 Pages
Abstract

Data augmentation integrates domain knowledge into a dataset by making domain-informed modifications to existing data points. For example, image data can be augmented by duplicating images in different tints or orientations, thereby incorporating the knowledge that images may vary in these dimensions. Recent work by Teshima and Sugiyama has explored the integration of causal knowledge (e.g, A causes B causes C) up to conditional independence equivalence. We suggest a related approach for settings with additive noise that can incorporate information beyond a Markov equivalence class. The approach, built on the principle of independent mechanisms, permutes the residuals of models built on marginal probability distributions. Predictive models built on our augmented data demonstrate improved accuracy, for which we provide theoretical backing in linear Gaussian settings.

View on arXiv
Comments on this paper