Jul, 2023

通过生成对抗网络对连续处理进行因果推断的解缠表示学习

TL;DR我们提出了一种用于连续治疗的去混淆表示学习 (DRL) 框架,通过生成与治疗变量解耦的协变量表示来进行反事实结果估计。该框架是一个非参数模型,能够消除治疗与协变量之间的线性和非线性依赖关系,并在模型中嵌入了反事实推理网络以实现去混淆和可信的推理。在合成数据集上广泛实验表明,该 DRL 模型在学习去混淆表示方面表现卓越,并且胜过了针对连续治疗的最先进的反事实推理模型。此外,我们将该 DRL 模型应用于实际的医疗数据集 MIMIC,并展示了红细胞宽度分布与死亡率之间的详细因果关系。