因果模型
CausalModel
是一个核心对象来执行 识别 和寻找工具变量。
在介绍因果模型之前,我们首先需要阐明 干涉 的定义。干涉是对整个人群的,并给每一个人一些操作。 [Pearl] 定义了 \(do\)-operator 来描述了这样的操作。概率模型不能够服务预测干涉效果,这导致了对因果模型的需求。
对 causal model 的正式定义归因于 [Pearl] 。一个因果模型是一个三元组
其中
\(U\) 是 外生的 (由模型外的因素决定的变量);
\(V\) 是 内生的 决定于 \(U \cup V\), 和 \(F\) 是这样的一组函数
其中 \(pa_i \subset V \backslash V_i\).
例如, \(M = \left< U, V, F\right>\) 是一个因果模型其中
这样
注意每个因果模型都可以和一个DAG关联并编码变量之间必要的因果关系信息。
YLearn使用 CausalModel
来表示一个因果模型并支持许多关于因果模型的操作比如 识别 。
识别
为了表征干涉的效果,需要考虑 因果效应 ,其是一个因果估计量包含 \(do\)-operator 。把因果效应转变为对应的统计估计量的行为被称为 识别 且
在YLearn中的 CausalModel
里实现。注意不是所有的因果效应都能被转变为统计估计量的。我们把这样的因果效应称为不可识别的。我们列出几个 CausalModel 支持的识别方法。
后门调整
\(X\) 对 \(Y\) 的因果效应由下式给出
\[P(y|do(x)) = \sum_w P(y|x, w)P(w)\]
如果变量的集合 \(W\) 满足后门准则关于 \((X, Y)\).
前门调整
\(X\) 对 \(Y\) 的因果效应由下式给出
如果变量的集合 \(W\) 满足前门准则关于 \((X, Y)\) and if \(P(x, w) > 0\).
通用识别
[Shpitser2006] 给出一个必要充分的图的条件这样任意一个集合的变量对另一个任意集合的因果效应能够被唯一的识别无论它是不是可识别的。 我们称验证这个条件对应的行动为 通用识别。
找到工具变量
当有未观测到的 \(X\) 和 \(Y\) 的混淆因素时,工具变量在识别和估计 \(X\) 对 \(Y\) 的因果效应很有用处。 一组变量 \(Z\) 被称为一组 工具变量 如果 \(Z\) 中任意的 \(z\) :
\(z\) 对 \(X\) 有因果效应。
\(z\) 对 \(Y\) 的因果效应完全由于 \(X\) 。
没有后门路径从 \(z\) 到 \(Y\) 。
例子1: 用通用识别方法识别因果效应
对于图中的因果结构,我们想要使用 通用识别 方法识别 \(X\) 对 \(Y\) 的因果效应。第一步是用 CausalModel
表示因果结构。
from ylearn.causal_model.graph import CausalGraph
causation = {
'X': ['Z2'],
'Z1': ['X', 'Z2'],
'Y': ['Z1', 'Z3'],
'Z3': ['Z2'],
'Z2': [],
}
arcs = [('X', 'Z2'), ('X', 'Z3'), ('X', 'Y'), ('Z2', 'Y')]
cg = CausalGraph(causation=causation, latent_confounding_arcs=arcs)
然后我们需要为编码在 cg
中的因果结构定义一个 CausalModel
的实例,从而进行识别。
from ylearn.causal_model.model import CausalModel
cm = CausalModel(causal_model=cg)
stat_estimand = cm.id(y={'Y'}, x={'X'})
stat_estimand.show_latex_expression()
>>> :math:`\sum_{Z3, Z1, Z2}[P(Z2)P(Y|Z3, Z2)][P(Z1|Z2, X)][P(Z3|Z2)]`
结果是想要的识别的在给定的因果结构中 \(X\) 对 \(Y\) 的因果效应。
例子2: 使用后门调整识别因果效应
对于图中的因果结构,我们想要使用 后门调整 方法识别 \(X\) 对 \(Y\) 的因果效应。
from ylearn.causal_model.graph import CausalGraph
from ylearn.causal_model.model import CausalModel
causation = {
'X1': [],
'X2': [],
'X3': ['X1'],
'X4': ['X1', 'X2'],
'X5': ['X2'],
'X6': ['X'],
'X': ['X3', 'X4'],
'Y': ['X6', 'X4', 'X5', 'X'],
}
cg = CausalGraph(causation=causation)
cm = CausalModel(causal_graph=cg)
backdoor_set, prob = cm3.identify(treatment={'X'}, outcome={'Y'}, identify_method=('backdoor', 'simple'))['backdoor']
print(backdoor_set)
>>> ['X3', 'X4']
例子3: 找到合理的工具变量
我们想要为 \(t\) 对 \(g\) 的因果效应找到合理的工具变量。
causation = {
'p':[],
't': ['p'],
'l': ['p'],
'g': ['t', 'l']
}
arc = [('t', 'g')]
cg = CausalGraph(causation=causation, latent_confounding_arcs=arc)
cm = CausalModel(causal_graph=cg)
cm.get_iv('t', 'g')
>>> No valid instrument variable has been found.
我们依然想要在这个新的因果结构中,为 \(t\) 对 \(g\) 的因果效应找到合理的工具变量。
causation = {
'p':[],
't': ['p', 'l'],
'l': [],
'g': ['t', 'l']
}
arc = [('t', 'g')]
cg = CausalGraph(causation=causation, latent_confounding_arcs=arc)
cm = CausalModel(causal_graph=cg)
cm.get_iv('t', 'g')
>>> {'p'}
类结构
- class ylearn.causal_model.CausalModel(causal_graph=None, data=None)
- 参数
causal_graph (CausalGraph, optional, default=None) – CausalGraph的实例,编码了因果结构
data (pandas.DataFrame, optional, default=None) – 用于发现因果结构的数据,如果causal_graph没有提供。
- id(y, x, prob=None, graph=None)
识别因果量 \(P(y|do(x))\) 如果可识别否则返回 raise
IdentificationError
。 注意这里我们仅考虑半马尔可夫因果模型,其中每个未观测到的变量正好是两个节点的父节点。这是因为任何的有未观测的变量的因果模型可以被转变为 一个编码了同样集合的条件独立性的半马尔可夫模型。- 参数
y (set of str) – 结果的名字的集合。
x (set of str) – 治疗的名字的集合。
prob (Prob, optional, default=None) – 编码在图中的概率分布。
graph (CausalGraph) – CausalGraph编码了对应的因果结构中的信息。
- 返回
转变的因果效应的概率分布。
- 返回类型
- 抛出
IdentificationError – 如果感兴趣的因果效应不能识别,则raise IdentificationError。
- is_valid_backdoor_set(set_, treatment, outcome)
决定给定的集合是否是对结果的治疗的因果效应的一个合理的后门调整集合。
- 参数
set (set) – 调整集合。
treatment (set or list of str) – 治疗的名字。对单个治疗,str也是可以接受的。
outcome (set or list of str) – 结果的名字。对单个结果,str也是可以接受的。
- 返回
True,如果在现在的因果图中,给定的集合是对结果的治疗的因果效应的一个合理的后门调整集合。
- 返回类型
bool
- get_backdoor_set(treatment, outcome, adjust='simple', print_info=False)
对给定的治疗和结果返回后门调整集合。
- 参数
treatment (set or list of str) – 治疗的名字。对单个治疗,str也是可以接受的。
outcome (set or list of str) – 结果的名字。对单个结果,str也是可以接受的。
adjust (str) –
设置后门集合的样式。可选的选项是
simple: 直接返回治疗的父节点集合
minimal: 返回最小的后门调整集合
all: 返回所有合理的后门调整集合。
print_info (bool, default=False) – 如果为True,打印识别的结果。
- 返回
第一个元素是调整列表,同时第二个是编码的Prob。
- 返回类型
两个元素的元组
- 抛出
IdentificationError – Raise error如果样式不在simple,minimal或者all或者没有集合能满足后门准则。
- get_backdoor_path(treatment, outcome)
返回所有的连接治疗和结果的后门路径。
- 参数
treatment (str) – 治疗的名字。
outcome (str) – 结果的名字。
- 返回
一个包含图中所有合理的治疗和结果之间的后门路径的列表。
- 返回类型
list
- has_collider(path, backdoor_path=True)
如果现在图的path中有一个对撞,返回True,否则返回False。
- 参数
path (list of str) – 包含路径中节点的列表。
backdoor_path (bool, default=True) – 该路径是否是一个后门路径。
- 返回
True,如果path有一个对撞。
- 返回类型
bool
- is_connected_backdoor_path(path)
测试是否一个后门路径是连接的。
- 参数
path (list of str) – 描述这个路径的列表。
- 返回
True,如果路径是一个d-connected的后门路径,否则False。
- 返回类型
bool
- is_frontdoor_set(set_, treatment, outcome)
决定给定的集合是否是对结果的治疗的因果效应的一个合理的前门调整集合。
- 参数
set (set) – 等待决定是否是合理的前门调整集合的集合。
treatment (str) – 治疗的名字。
outcome (str) – 结果的名字。
- 返回
True如果给定的集合是对结果的治疗的因果效应的一个合理的前门调整集合。
- 返回类型
bool
- get_frontdoor_set(treatment, outcome, adjust='simple')
返回用于调整治疗和结果之间因果效应的前门集合。
- 参数
treatment (set of str or str) – 治疗的名字。应该只包含一个元素。
outcome (set of str or str) – 结果的名字。应该只包含一个元素。
adjust (str, default='simple') –
可选的选项包括 ‘simple’: 返回有最少数量元素的前门集合。
’minimal’: 返回有最少数量元素的前门集合。
’all’: 返回所有可能的前门集合。
- 返回
2个元素(adjustment_set, Prob)
- 返回类型
元组
- 抛出
IdentificationError – Raise error如果样式不在simple,minimal或者all或者没有集合能满足前门准则。
- get_iv(treatment, outcome)
为结果的治疗的因果效应找到工具变量。
- 参数
treatment (iterable) – 治疗的名字(们)。
outcome (iterable) – 结果的名字(们)。
- 返回
一个合理的工具变量集合将会是空的如果没有这样的集合。
- 返回类型
set
- is_valid_iv(treatment, outcome, set_)
决定给出的集合是否是一个合法的工具变量集合。
- 参数
treatment (iterable) – 治疗的名字(们)。
outcome (iterable) – 结果的名字(们)。
set (set) – 等待测试的集合。
- 返回
True如果集合是一个合理的工具变量集合否则False。
- 返回类型
bool
- identify(treatment, outcome, identify_method='auto')
识别因果效应表达式。识别是转变任何因果效应量的操作。比如,用do operator的量,变为对应的统计量这样它就可以用给出的数据估计因果效应。但是, 注意不是所有的因果量都是可识别的,这种情况下,一个IdentificationError被抛出。
- 参数
treatment (set or list of str) – 治疗名字的集合。
outcome (set or list of str) – 结果名字的集合。
identify_method (tuple of str or str, optional, default='auto') –
如果传入的值是元组或者列表,那么它应该有两个元素, 其中第一个是识别方法,第二个是返回的集合样式。
可选的选项:
’auto’ : 使用所有可能的方法进行识别
’general’: 通用识别方法,看id()
(‘backdoor’, ‘simple’): 返回治疗和结果的所有的直接的混淆因素的集合作为后门调整集合。
(‘backdoor’, ‘minimal’): 返回所有的可能的有最小数量元素的后门调整集合。
(‘backdoor’, ‘all’): 返回所有的可能的后门调整集合。
(‘frontdoor’, ‘simple’): 返回所有的可能的有最小数量元素的前门调整集合。
(‘frontdoor’, ‘minimal’): 返回所有的可能的有最小数量元素的前门调整集合。
(‘frontdoor’, ‘all’): 返回所有的可能的前门调整集合。
- 返回
一个python字典,其中字典中的键是识别方法,值是对应的结果。
- 返回类型
dict
- 抛出
IdentificationError – 如果因果效应不可识别或者identify_method给的不正确。
- estimate(estimator_model, data=None, *, treatment=None, outcome=None, adjustment=None, covariate=None, quantity=None, **kwargs)
估计新的数据集中识别的因果效应。
- 参数
estimator_model (EstimatorModel) – 任何在EstimatorModel中实现的合适的估计器模型可以在这里使用。
data (pandas.DataFrame, optional, default=None) – 用于估计的因果效应的数据集。如果是None,使用用于因果图发现的数据。
treatment (set or list, optional, default=None) – 治疗的名字们。如果是None,用于后门调整的治疗被当作治疗。
outcome (set or list, optional, default=None) – 结果的名字们。如果是None,用于后门调整的结果被当作结果。
adjustment (set or list, optional, default=None) – 调整集合的名字们。如果是None,调整集合由CausalModel找到的最简单的后门集合给出。
covariate (set or list, optional, default=None) – 协变量集合的名字。如果是None则忽略。
quantity (str, optional, default=None) – 估计因果效应时,感兴趣的量。
- 返回
估计的数据中的因果效应。
- 返回类型
np.ndarray or float
- identify_estimate(data, outcome, treatment, estimator_model=None, quantity=None, identify_method='auto', **kwargs)
组合识别方法和估计方法。然而,既然现在实现的估计器模型自动假设(有条件地)无混淆(除了有关iv的方法)。我们可能仅考虑使用后门集合调整来实现无混淆条件。
- 参数
treatment (set or list of str, optional) – 治疗的名字们。
outcome (set or list of str, optional) – 结果的名字们。
identify_method (tuple of str or str, optional, default='auto') –
如果传入的值是元组或者列表,那么它应该有两个元素, 其中第一个是识别方法,第二个是返回的集合样式。
可选的选项:
’auto’ : 使用所有可能的方法进行识别
’general’: 通用识别方法,看id()
(‘backdoor’, ‘simple’): 返回治疗和结果的所有的直接的混淆因素的集合作为后门调整集合。
(‘backdoor’, ‘minimal’): 返回所有的可能的有最小数量元素的后门调整集合。
(‘backdoor’, ‘all’): 返回所有的可能的后门调整集合。
(‘frontdoor’, ‘simple’): 返回所有的可能的有最小数量元素的前门调整集合。
(‘frontdoor’, ‘minimal’): 返回所有的可能的有最小数量元素的前门调整集合。
(‘frontdoor’, ‘all’): 返回所有的可能的前门调整集合。
quantity (str, optional, default=None) – 估计因果效应时,感兴趣的量。
- 返回
估计的数据中的因果效应。
- 返回类型
np.ndarray or float