发布于 

因果推断简述

1. 因果推断思想

什么是因果关系?

比较哲学的定义:因果关系(Causality)是一个事件和第二个事件之间的作用关系,其中后一事件被认为是前一事件的结果。而因果推断(Causal Inference),则是在一个较大的系统中确定指定现象(或事件)的实际、独立效果的过程。

说人话:因果关系试图回答的是 what-if 问题,即假如其他一切不变,只改变一个变量 x,如果 y 发生了变化,则认为 x-y 有因果关系。而这一推断过程,称之为因果推断。

因果关系有什么用?

通常我们描述两个事件之间的关系,会常用“相关性”和“因果性”来描述,不妨我们先对比一下这两个区别。

下图是美国每年科研支出与自杀数的关系,可以看到相关性达到了 99% 以上,但是我们可以说是每年科研支出上升导致了自杀数的上升吗? 图 1 而因果关系就是用来解决这一关系的。相关不一定代表因果,同样的,因果也不一定代表相关。回到最初的问题,那我们为什么需要知道因果呢?或者说知道了因果有什么用呢?

简单的回答:因为因果关系可以用来指导业务决策。举个例子,比如我们的一个分析需求是提升产品的用户留存率,通常的做法是以已经流失的用户作为数据,挖掘变量训练一个模型对用户的流失率进行预测,再针对容易流失的用户进行干预。但是实际上,更有效的方式是寻找“留存率提升空间最大的群体”,而不是“最容易流失的群体”。但流失用户识别模型,是很难发掘前者的。

如果我们是全知全能的,比如在上面的例子中,我们可以创造一个平行宇宙,一个对所有用户提供干预,一个不干预,从而获得对每个人干预的效果,再对比干预的成本就可以决定要不要实际使用干预措施。但是现实里,我们不是全知全能的,也就是存在数据缺失(missing data),所以我们才需要研究各种方法来挖掘因果关系,也促成了数据科学发展中的一个重要方向。

怎么衡量因果关系?

这里讲解我们在实验中或观测数据中常遇到的一个问题,以及如何应用因果推断来解决。

