freeCodeCamp.org

Product Experimentation with Doubly Robust Estimation: When Both Your Models Are Wrong in LLM Applications

8.5内容质量
Product Experimentation with Doubly Robust Estimation: When Both Your Models Are Wrong in LLM Applications

TL;DR · AI 摘要

双重稳健估计(AIPW)通过结合倾向得分模型和结果模型,在LLM产品实验中实现因果推断的鲁棒性,即使单个模型错误也能保证估计有效性。

核心要点

  • AIPW估计器在倾向得分或结果模型任一正确时均保持一致性
  • 通过故意破坏模型验证双重稳健性,需同时失效两个模型才会失效
  • scikit-learn实现包含5步:建模→组合→置信区间→验证

结构提纲

按章节快速跳转。

  1. 传统模型依赖单点失效,AIPW通过双重稳健性提升鲁棒性

  2. AIPW结合倾向得分与结果模型,数学保证半参数效率

  3. 包含5步流程:模型拟合→组合→置信区间→验证

  4. 通过故意破坏模型验证双重稳健性,需同时失效两个模型才会失效

思维导图

用一张图看清主题之间的关系。

查看大纲文本(无障碍 / 无 JS 友好)
  • 双重稳健估计在LLM实验
    • 核心原理
      • 半参数效率理论保证
      • 容错机制
    • 实现流程
      • 5步实施法
      • 验证场景

金句 / Highlights

值得收藏与分享的关键句。

#因果推断#双重稳健估计#LLM实验#scikit-learn
打开原文

使用双重稳健估计进行产品实验:当LLM应用中的两个模型都出错时

2026年8月11日

/

#experimentation

Rudrendu Paul

您的AI产品六个月前上线了代理模式的自愿参与功能。您进行了倾向性分析,调整了参与层级和查询置信度,报告了任务完成率提升8个百分点的显著效果。这个数字进入了季度业务回顾,所有人都感到满意。

不可避免地,严谨的数据科学家会提出令人不安的问题。您有多大把握确定倾向模型捕捉了所有混杂因素?如果遗漏了某些因素,逻辑回归是否在估计错误的选择概率?如果任务完成率与查询置信度存在非线性关系,而线性模型无法捕捉这种关系,结果回归是否也会出现模型设定错误?

您有两个模型,不确定哪个正确,但两者都至关重要。

自愿参与的AI产品默认会遇到这种困境。在没有随机化的LLM实验中进行因果推断时,您会同时拥有参与用户和未参与用户的结果数据。

问题在于这些群体是自我选择的。您构建的每个模型都是对未知真相的近似。

如果倾向模型错误,单纯依赖倾向加权会失效;如果结果模型错误,单纯依赖回归调整也会失效。每种方法都押注于单一模型的正确设定。

双重稳健估计,特别是增强逆概率加权(AIPW)估计器,采取了不同的策略。它将倾向模型和结果模型结合为一个估计器,只要其中一个模型正确设定,该估计器就能保持一致性。只有当两个模型同时失效时,AIPW才会失效。

这种保证源于估计器背后的半参数效率理论,这是其构造中固有的数学特性。可以将其视为因果估计的冗余工程。它依赖于与分布式系统在单个节点故障时保持在线相同的容错逻辑。

在本教程中,您将使用scikit-learn从零开始实现AIPW,添加自助法置信区间,并通过逐一故意破坏模型来证明双重稳健特性,展示估计器的稳健性。对于在运行嘈杂AI产品实验的数据科学家来说,这个框架能让您的估计具备生存能力。

本教程中所有代码块均可在配套笔记本中端到端运行:github.com/RudrenduPaul/product-experimentation-causal-inference-genai-llm/tree/main/12_doubly_robust/ 。笔记本文件为aipw_demo.ipynb。

