基本信息

人们对将机器学习应用于医疗保健、经济和教育等领域的因果推理问题产生了浓厚的兴趣。特别是个体层面的因果推理具有重要的应用，例如精准医疗。我们给出了一种新的理论分析和算法系列，用于在强可忽略性的假设下根据观察数据预测个体治疗效果（ITE）。该算法学习“平衡”表示，使得诱导的处理分布和控制分布看起来相似。我们给出了一种新颖、简单且直观的泛化误差界，表明表示的预期 ITE 估计误差受到该表示的标准泛化误差与该表示引起的处理分布和控制分布之间的距离之和的限制。我们使用积分概率度量来测量分布之间的距离，得出 Wasserstein 和最大平均差异 (MMD) 距离的明确界限。对真实数据和模拟数据的实验表明，新算法可以匹配或优于最先进的算法 ...