辛普森悖论(Simpson's Paradox)

例子:一组数据展示了不同性别的病人在接受药物治疗/未接受药物治疗之后康复的比例。

接受药物治疗 未接受药物治疗
男性 81/87(93%) 234/270(87%)
女性 192/263(73%) 55/80(69%)
总计 273/350(78%) 289/350(83%)

问题:我们应该看拆分性别的数据,还是总计的数据呢?

答案:应该看拆分性别的数据,结论是药物对治疗有帮助。

如果根据这组数据,我们发现无论男女,接受药物治疗后的康复比例均高于未接受药物。总计之中,接受治疗的康复比例是低于未接受治疗的,原因是男女比例的不同。所以我们可以得出结论,该药物对于治疗病情是有效的。如果我们现在换一个数据拆分的维度,还是同样的数据,只是把性别换成了血压分类:

接受药物治疗 未接受药物治疗
低血压 81/87(93%) 234/270(87%)
高血压 192/263(73%) 55/80(69%)
总计 273/350(78%) 289/350(83%)

问题:我们应该看拆分血压的数据,还是总计的数据呢?

答案:应该看总计的数据,结论是药物对治疗无帮助/加重病情。

为什么同样的数据,有的时候需要看整体,而有的时候需要看拆分的呢?原因是我们需要从数据中识别因果关系,从而应用得来准确的结论。

DAG(Directed Acyclic Graphs,有向无环图)——把假设可视化

DAG 是一种把因果关系假设呈现的工具,图中每一个箭头则代表一种因果关系。例如性别是一个混杂因素(Confounder),即药物不会影响性别,但性别影响药物效果,这时我们便需要观测拆分性别后的数据。 图 2 在第二组数据中,血压是药物发生作用的中介变量(mediator),即吃药可能会导致血压变化,从而影响药物效果。这个时候就应该看整体数据。 图 3 在现实情况中,通常因果关系不会这么明显,而影响我们得到正确因果关系的有两种常见偏差。

混杂偏差(Confounding Bias)——Common Causes

类似上边不同性别对于药物效果的影响,混杂偏差主要出现在有混杂变量存在,但该变量不容易或不能被观察到的场景中。 图 4

选择偏差(Selection Bias)——Conditioning on Common Effects

类似上面血压的例子,血压并不是一个后门阻断,如果根据血压来选择实验/对照则会造成选择偏差。但通常背后的隐含变量是无法观测清楚的。 图 5

2. 因果推断概念

因果推断主要是为了解决两个主要问题:

  1. Causal Discovery(因果关系挖掘):温度升高是否导致电费增加?或者在商品价格、转化率、上市时间等多个变量中探究一个因果图,探究哪些变量有因果关系?
  2. Causal Effect(因果效应推理):假设已经知道温度升高时电费增加的原因,我们想知道温度从 20 升到 30 度,会对电费带来多少增加?

这两个问题也对应因果推断中的两个主要框架(Framework):

  1. Structural Causal Models(SCM)。我们首先需要先得到一个因果图,然后对于因果,使用 Structural Equations 来描述它。
  2. Potential Outcome Framework。使用多种手段与方法控制变量,从中得到仅某个因变化而带来的预期结果。

被估量(Estimand)、估计值(Estimate)与估计器/估计量(Estimator)

图 6 被估量是一个使用统计分析方法来估计的统计量(An estimand is a quantity that is to be estimated in a statistical analysis)。

估计值是根据有限数据估计出近似于被估量的统计量(The approximation of the estimand using a finite data sample)。

估计器是根据有限数据估计被估量的方法(The method or formula for arriving at the estimate for an estimand)。

说人话,我们先从什么是估计(estimation)开始。

举个例子,我们想知道全球所有人的平均年龄,则全球人口是我们的样本(population),而平均年龄是我们估计的参数(parameter)。因为平均年龄可以告诉我们年龄的分布情况,所以它是一个参数。如果你真的有全世界人口的年龄,那就不叫估计了。可是现实中,我们无法得到所有人年龄的数据,于是从各国都抽取部分,被抽取的人就组成了我们的样本(sample)。然后我们计算出一个平均年龄,因为这个平均数并不是总人口的真实均值,所以这个均值是被估计的。

现在,我们要升级一下思维:X(学历)和 Y(工资)之间存在一个参数,这个参数决定了 X 和 Y 之间的关系,定义为 β。我们想要知道,全世界范围内,学历到底怎样影响工资,即:β = ?。所以 β(总体的真实参数)是被估量(Estimand),是一个永远无法得知的值。但是我们抽了样本,我们需要一个方法(method or rule),根据我们给的数据计算出一个估计值(Estimate)。而这个方法呢,就叫做估计器(Estimator)。

Causal Discovery

挖掘因果关系的主要方法是贝叶斯网络,及我们上边所提到的 DAG。

举几个例子: 图 7 图 8 图 9 图 10 图 11 图 12 图 13 图 14

四象限

如果上策略:接受 D(1)=1 如果上策略:不接受 D(1)=0
如果没上策略:接受 D(0)=1 Always-Takers:上不上策略都接受 Defiers:上策略不接受,不上就接受(反人格)
如果没上策略:不接受 D(0)=0 Compliers:上策略接受,不上就不接受 Never-Takers:上不上策略都不接受

估计量

ATE(Average Treatment Effect)

群体中的平均效应(The average treatment effect across the population)。 图 15

ITT(Intention-to-treatment Estimation)

有意向干预的效果。即实验设计者可能希望 assigned to treatment 的实验对象(with Z=1)都真的接受了 treatment(with D=1),然而现实中会出现 assigned to treatment 的实验对象(with Z=1)可能并未真的接受 treatment(with D=0),即我们无法强制 assigned to treatment 组的个体真的接受 treatment。所以这里出现了一个漏斗。 图 16

ATT(Average Treatment on the Treated)

群体中实际受到了措施的样本的平均效应(The average treatment effect among those who actually received the treatment in your study)。 图 17

LATE(Local Average Treatment Effect)

treatment effects on compliers。

简单来讲,它刻画了局部处理效应,该结果是否能代表大盘取决于受到策略影响的人群分布情况。对该概念的深入理解依赖于读者对工具变量估计框架的理解。具体介绍见上文工具变量相关文档。

也存在一些其他估计量,比如 CATE(Conditional Average Treatment Effect,条件群体平均措施效应),ATU(Average treatment effect in the untreated,群体中实际未受到措施的样本的平均效应,The average treatment effect among those who actually did not received the treatment in your study)等。 图 18

3. 方法

通常有五大类:匹配、时间序列、增效模型、双重机器学习、工具变量。

3.1. 匹配 Matching

定义:在观测数据中,个体是否接受处理(如参与政策、接受治疗)通常不是随机的,而是由其特征(如年龄、收入、健康状况等)决定的,这会导致选择偏差(Treatment Selection Bias)。对每一个实验组的样本,通过特征匹配或将多维特征映射到倾向性得分(propensity score)上再匹配的方式找到反事实对照组。

下面以倾向性得分即 PSM 为例,场景为教育程度与工资水平的关系: 图 19 步骤:

  1. 计算倾向性得分:估计 X ➡️ T,计算每个个体接受处理的概率(不同家庭条件上大学的概率),即倾向性得分。
  2. 共同支撑集检测:观测处理组和控制组的倾向得分分布重叠,重叠大则效果好(能匹配到类似的人)。
  3. 匹配:根据倾向得分值,为每个处理组个体匹配一个或多个控制组个体。常用匹配方法包括:最近邻、卡尺匹配、分层匹配、核匹配、马氏距离匹配等等。
  4. 平衡性检测:检测两组是否平衡,比如协变量的均值/方差等。
  5. 因果估计:ATE、ATT。

PROS

  1. 降维处理:将高维协变量压缩为一维倾向得分,简化匹配过程。
  2. 灵活性:适用于二分类处理变量(如是否参与政策),可结合多种匹配方法。
  3. 接近随机实验:通过匹配消除可观测变量的选择偏差,使结果更接近因果推断。

CONS

  1. 依赖可观测变量:无法解决未被测量的混杂因素(如个体偏好、未记录的健康状况),若存在严重不可观测偏差,结果可能有误。
  2. 共同支持限制:若处理组和控制组的倾向得分分布重叠较少(如极端样本),匹配效果较差,需剔除不满足条件的样本。
  3. 模型依赖性:倾向得分的估计依赖模型设定(如线性回归、逻辑回归),若模型不满足 Groundtruth 上的关系就会产生误差。

3.2. 时间序列 Time Series

3.2.1. DID

图 20 DID 即双重差分(difference in difference),用差值比差值,是很简单的群体效应估计方法。只需要将样本数据随机分成两组,对其中一组进行干预,评估时使用实验期间对比实验前的差值,然后评估实验组与对照组的差值的差值即可。

CONS

需要满足平行趋势假设,也就是实验前,两组的表现是平行一致的。

其他

业务上通常会使用 PSM + DID 一起用,原因是两者可以巧妙地相互补足短板,PSM 做样本上的降噪(横向),DID 做时间上的降噪(纵向)。但需要注意的是,因为经过两侧模型的预估,我们犯第一类/第二类错误的概率也在累积。

3.2.2. 合成控制

核心思想:与 Matching 不同于“找一个对照组”,合成控制是“生成一个对照组”。通过构建一个由多个控制组单元(如其他地区、国家)加权合成的“虚拟控制组”,使其在干预前的特征与处理组尽可能相似,从而通过比较处理组与合成控制组在干预后的差异,估计政策的净效应。

例子:评估加州控烟政策的效果(Abadie et al., 2010)

处理组:美国加州 1988 年实施控烟政策(提高烟草税、禁止公共场所吸烟)。

控制组:其他未实施该政策的州。

协变量:干预前的人均香烟销量、收入、人口结构、医疗支出等,通过加权最小二乘法求解权重,使得各州的加权合成的控制组与实验组在实验干预前的表现尽可能接近。

CONS

  1. 合成控制组的效果高度依赖协变量的选择。
  2. 相对于 PSM 来讲解释性差。

3.3. 增效模型 Uplift

核心思想:建立一个模型(S-Learner,T-Learner,X-Learner 等等)来分别预估两个概率(上策略后接受的概率、不上策略后接受的概率),从而可以将有限的资源花给 Compliers,最大化策略效果。常用在发优惠券、激励广告等有成本的激励策略上。

3.4. 双重机器学习 DML

核心思想:我们只关心 Treatment T 对 outcome Y 的影响,因此我们可以首先使用 X 回归 T,得到一个 T 的残差,然后使用 X 回归 Y,得到一个 Y 的残差,最后使用 T 的残差回归 Y 的残差,估计的参数即我们想要的 ATE。即: 图 21 以教育程度与工资水平的关系为例: 图 22 我们先估计 X ➡️ College 的关系获得残差: 图 23 再估计 X ➡️ Wage 的关系获得残差: 图 24 最后通过残差估计残差: 图 25 CONS

DML 需要建立在 Unconfoundedness 假设上:因为需要用混淆变量估计 Treatment 和 Outcome,所以所有混淆变量都需要被找到。

3.5. 工具变量 IV

核心思想:我们在预估的时候会存在内生变量的影响。为了解决这个问题,我们可以找一个与 x 相关、与 y 不直接相关、但与 e 不相关的变量作为工具,代替 x 放入到方程中。

以教育程度与工资水平的关系为例: 图 26 在研究受教育程度对工资水平的影响时,通常会遗漏重要的解释变量“个人能力”,以致于估计出来的系数包含了个人能力对工资的影响。由于“住在大学附近”并不能直接影响工资水平,且和误差项中的“个人能力”不相关,但是可能会影响个人是否上大学,因此可以用“住在大学附近”作为工具变量替换“教育水平”。

实际操作的方式是一个二阶段估计:一阶段估计“住在大学附近”➡️“教育水平”,获得一个工具变量“hat(教育水平)”;二阶段使用“hat(教育水平)”➡️“工资水平”。

在寻找工具变量的时候需要满足:

  1. 外生性 图 27
  2. 相关性 图 28 一些有趣的工具变量:

  3. 墨西哥移民规模是否影响个体就业选择 - 工具变量:用移民来源社区降雨量替代移民来源社区。 - 相关性:样本所属的墨西哥社区主要从事依赖降雨的农业生产,降雨量会影响社区农业收入进而影响农业预期收入,并最终影响移民决策。

  4. 子女数目增加对父母劳动市场参与 - 工具变量:头两个孩子的性别构成。 - 相关性:父母一般希望子女的性别构成多样化,所以如果头两胎都是男孩或女孩,那么他们就希望再生一个性别不一样的;但如果前两胎一个男孩一个女孩,那性别多样化的任务已经完成了,就不会再生了。
  5. 越战经历对之后收入的影响 - 工具变量:作者发现国防部征兵的时候是给适龄的男性抽一个号码(Draft Lottery),然后定一个上限,号码小于这个上限的人在征兵范围之内,于是他定义征兵号是否小于这个上限为 draft eligibility,作为工具变量。这是一个巧妙利用自然实验的例子。 - 从因果推断角度理解,AB 实验干预其实是一个天然的工具变量,关于 AB 实验详见一文带你速通 AB 实验。 - 相关性:在这个范围内的人,才能被征兵。
  6. 更多:IV 在哪里?奇思妙想的工具变量

CONS

  1. IV 方法在计量中很流行,但在工业界应用较少:实际难论证工具变量和 Y 之间没有其他中介变量。
  2. IV 存在 P-hacking 风险(Brodeur et al. 2020),降雨量作为 IV 的泛滥:https://osf.io/preprints/socarxiv/9qj4f

3.6. 总结

模块 方法 被估量 实验变量类型 高维数据控制 High dimensional control 无混淆性 Unconfoundedness 其他假设 双重鲁棒性 doubly robustness 置信区间 Confidence interval
匹配 matching cem ATT 离散 1 1
匹配 matching psm ATT 离散 1 1 1
匹配 matching iptw ATE 离散 1 1 1
时间序列 Time series did ATT 平行趋势 Parallel trend 1
时间序列 Time series Synthetic control ATT 稳定线性组合 Stable linear combination 1
时间序列 Time series psm + did ATT 1 平行趋势 Parallel trend 1
uplift model s-learner ITE/ATT/ATE 1 1
uplift model T-learner ITE/ATT/ATE 1 1
uplift model x-learner ITE/ATT/ATE 1 1
uplift model linearDR-learner ITE/ATT/ATE 1 1 1 1
uplift model forestDR-learner ITE/ATT/ATE 1 1 1 1
uplift model orthoforestDR-learner ITE/ATT/ATE 1 1 1 1
DML LinearDML ITE/ATT/ATE/MTE 离散/连续 1 1 1
DML CausalForestDML ITE/ATT/ATE/MTE 1 1 1
DML OrthoforestDML ITE/ATT/ATE/MTE 1 1 1
IV 2SLS LATE IV 假设 1

4. 大模型与因果推断

在大模型与因果推断的结合上,目前还较浅,主要聚焦于三个方向:

  1. 通过因果图的方式,用大模型对因果图进行增强识别。利用因果发现算法与大模型相结合,以获取更精准的因果图。
  2. Agent,利用大模型的智能基础及其工具使用能力,配合外部工具,综合构建一个智能代理,帮助用户完成一些因果推断任务。
  3. ABM(agent base modeling)基于代理的建模,通过多个智能体之间的相互交互形成现实生活中的场景,即形成一个仿真生态,进而辅助策略的研究和探索。

5. Ref