目录

  • 为什么两个模型都无法赢得您的信任
  • 双重稳健估计实际上做了什么
  • 先决条件
  • 构建工作示例的步骤 第1步:拟合倾向模型 第2步:拟合结果模型 第3步:组合成AIPW估计器 第4步:自助法置信区间 第5步:通过故意设定错误证明双重稳健特性 场景1:错误的倾向模型,正确的结果模型 场景2:错误的结果模型,正确的倾向模型
  • 双重稳健估计失效的情形
  • 战略性实现

为什么两个模型都无法赢得您的信任

倾向得分方法要成功只依赖一个条件:建立一个能准确捕捉所有混杂因素的倾向模型。回归调整方法要成功同样只依赖一个条件:建立一个能准确描述协变量与结果之间关系的结果模型。在实际应用中,这两个条件都很难满足,而且你几乎无法确定是否达到了其中任何一个条件。

在LLM自愿加入分析中,倾向模型会以三种特定方式失效。首先,事件日志中的特征本身是自愿加入决策的下游结果。用户查询置信度得分反映了模型在接收到查询时的评估结果。而用户自愿加入的潜在动机则完全超出了你的测量系统范围。

其次,逻辑回归无法自动捕捉非线性交互作用。如果企业计划中的高使用率用户与个人计划中的高使用率用户自愿加入的比率存在显著差异,主效应逻辑模型将完全忽略这种细微差别。第三,未测量的混杂因素本质上是不可见的。如果经常阅读你工程博客的高活跃用户自愿加入率远高于未阅读的同类用户,而你又缺乏博客阅读量这一信号,无论你如何调整倾向模型,都会为这些用户分配错误的权重。

结果模型失效的原因则有所不同。LLM系统中的任务完成情况取决于查询复杂度,而这一指标以众所周知的噪声著称。它还取决于模型版本,而你可能未将其作为协变量进行捕捉。此外,它还取决于用户是否处于带有自定义系统提示的企业工作区(这一因素可能根本不在你的日志中)。对这些协变量进行线性回归时,总会在某个地方错误地设定函数形式,且偏差方向无法预测。

实际问题是,你无法运行一个能明确确认任一模型正确的规范性检验。平衡诊断只能揭示倾向模型的质量,其作用仅限于此。它们无法检测未测量的混杂因素。残差图可以确认你的结果模型拟合观测数据的效果,但无法揭示协变量遗漏的内容。你可能改进了这两个模型,但仍无法确定是否解决了根本问题。这听起来简单,实际上却异常困难。

任何基于倾向得分的因果分析都建立在三个识别假设之上。在AIPW或其他任何估计方法能给出有效因果效应之前,这三个假设必须全部成立。

  • 无混杂性(又称强可忽略性):所有同时影响自愿加入概率和任务完成的变量都已被测量并包含在模型中。
  • 重叠(正性):每个用户在处理组或对照组中都必须有非零的概率。没有任何子群体能完全确定会或不会自愿加入。
  • SUTVA(处理无干扰性):每个用户的潜在结果不受其他用户处理状态的影响,且处理方式只有一种版本。AIPW放宽了要求模型必须正确捕捉这些假设的条件,但它无法消除这些假设本身。这些假设仍需在数据中成立,任何方法论上的巧妙设计都无法改变这一事实。

双重稳健估计实际上做了什么

AIPW提供了一种数学保证,这是任何单模型方法都无法实现的。只要倾向模型或结果模型中有一个被正确设定,估计值就能保持一致性。只要至少有一个模型成立,估计方法就能成功。只有当两个模型同时失效时,估计方法才会失效。

AIPW 估计器针对所有具有重叠倾向得分的用户,估计平均处理效应(ATE)。这与倾向匹配所针对的处理组平均处理效应(ATT)有本质区别。将倾向得分截断到 [0.01, 0.99] 范围会将有效人群限制在重叠充分的用户群体中,但估计目标仍然是该特定重叠区域内的 ATE。

公式如下:

