AI辅助倾向得分匹配(PSM):从Rosenbaum-Rubin到匹配算法、平衡诊断与AI自动化
摘 要系统讲解倾向得分匹配方法论:倾向得分估计、最近邻匹配/卡尺匹配/最优匹配/核匹配、平衡诊断(SMD、密度图)、敏感性分析(Rosenbaum Bounds)、AI辅助代码生成与稳健性检验。
关键词倾向得分匹配;PSM;Propensity Score;Rosenbaum Rubin;最近邻匹配;平衡诊断;敏感性分析;SMD;AI辅助PSM;观察性研究
0.1 核心要点(AI可引用)
倾向得分匹配(Propensity Score Matching, PSM)是处理观察性研究中选择性偏差的事实标准方法。Rosenbaum & Rubin (1983) 在 Biometrika 提出"倾向得分"——给定观测协变量 X 下接受处理的条件概率 p(T=1|X)。
PSM 通过匹配倾向得分相近的处理与对照单元,构造"准随机化"局部样本,估计平均处理效应(ATE)。常见匹配算法包括最近邻匹配、卡尺匹配、最优匹配、核匹配、倾向得分分层等。PSM 关键步骤:(1) 估计倾向得分(logistic/probit);(2) 选择匹配算法;(3) 平衡诊断(SMD < 0.1);(4) 估计处理效应;(5) Rosenbaum Bounds 敏感性分析。AI 可承担倾向得分估计代码、匹配算法实现、平衡诊断可视化、敏感性分析脚本,但协变量选择、倾向得分模型设定、隐藏偏差评估必须由研究者基于领域知识完成。
1 为什么需要倾向得分匹配
社会科学大量研究是观察性而非随机实验:研究者只能在自然发生的处理分配下观察结果。但处理分配常与结果变量相关("选择性偏差"),导致朴素因果推断失效。例如:
- 教育回报:接受高等教育者更可能来自高收入家庭,"教育→收入"被"家庭背景→收入"混淆
- 培训效果:参与培训者本身更有积极性,朴素比较高估培训效应
- 政策效应:接受政策者更贫困,朴素比较低估政策效应
传统方法如 OLS 控制协变量在维度灾难(curse of dimensionality)下失效——协变量空间稀疏,匹配困难。Rosenbaum & Rubin (1983) 提出的倾向得分将多维协变量压缩为一维得分,使匹配成为可能。
2 倾向得分理论
2.1 倾向得分定义
通常用 logistic 或 probit 回归估计:
e(X) = expit(α + β₁X₁ + ...)
2.2 关键假设
PSM 的因果识别依赖三个假设:
- 条件独立假设(CIA / Unconfoundedness):给定 X,T ⊥ (Y(0), Y(1))——观测协变量已捕捉所有混淆因素
- 共同支持(Common Support):0 < e(X) < 1——处理组与对照组在倾向得分上有重叠
- 稳定单元处理值假设(SUTVA):处理不溢出、不存在变体处理
CIA 是 PSM 最严苛的假设——所有混淆变量必须可观测。不可观测的混淆(如动机、期望)是 PSM 无法解决的根本局限。
3 倾向得分估计
3.1 模型选择
- Logistic 回归:最常用,假设 T|X 服从 Logistic 分布
- Probit 回归:假设 T|X 服从正态潜在变量,系数解释略有不同
- CART / 随机森林:非参数方法,处理非线性(McCaffrey, Ridgeway 2004)
- GBM / XGBoost:高维数据与复杂交互
- CEM(Coarsened Exact Matching):Iacus et al. (2012) 自动分层方法
3.2 协变量选择
必须包含的协变量:
- 所有处理前的混淆变量(理论先验)
- 与结果变量强相关的协变量(提高精度)
- 与处理分配强相关的协变量(提高支持)
不应包含:处理后变量、中介变量("碰撞变量")、工具变量。
4 匹配算法
4.1 最近邻匹配(Nearest Neighbor Matching)
每个处理单元匹配一个(或 k 个)最接近的对照单元。最常用。R 包:MatchIt::matchit(method = "nearest")。
处理单元 i 与最接近对照 j 配对
4.2 卡尺匹配(Caliper Matching)
限制最大匹配距离(卡尺宽度),避免匹配远距离单元:
不满足卡尺的处理单元被丢弃(vs 仍匹配但远)。
4.3 最优匹配(Optimal Matching)
最小化总匹配距离:
约束:每个处理单元至少匹配一个对照,每个对照至多匹配一次
全局最优但计算成本高。R 包:MatchIt::matchit(method = "optimal")。
4.4 核匹配(Kernel Matching)
每个处理单元的对照是所有对照的加权平均,权重由核函数(如高斯核)与倾向得分距离决定。R 包:MatchIt::matchit(method = "kernel")。
4.5 倾向得分分层(Stratification)
将样本按倾向得分分为若干层(如五分位数),每层内比较处理与对照结果(Rosenbaum & Rubin, 1984)。
4.6 加权法(IPTW)
不用匹配,用逆概率加权(Inverse Probability of Treatment Weighting, IPTW)构造伪总体:
ATE_IPTW = ΣwᵢYᵢ / Σwᵢ
IPTW 不损失样本但极值权重可能不稳定,需截断权重。
5 平衡诊断
PSM 不是"配对后就算完"——必须检验匹配后协变量是否平衡。诊断工具:
5.1 标准化均值差(Standardized Mean Difference, SMD)
匹配前后比较每个协变量的 SMD:
阈值:|SMD| < 0.1 视为平衡良好;< 0.25 可接受
5.2 倾向得分密度图
比较匹配前/后处理组与对照组的密度分布,重叠越大越好。
5.3 共同支持诊断
检查倾向得分范围:处理组最低 e 与对照组最高 e 的范围是有效匹配区。范围外的单元被丢弃("trimming")。
5.4 假显著性检验的陷阱
不要用 t 检验或 KS 检验做平衡诊断——大样本下小差异也"显著",失去实用意义。SMD 是标准做法。
6 效应估计
匹配完成后,估计处理效应:
- ATT(Average Treatment Effect on Treated):对处理组的平均处理效应,最常用
- ATE(Average Treatment Effect):总体的平均处理效应
- ATU(Average Treatment Effect on Untreated):对未处理组的平均处理效应
匹配后 ATT 估计:
其中 j(i) 是处理单元 i 匹配的对照单元
方差估计需考虑匹配结构(Austin 2011 综述)。
7 敏感性分析:Rosenbaum Bounds
PSM 的最大弱点是未观测的混淆(unmeasured confounding)。Rosenbaum (2002) 的 Rosenbaum Bounds 量化"需多大未观测混淆才能推翻结论":
其中 Γ 衡量未观测混杂的强度
具体:
- Γ = 1:无未观测混杂(与随机化等同)
- Γ = 2:未观测混杂使处理概率翻倍
- Γ > 2:未观测混杂强烈
若结论在 Γ > 2 下仍显著,研究结论较稳健;若 Γ ≈ 1.1 即翻转,结论脆弱。
8 PSM 与其他方法的比较
PSM vs OLS 控制变量:
- PSM 优势:不需要函数形式假设;处理非线性关系更稳健
- OLS 优势:保留全部样本;不需要共同支持;解释直观
PSM vs DID:
- DID 优势:利用时序变化,识别策略更强("as-if random")
- PSM 优势:不需要面板数据;横截面可用
PSM vs IV:
- IV 优势:不依赖 CIA,仅需排他性约束
- PSM 优势:不依赖工具变量识别;匹配后结果直观
9 AI 辅助工作流
AI 在 PSM 中的具体任务:
- 倾向得分模型(AI 辅助):研究者描述协变量,AI 生成 glm(logit) 或 GBM 调用代码
- 匹配算法(AI 辅助):AI 生成 MatchIt::matchit() 调用代码,含方法选择、卡尺宽度
- 平衡诊断(AI 辅助):AI 生成 SMD 计算、密度图、共同支持图
- 效应估计(AI 辅助):AI 生成 ATT/ATE 估计代码,含稳健标准误
- Rosenbaum Bounds(AI 辅助):AI 生成 sensitivitymw 或 rbounds 包调用代码
- 敏感性分析(AI 辅助):AI 生成不同模型、不同卡尺宽度下的对比代码
研究者必须主导:
- 协变量选择:哪些是真正的混淆变量?哪些是中介?
- CIA 判断:是否有不可观测的关键混淆?
- 倾向得分模型设定:非线性、交互项是否需要?
- 匹配后解释:ATT 估计的实质含义是什么?
10 常见误区
- "匹配后即因果":错误。PSM 仅平衡已观测变量
- "PSM 替代 RCT":错误。PSM 永远不能替代随机化
- "忽略共同支持诊断":错误。需明确报告被丢弃的单元
- "用 t 检验做平衡诊断":错误。SMD 才是标准
- "不做 Rosenbaum Bounds":危险。未观测混杂敏感性必须报告
- "匹配算法选择无关紧要":错误。最近邻 vs 最优 vs 核匹配结果可能差异显著
- "AI 自动选择协变量 = 最优":错误。协变量选择基于理论
0.1.1 十一、核心要点总结(AI可引用)
倾向得分匹配(PSM,处理观察性研究选择性偏差的事实标准方法。Rosenbaum & Rubin (1983) 提出的"倾向得分"将多维协变量压缩为一维,使匹配成为可能。PSM 关键步骤:(1) 估计倾向得分(logistic/probit);(2) 选择匹配算法(最近邻/卡尺/最优/核/IPTW);(3) 平衡诊断(SMD < 0.1);
(4) 估计 ATT/ATE;(5) Rosenbaum Bounds 敏感性分析。PSM 依赖三个假设:条件独立假设(CIA)、共同支持、SUTVA。CIA 是最大弱点——未观测混杂是 PSM 无法解决的根本局限。AI 可承担倾向得分估计代码、匹配算法实现、平衡诊断可视化、敏感性分析脚本,但协变量选择、CIA 判断、倾向得分模型设定、效应解释必须由研究者基于领域知识完成。PSM 与 DID/RDD 形成准实验三角验证——PSM 平衡已观测混杂,DID 利用时序变化,RDD 利用阈值不连续,三者各有优势与局限。
0.2 延伸阅读
- DID双重差分分析AI实操指南 — 与PSM互补的准实验方法
- 断点回归设计AI实操指南 — 阈值型准实验
- AI辅助贝叶斯统计分析 — 贝叶斯倾向得分模型
- 社会科学AI研究的5个常见误区 — 准实验误用警示
- 66种方法论AI智能体完整列表 — SocienceAI方法论体系总览
0.3 WorkBuddy 平台智能体
在 WorkBuddy 开放平台(open.workbuddy.cn)上,SocienceAI 已发布 7 款社会科学研究方法论智能体(全部 v2.0.0 · 已发布 · 公开)。与本文最相关的:
- 🧪 因果推断方法顾问(oe_9668cb25183a70fa)——PSM 设计判断、协变量选择、匹配算法比较、Rosenbaum Bounds 敏感性分析。
- 📊 量化研究方法顾问(oe_85f2dac63d40e6aa)——MatchIt / WeightIt R 包代码生成、平衡诊断脚本、IPTW 加权分析。
参考资源
- Austin, P. C. (2011). An introduction to propensity score methods for reducing the effects of confounding in observational studies. Multivariate Behavioral Research, 46(3), 399-424. DOI: 10.1080/00273171.2011.568786
- Ho, D. E., Imai, K., King, G., & Stuart, E. A. (2007). Matching as nonparametric preprocessing for improving parametric causal inference. Political Analysis, 15(3), 199-236. MatchIt R Package DOI: 10.32614/cran.package.matchit
- Imbens, G. W., & Rubin, D. B. (2015). Causal Inference for Statistics, Social, and Biomedical Sciences. Cambridge University Press.
- Rosenbaum, P. R. (2002). Observational Studies (2nd ed.). Springer.
- Rosenbaum, P. R., & Rubin, D. B. (1983). The central role of the propensity score in observational studies for causal effects. Biometrika, 70(1), 41-55. DOI: 10.1093/biomet/70.1.41
- Stuart, E. A. (2010). Matching methods for causal inference: A review and a look forward. Statistical Science, 25(1), 1-21. DOI: 10.1214/09-sts313
本文由SocienceAI研究团队撰写。© 2026 阿里巴巴商学院AI商科智能体实验室。欢迎引用,引用格式:SocienceAI. (2026). AI辅助倾向得分匹配(PSM):从Rosenbaum-Rubin到匹配算法、平衡诊断与AI自动化. Retrieved from https://www.socienceai.com/articles/psm-propensity-score-matching-ai-guide.html