DoubleGen: Debiased Generative Modeling of Counterfactuals

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Luedtke, Alex, Fukumizu, Kenji
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912596437762048
author Luedtke, Alex
Fukumizu, Kenji
author_facet Luedtke, Alex
Fukumizu, Kenji
contents Generative models for counterfactual outcomes face two key sources of bias. Confounding bias arises when approaches fail to account for systematic differences between those who receive the intervention and those who do not. Misspecification bias arises when methods attempt to address confounding through estimation of an auxiliary model, but specify it incorrectly. We introduce DoubleGen, a doubly robust framework that modifies generative modeling training objectives to mitigate these biases. The new objectives rely on two auxiliaries -- a propensity and outcome model -- and successfully address confounding bias even if only one of them is correct. We provide finite-sample guarantees for this robustness property. We further establish conditions under which DoubleGen achieves oracle optimality -- matching the convergence rates standard approaches would enjoy if interventional data were available -- and minimax rate optimality. We illustrate DoubleGen with three examples: diffusion models, flow matching, and autoregressive language models.
format Preprint
id arxiv_https___arxiv_org_abs_2509_16842
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DoubleGen: Debiased Generative Modeling of Counterfactuals
Luedtke, Alex
Fukumizu, Kenji
Machine Learning
Methodology
Generative models for counterfactual outcomes face two key sources of bias. Confounding bias arises when approaches fail to account for systematic differences between those who receive the intervention and those who do not. Misspecification bias arises when methods attempt to address confounding through estimation of an auxiliary model, but specify it incorrectly. We introduce DoubleGen, a doubly robust framework that modifies generative modeling training objectives to mitigate these biases. The new objectives rely on two auxiliaries -- a propensity and outcome model -- and successfully address confounding bias even if only one of them is correct. We provide finite-sample guarantees for this robustness property. We further establish conditions under which DoubleGen achieves oracle optimality -- matching the convergence rates standard approaches would enjoy if interventional data were available -- and minimax rate optimality. We illustrate DoubleGen with three examples: diffusion models, flow matching, and autoregressive language models.
title DoubleGen: Debiased Generative Modeling of Counterfactuals
topic Machine Learning
Methodology
url https://arxiv.org/abs/2509.16842