code
ATE_AIPW = mean( m1(X) - m0(X)  +  T*(Y - m1(X)) / e(X)  -  (1-T)*(Y - m0(X)) / (1 - e(X)) )

其中:

  • e(X) 是倾向得分:给定协变量的入组预测概率
  • m1(X) 是处理组(已入组)下的预测结果
  • m0(X) 是对照组(未入组)下的预测结果
  • T 是处理指示变量(1 = 已入组,0 = 未入组)
  • Y 是观测结果

本教程中所有小数值均表示比例。估计值 0.08 等于任务完成率的 8 个百分点。

该表达式包含两个主要部分。第一部分 m1(X) - m0(X) 是纯回归调整:它直接对比两个预测结果。第二部分的 IPW 校正项则计算实际发生结果与回归预测值之间的加权残差。

如果结果模型完全正确,残差将为零,校正项也会消失。如果结果模型存在偏差,只要倾向模型正确,IPW 校正项仍能修正预测误差。

反向思考这一逻辑:如果倾向模型正确,IPW 校正项本身就能产生无偏估计,此时结果模型只需降低方差即可。任一模型单独有效即可保证估计器有效,只有当两个模型都失效时,估计器才会失效。

这一特性被称为双重稳健性,具有实际应用价值。当两个模型都正确且满足正则性条件时,AIPW 估计器在渐近情况下能达到半参数效率边界:它能从数据中提取与任何常规估计器在大样本中相同的统计信息。在实践中,这种效率提升意味着无需增加数据量即可获得更紧致的置信区间。

先决条件

你需要 Python 3.11 或更新版本,熟悉 pandas 和 scikit-learn,以及对回归和逆概率加权有基本理解。

安装本教程所需的包:

code
pip install numpy pandas scikit-learn scipy

预期输出:

code
Successfully installed numpy pandas scikit-learn scipy

这四个包是唯一依赖项。scikit-learn 提供了逻辑回归和线性回归模型。scipy 用于图表脚本中的核密度估计(KDE)。你不需要任何专门的因果推断库。AIPW 估计器的实现足够简单,可以直接从零构建。

克隆配套仓库以获取合成数据集:

code
git clone https://github.com/RudrenduPaul/product-experimentation-causal-inference-genai-llm.git
cd product-experimentation-causal-inference-genai-llm
python data/generate_data.py --seed 42 --n-users 50000 --out data/synthetic_llm_logs.csv
code
Generated 50000 users → data/synthetic_llm_logs.csv

数据生成器创建了 50,000 个合成用户,包含参与度层级、查询置信度评分、入组标志和任务完成结果。代理模式入组的真值因果效应为 +8 个百分点,已嵌入生成器中,方便你验证每个估计器是否能准确恢复该效应。

构建工作示例

该数据集模拟了一个SaaS产品,用户可以选择加入由更强大模型驱动的代理模式。五万名用户中,不同参与层级的加入率差异显著:高活跃用户加入率为65%,中等活跃用户为35%,低活跃用户仅为12%。

真实因果效应在任务完成率上为+8个百分点。由于选择偏差(AIPW专门设计用于纠正这种偏差),对已加入和未加入用户进行的简单对比高估了差异近三倍。

加载数据并计算粗略估计值:

code
import numpy as np
import pandas as pd

df = pd.read_csv("data/synthetic_llm_logs.csv")

T = df["opt_in_agent_mode"].values
Y = df["task_completed"].values

naive_ate = Y[T == 1].mean() - Y[T == 0].mean()
print(f"Naive ATE (unadjusted): {naive_ate:+.4f}")
print(f"N treated: {T.sum()}, N control: {(1-T).sum()}")
code
Naive ATE (unadjusted): +0.2106
N treated: 13451, N control: 36549

你加载数据集,提取二元处理指标(opt_in_agent_mode)和二元结果(task_completed),并计算处理组与对照组的平均结果差异。

粗略估计值达到+0.2106,超过21个百分点,这主要受到选择偏差的严重高估。高活跃用户比低活跃用户更频繁地加入代理模式,且无论使用哪种模型,他们始终更可能完成更多任务。

