2. PSI 数学原理:从公式推导到分箱策略

群体稳定性指标(Population Stability Index, PSI)是 AML 模型中衡量模型评分分布随时间变化的核心量化工具。其数学本质是 两个概率分布之间的相对熵(Kullback-Leibler Divergence)的对称化变体。理解 PSI 的推导过程、期望分布与实际分布的关系以及分箱策略,是后续阈值调优的理论基础。

2.1 PSI 计算公式推导

PSI 的计算建立在将连续评分划分为 k 个离散区间(分箱)的基础上。设:

  • 实际分布(Actual Distribution):当前监控周期内,落在第 i 个分箱的样本占比,记为 Ai。
  • 期望分布(Expected Distribution):模型开发或基准周期内,落在第 i 个分箱的样本占比,记为 Ei。

PSI 的完整计算公式为:

PSI = Σ (Aᵢ - Eᵢ) × ln(Aᵢ / Eᵢ)   (i = 1, 2, ..., k)

该公式可拆解为两个步骤:

  1. 计算每个分箱的贡献值:(Aᵢ - Eᵢ) × ln(Aᵢ / Eᵢ)。其中 ln(Aᵢ / Eᵢ) 衡量两个比例之间的差异方向与幅度,(Aᵢ - Eᵢ) 作为权重放大差异。
  2. 求和:将所有分箱的贡献值相加,得到全局的 PSI 值。
数学性质说明: 当实际分布与期望分布完全一致时,每个分箱的 Aᵢ = Eᵢ,则 ln(1) = 0,PSI = 0。分布差异越大,PSI 值越高。

2.2 期望分布与实际分布

在 AML 模型监控中,两个分布的定义直接决定了 PSI 的监控对象:

分布类型 定义 数据来源 典型用途
期望分布 (E) 模型开发时训练集/验证集的评分分布,或模型上线初期的稳定分布 开发数据集、OOT 测试集、上线后首月数据 作为基准参考,代表模型“正常”状态下的评分行为
实际分布 (A) 当前监控周期(如周、月)内模型输出的评分分布 生产环境实时评分日志、批量评分结果 与期望分布对比,检测是否存在漂移
常见陷阱: 期望分布不应使用全量历史数据简单平均。若业务在开发后发生结构性变化(如新客群准入),应重新校准期望分布,否则 PSI 会持续偏高,导致告警疲劳。

2.3 分箱策略:等宽、等频与分位数

分箱策略直接影响 PSI 的敏感度和稳定性。三种主流策略的对比如下:

策略 方法 优点 缺点 AML 场景适用性
等宽分箱 将评分范围 [min, max] 等分为 k 个宽度相同的区间 实现简单,区间边界固定,易于解释 数据稀疏时部分分箱可能无样本,导致 ln(0) 计算异常 低;评分分布通常非均匀,等宽易产生空箱
等频分箱 使每个分箱包含大致相同数量的样本 避免空箱,每个分箱统计意义均衡 边界随数据分布变化,不同周期边界可能不同 中;适用于样本量充足、分布稳定的场景
分位数分箱 基于期望分布的分位数(如 10%、20%…90%)确定边界 边界固定且反映数据内在分布形态,抗噪声能力强 需要预先计算分位数,对极端值敏感 高;AML 模型评分通常呈偏态分布,分位数法最稳健
推荐实践: 在 AML 模型监控中,优先使用 分位数分箱(通常 10 箱或 20 箱),边界基于期望分布计算并固定。这样既能保证每个分箱有足够样本,又能使不同监控周期的 PSI 值具有可比性。

2.4 PSI 计算流程与分箱策略选择

下图展示了从原始评分到 PSI 值的完整计算链路,以及分箱策略在其中的位置:

PSI 计算流程与分箱策略选择 原始评分数据 期望分布基准 分箱策略选择 等宽 / 等频 / 分位数 计算各分箱占比 (Aᵢ, Eᵢ) 贡献值 = (Aᵢ - Eᵢ) × ln(Aᵢ / Eᵢ) PSI = Σ 贡献值 分箱策略影响 PSI 敏感度

2.5 分箱策略对 PSI 值的敏感性分析

不同分箱策略下,相同的分布偏移会产生不同的 PSI 值。以下是一个模拟示例,展示三种策略对同一组数据的计算结果差异:

分箱策略 分箱数 期望分布 (E) 实际分布 (A) PSI 值 敏感性评价
等宽 (0-20, 20-40, …) 5 [0.05, 0.15, 0.40, 0.30, 0.10] [0.02, 0.08, 0.35, 0.40, 0.15] 0.042 中等;尾部箱样本少,贡献不稳定
等频 (各 20%) 5 [0.20, 0.20, 0.20, 0.20, 0.20] [0.15, 0.18, 0.22, 0.25, 0.20] 0.018 较低;均匀分布稀释了局部偏移
分位数 (P10, P20, …) 10 [0.10, 0.10, …, 0.10] [0.06, 0.08, 0.09, 0.11, 0.12, 0.13, 0.11, 0.10, 0.10, 0.10] 0.035 高;能捕捉到评分中低分段的偏移
关键结论: 分位数分箱在 AML 场景中表现最优,因为它能自适应评分分布的非对称性,在样本密集区域(如中低分段)提供更精细的粒度,从而更早发现风险偏移。

2.6 分箱策略选择决策树

以下决策树可帮助在实际项目中快速选择合适的分箱策略:

分箱策略选择决策树 评分数据分布已知? 是 → 检查分布形态 否 → 使用等频分箱 偏态分布 → 分位数 均匀分布 → 等宽 推荐:分位数分箱(10 箱或 20 箱),边界固定于期望分布 避免空箱、抗噪声、跨周期可比
实施建议: 在 AML 模型监控系统中,建议将分箱边界作为模型配置的一部分固化下来。每次计算 PSI 时,使用相同的分箱边界,确保不同时间点的 PSI 值具有纵向可比性。分箱数推荐 10~20 箱,过少会降低灵敏度,过多则引入噪声。