Estimating the counterfactual outcome of treatment is essential for decision-making in public health and clinical science, among others. Often, treatments are administered in a sequential, time-varying manner, leading to an exponentially increased number of possible counterfactual outcomes. Furthermore, in modern applications, the outcomes are high-dimensional and conventional average treatment effect estimation fails to capture disparities in individuals. To tackle these challenges, we propose a novel conditional generative framework capable of producing counterfactual samples under time-varying treatment, without the need for explicit density estimation. Our method carefully addresses the distribution mismatch between the observed and counterfactual distributions via a loss function based on inverse probability weighting. We present a thorough evaluation of our method using both synthetic and real-world data. Our results demonstrate that our method is capable of generating high-quality counterfactual samples and outperforms the state-of-the-art baselines.
翻译:估计治疗的反事实结果对于公共卫生和临床科学等领域的决策至关重要。通常,治疗以序贯、时变的方式实施,导致可能的反事实结果数量呈指数级增长。此外,在现代应用中,结果具有高维性,传统的平均治疗效果估计无法捕捉个体间的差异。为应对这些挑战,我们提出了一种新颖的条件生成框架,能够在时变治疗下生成反事实样本,无需显式密度估计。我们的方法通过基于逆概率加权的损失函数,仔细解决了观测分布与反事实分布之间的分布不匹配问题。我们使用合成数据和真实数据对所提方法进行了全面评估。结果表明,我们的方法能够生成高质量的反事实样本,并优于当前最先进的基线方法。