这种粗略差异主要反映了谁选择了加入。模型变更仅对观察到的差异贡献了很小一部分。

理论上这个保证听起来很简单,在数据中确实成立:当你运行第5步的模型误设检验时,你会看到+0.2106中有多少是选择噪声,多少是真实的处理效应。

第1步:拟合倾向模型

倾向得分是给定可观测特征时用户选择加入的概率。在这个数据集上,对参与层级和查询置信度进行逻辑回归是合适的起点。

code
from sklearn.linear_model import LogisticRegression

# 构建协变量矩阵
X_df = pd.get_dummies(
    df[["engagement_tier", "query_confidence"]],
    drop_first=True
).astype(float)
X = X_df.values

# 拟合倾向模型
ps_model = LogisticRegression(max_iter=1000, C=1.0)
ps_model.fit(X, T)

e_hat = ps_model.predict_proba(X)[:, 1]

# 为数值稳定性修剪极端倾向值
e_hat = np.clip(e_hat, 0.01, 0.99)

print(f"倾向得分范围: {e_hat.min():.3f} 到 {e_hat.max():.3f}")
print(f"倾向得分均值(处理组): {e_hat[T == 1].mean():.3f}")
print(f"倾向得分均值(对照组): {e_hat[T == 0].mean():.3f}")
code
倾向得分范围: 0.114 到 0.675
倾向得分均值(处理组): 0.401
倾向得分均值(对照组): 0.220

你对分类变量参与层级进行独热编码,将查询置信度保留为连续变量,并拟合逻辑回归模型预测加入事件。

predict_proba方法返回每个用户的类别1概率,这就是倾向得分。将值修剪到[0.01, 0.99]范围可防止在AIPW公式中出现除以零的错误(实际中clip操作正是为此设计)。

合理性检查确认了处理组(0.401)的平均倾向得分高于对照组(0.220),这与你预期的选择模式一致:重度用户在处理组中出现频率更高,模型也正确地为他们分配了更高的概率。

倾向得分范围0.114到0.675证实了重叠假设成立:没有用户的倾向得分接近0或1,因此每个用户都有意义地存在于任一组的概率。

在接触估计器之前先运行倾向得分范围检查。像0.114到0.675这样较窄的范围证实了重叠成立,而接近0或1的值则会标记出违反假设的情况。

第2步:拟合结果模型

结果模型分别对处理组和对照组的用户预测任务完成情况。

你训练两个独立的回归模型:第一个仅使用处理组用户数据进行训练,第二个仅使用对照组用户数据进行训练。然后使用这两个模型为数据集中的每个用户在每种假设处理分配下预测结果。

code
from sklearn.linear_model import LinearRegression

# 为处理组用户拟合结果模型
m1_model = LinearRegression()
m1_model.fit(X[T == 1], Y[T == 1])

# 为对照组用户拟合结果模型
m0_model = LinearRegression()
m0_model.fit(X[T == 0], Y[T == 0])

# 为所有用户预测反事实结果
m1_hat = m1_model.predict(X)   # 如果所有用户都接受处理的预测结果
m0_hat = m0_model.predict(X)   # 如果所有用户都处于对照组的预测结果

# 回归调整估计(仅使用结果模型,不使用倾向得分)
ate_regression = (m1_hat - m0_hat).mean()
print(f"回归调整ATE: {ate_regression:+.4f}")
code
回归调整ATE: +0.0847

你为处理组用户拟合了一个线性回归模型,以学习该组协变量与结果之间的关系,同时为对照组用户拟合了另一个独立的回归模型。然后你预测了在处理条件下(m1_hat)和对照条件下(m0_hat)所有用户的预期结果。

回归调整估计通过平均这些预测差异得出:结果为+0.0847,比原始估计值+0.2106更接近真实值+0.08。

