Mar, 2024

异质剂量 - 反应曲线估计的对比平衡表示学习

TL;DR对于决策制定来说,估计个体对不同治疗剂量的潜在反应至关重要,尤其在精准医学和管理科学领域。本文通过学习与治疗变量独立的协变量表示来预测反事实结果,然而,这种独立性约束忽略了对于反事实预测有用的大量协变量信息,特别是在治疗变量是连续的情况下。为了解决这个问题,本文首先从理论上证明了平衡和预测表示在无偏估计异质剂量响应曲线方面的重要性,即学习得到的表示需满足协变量与两个治疗变量及潜在反应之间的条件独立性。基于此,我们提出了一种新颖的对比平衡表示学习网络,使用部分距离度量,称为 CRNet,用于估计异质剂量响应曲线,同时不丢失治疗剂量的连续性。我们在合成和真实世界数据集上进行了广泛实验,证明了我们的方法显著优于之前的方法。