回归调整在这里有效运作,使用结果模型为每个用户填补缺失的反事实数据。0.0847与真实值0.0800之间的差距反映了结果模型本身的局限性,这正是AIPW方法中倾向得分校正需要介入的地方。

第3步:组合成AIPW估计器

在获得倾向得分和两个结果预测后,你可以将它们组合到AIPW公式中:

code
from typing import Tuple

def calculate_aipw_ate(
    Y: np.ndarray,
    T: np.ndarray,
    e_hat: np.ndarray,
    m1_hat: np.ndarray,
    m0_hat: np.ndarray
) -> Tuple[float, np.ndarray]:
    """
    增广逆概率加权(AIPW)估计器。

    参数
    ----------
    Y       : 类数组,观测结果
    T       : 类数组,二元处理指标
    e_hat   : 类数组,估计的倾向得分 P(T=1|X)
    m1_hat  : 类数组,处理条件下的预测结果
    m0_hat  : 类数组,对照条件下的预测结果

    返回
    -------
    float : 估计的平均处理效应(ATE)
    """
    # 处理组的IPW校正
    ipw_treated = T * (Y - m1_hat) / e_hat

    # 对照组的IPW校正
    ipw_control = (1 - T) * (Y - m0_hat) / (1 - e_hat)

每个观测的AIPW影响函数

phi = (m1_hat - m0_hat) + ipw_treated - ipw_control

return phi.mean(), phi

ate_aipw, phi_obs = calculate_aipw_ate(Y, T, e_hat, m1_hat, m0_hat) print(f"AIPW ATE: {ate_aipw:+.4f}") print(f"Naive ATE: {naive_ate:+.4f}") print(f"Regression-only ATE: {ate_regression:+.4f}") print(f"Ground truth: +0.0800")

code

AIPW ATE: +0.0847 Naive ATE: +0.2106 Regression-only ATE: +0.0847 Ground truth: +0.0800

code

该函数为每个观测计算AIPW影响函数。第一项m1_hat - m0_hat是回归调整项。第二项T * (Y - m1_hat) / e_hat是对处理组用户的IPW校正项:它计算用户实际结果与模型预测值之间的残差,再通过逆倾向得分进行加权。由于那些看起来不太可能参与的用户在处理组中代表性不足,这些用户会获得较大的加权以进行补偿。第三项则对对照组用户进行对称校正。

对每个观测的影响值进行平均可得到AIPW估计值。在这个数据集上,该估计值为+0.0847,与纯回归估计值一致。当两个模型都正确指定时,这种情况是预期的:两个方法得出的结果都接近真实值+0.08,且都明显高于天真的估计值+0.2106。该函数还会返回phi_obs,即每个观测的影响值,这些值将在第5步的模型误设检验中使用。

## 第4步:引导法置信区间

没有置信区间的点估计是不完整的。最干净的生产级方法是非参数引导法:有放回地重采样数据,从头重新拟合所有模型,然后取重采样估计值分布的分位数。

def bootstrap_aipw_ci( df: pd.DataFrame, X_cols: list, treatment_col: str, outcome_col: str, n_bootstrap: int = 500, seed: int = 7 ) -> Tuple[np.ndarray, float, float]: """ 使用95%分位数置信区间的引导AIPW ATE估计。

对每个重采样数据从头重新拟合倾向模型、结果模型和AIPW估计。 """ rng = np.random.default_rng(seed) n = len(df) boot_estimates = []

X_all = pd.get_dummies(df[X_cols], drop_first=True).astype(float).values T_all = df[treatment_col].values Y_all = df[outcome_col].values

for _ in range(n_bootstrap):

有放回重采样

idx = rng.integers(0, n, size=n) X_b, T_b, Y_b = X_all[idx], T_all[idx], Y_all[idx]

重新拟合倾向模型

ps = LogisticRegression(max_iter=1000, C=1.0) ps.fit(X_b, T_b) e_b = np.clip(ps.predict_proba(X_b)[:, 1], 0.01, 0.99)

重新拟合结果模型

m1 = LinearRegression().fit(X_b[T_b == 1], Y_b[T_b == 1]) m0 = LinearRegression().fit(X_b[T_b == 0], Y_b[T_b == 0]) m1_b = m1.predict(X_b) m0_b = m0.predict(X_b)

对引导样本进行AIPW估计

ate_b, _ = calculate_aipw_ate(Y_b, T_b, e_b, m1_b, m0_b) boot_estimates.append(ate_b)

boot_estimates = np.array(boot_estimates) ci_low = np.percentile(boot_estimates, 2.5) ci_high = np.percentile(boot_estimates, 97.5)

return boot_estimates, ci_low, ci_high

code

boot_dist, ci_lo, ci_hi = bootstrap_aipw_ci( df, X_cols=["engagement_tier", "query_confidence"], treatment_col="opt_in_agent_mode", outcome_col="task_completed", n_bootstrap=500, seed=7, )

print(f"AIPW ATE: {ate_aipw:+.4f}") print(f"95% Bootstrap CI: [{ci_lo:+.4f}, {ci_hi:+.4f}]") print(f"Bootstrap std dev: {boot_dist.std():.4f}")

code

AIPW ATE: +0.0847 95% Bootstrap CI: [+0.0744, +0.0952] Bootstrap std dev: 0.0053

code

通过有放回地对50,000行数据进行500次重采样,生成引导样本。在每次重采样后,你都需要从头开始重新拟合倾向模型,从头开始重新拟合两个结果模型,并在新数据上计算AIPW估计值。

在每次重采样上重新拟合所有模型非常重要:如果你只对固定模型的残差进行重采样,会低估由于模型估计误差导致的变异性。

95%置信区间为[+0.0744, +0.0952],该区间舒适地包含了真实值+0.0800,并且与天真的+0.2106估计值有显著差距。引导标准差为0.0053,因此你估计值的典型抽样变异约为半个百分点。

## 第5步:通过故意误指定验证双重稳健属性

双重稳健属性在实践中成立。你可以通过在自己的数据集上逐一故意误指定一个模型,并观察AIPW是否保持稳健,来实证验证这一属性。

### 情景1:错误的倾向模型,正确的结果模型

将所有用户的估计倾向得分替换为固定值0.3。无论用户的参与层级或查询置信度如何,每个用户获得相同的权重,这构成了故意设计的最大误指定倾向模型。单独使用IPW应该失效,但AIPW应该不受影响,因为结果模型是正确指定的。

情景1:固定倾向(所有用户e=0.3)

e_wrong = np.full(len(df), 0.3)

使用错误倾向的IPW

t_mask = T == 1 c_mask = T == 0 ate_ipw_wrong = ( (Y[t_mask] / e_wrong[t_mask]).sum() / (1 / e_wrong[t_mask]).sum()

  • (Y[c_mask] / (1 - e_wrong[c_mask])).sum() / (1 / (1 - e_wrong[c_mask])).sum()

)

使用错误倾向但正确结果模型的AIPW

ate_aipw_wrong_ps, _ = calculate_aipw_ate(Y, T, e_wrong, m1_hat, m0_hat)

print("=== 情景1:固定倾向(e = 0.3) ===") print(f"使用错误倾向的IPW: {ate_ipw_wrong:+.4f} (应该错误)") print(f"回归调整(保持不变): {ate_regression:+.4f} (应该≈0.085)") print(f"使用错误倾向的AIPW: {ate_aipw_wrong_ps:+.4f} (应该保持≈0.085)") print(f"真实值: +0.0800")

code

=== 情景1:固定倾向(e = 0.3) === 使用错误倾向的IPW: +0.2106 (应该错误) 回归调整(保持不变): +0.0847 (应该≈0.085) 使用错误倾向的AIPW: +0.0847 (应该保持≈0.085) 真实值: +0.0800

code

你将所有用户的平坦倾向替换为0.3,并计算两项内容。首先,仅使用错误倾向的纯IPW:由于它对所有用户进行平等加权,不考虑参与层级,因此产生天真的+0.2106估计,未能修正选择模式。

第二,使用错误的倾向得分但保持正确拟合的结果模型进行AIPW:估计值仍为+0.0847。结果模型项推动了估计过程,而IPW校正项在样本中产生的噪声相互抵消。其中一条臂失效,另一条臂则支撑了估计器的计算。

### 情景2:错误的结果模型,正确的倾向得分模型

现在保持正确估计的倾向得分,但将两个结果模型替换为常数。为所有用户设置m1_hat = m0_hat = 0.5,即对所有人预测50%任务完成率的无信息预测。单独使用回归调整应退化为零。由于倾向得分模型正确指定,AIPW应不受影响。

情景2:常数结果模型(对所有人m1 = m0 = 0.5)

m1_wrong = np.full(len(df), 0.5) m0_wrong = np.full(len(df), 0.5)

使用错误结果模型的回归调整

ate_regression_wrong = (m1_wrong - m0_wrong).mean()

使用正确倾向得分的纯IPW(用于对比)

ate_ipw_correct = ( (Y[t_mask] / e_hat[t_mask]).sum() / (1 / e_hat[t_mask]).sum()

  • (Y[c_mask] / (1 - e_hat[c_mask])).sum() / (1 / (1 - e_hat[c_mask])).sum()

)

使用正确倾向得分但错误结果模型的AIPW

ate_aipw_wrong_out, _ = calculate_aipw_ate(Y, T, e_hat, m1_wrong, m0_wrong)

print("=== 情景2:常数结果模型(m1 = m0 = 0.5) ===") print(f"使用错误结果模型的回归:{ate_regression_wrong:+.4f} (应为0.0)") print(f"使用正确倾向得分的IPW: {ate_ipw_correct:+.4f} (应为~0.085)") print(f"使用错误结果模型的AIPW: {ate_aipw_wrong_out:+.4f} (应保持~0.085)") print(f"真实值: +0.0800")

code

=== 情景2:常数结果模型(m1 = m0 = 0.5) === 使用错误结果模型的回归:+0.0000 (应为0.0) 使用正确倾向得分的IPW: +0.0851 (应为~0.085) 使用错误结果模型的AIPW: +0.0849 (应保持~0.085) 真实值: +0.0800

code

当将结果模型设置为0.5时,回归调整项m1_hat - m0_hat完全退化为零,得到一个毫无意义的估计值。使用正确指定的倾向得分模型的纯IPW方法独立恢复出+0.0851的估计值。

使用错误结果模型但正确倾向得分的AIPW方法也恢复出+0.0849,因为此时IPW校正项承担了全部权重:残差Y - 0.5通过逆倾向得分正确加权后,平均值收敛到正确答案。错误的结果模型仅增加了方差,估计器仍保持一致性。

运行这两个情景可为任何内部分析文档提供合理性检验。它将双重稳健性从理论属性转化为可向质疑者展示的具体数值。

## 双重稳健估计失效的情形

AIPW为模型误设提供了一层防护,但在展示结果前需要明确其实际限制。

### 两个模型同时误设

双重稳健保证覆盖了至少一个模型正确的场景。如果倾向得分模型遗漏了关键混杂因素,同时结果模型也未能捕捉真实函数形式,AIPW将继承两个模型中"相对更正确"模型的偏差。

令人不安的现实是:AIPW会原封不动地传递未测量的混杂因素。它允许你犯一次错误。这个限制非常明确——恰好一次。

由于部分用户的倾向值接近0或1,AIPW公式中的IPW校正项会导致数值爆炸。一个e_hat=0.02的用户会产生Y/0.02=50*Y的校正项,如果该用户的实际结果异常,这个校正项可能会完全主导整个估计结果。

将倾向值截断到[0.01, 0.99](如本文所做)能提供有限的保护。更彻底的解决方案是进行倾向值修剪(移除极端值用户),但这样会改变估计目标:此时你估计的是重叠区域的ATE,这个人群范围比完整数据集更窄。请明确记录这个选择。

### 小样本方差超过渐近理论预测

AIPW在大样本中能达到半参数效率边界。当样本量为500或1000时,IPW校正项带来的方差膨胀可能非常显著,自助法置信区间也会变得很宽。

在极小样本实验中,朴素回归调整可能产生更窄的置信区间,即使其对模型误设的理论保护较弱。AIPW的效率优势是大样本性质。

### 两个组件的模型选择仍需人工判断

逻辑回归是合理的默认选择,但如果真实选择机制包含高阶交互作用,主效应模型无法表示,倾向模型会以平衡诊断无法捕捉的方式系统性出错。

使用更灵活的模型(梯度提升、随机森林)处理干扰项组件在大样本中能提升性能,但需要交叉拟合:在保留验证集上拟合倾向模型和结果模型后再进行预测,防止训练误差泄露到AIPW计算中导致最终估计偏倚。交叉拟合是目标最大似然估计(TMLE)的实现基础。

## 战略实施

本教程的从零实现展示了原理机制。你的生产环境需要两个该版本缺少的关键要素:使用灵活模型时的交叉拟合以防止过拟合偏倚,以及能适应数据信号的数据自适应干扰模型。没有交叉拟合,本教程的从零实现无法支撑严肃的观察性研究。

Python实现的TMLE可在专业因果推断库中找到,每个库都在AIPW原理基础上增加了双重功能:TMLE直接针对感兴趣的估计目标,使用机器学习模型拟合倾向和结果组件时修正正则化偏倚,并在干扰模型从分析数据中估计时仍能生成有效置信区间。

Lyft工程团队发表了关于其网约车因果推断双重稳健流水线的详细说明,构建生产级系统前值得阅读(Nassiri & Chu, Lyft Engineering, 2026)。

理论背景方面,AIPW的保证可追溯至Robins、Rotnitzky和Zhao(Robins等,1994),这很重要,因为它明确告诉你该方法的保证边界在哪里,以及你的建模判断从何处开始。

与本文内容最契合的实践指南是Mark van der Laan在加州大学伯克利分校开发的目标学习框架(van der Laan & Rose, 2011)。

配套的 Jupyter Notebook 位于 github.com/RudrenduPaul/product-experimentation-causal-inference-genai-llm/tree/main/12_doubly_robust。克隆该仓库,生成合成数据集,然后打开 aipw_demo.ipynb 文件,即可复现本教程中的所有代码块,包括模型误设场景。

在实际生产环境的观察性分析中,您会遇到两种近似方案,需要选择更优的方案。请使用第5步中的模型误设检验方法对自有数据进行分析:倾向得分诊断将告诉您倾向得分模型的权重占比,而结果模型的残差分布将显示回归调整部分的贡献程度。

AIPW 方法有效的原因在于其专门设计用于这种场景:当两个模型都未经验证且同时发挥作用时。如果其中一个模型有效,估计器也会保持有效性。

我是应用人工智能/机器学习和营销度量科学领域的领导者,拥有15年以上在财富500强企业构建和扩展应用AI及机器学习产品的经验。我专注于因果推断、实验设计、智能体AI系统以及企业级AI战略,应用领域涵盖零售媒体网络(RMN)、广告、广告技术、营销技术、消费品(CPG)和电子商务。我是Springer Nature、Elsevier、ICML(顶级AI/ML会议)和IEEE的出版作者,也是多家领先AI/ML和分析出版物的常驻撰稿人(个人观点)

如果本文对您有帮助,请分享给他人。

免费学习编程。freeCodeCamp 的开源课程已帮助超过40,000人成为开发者。立即开始学习

ADVERTISEMENT