「AI黑箱探测」事后模型归因解析Part 1

「AI黑箱探测」事后模型归因解析Part 1

关注并标星索信达

每天打卡阅读

更快走进金融人工智能世界

━━━━━━



我们是索信达集团旗下的金融人工智能实验室团队,微信公众号(datamargin)将不定期推送原创AI科学文章。我们的作品都是由实战经验丰富的AI科学技术人员或资深顾问精心准备,志在分享结合实际业务的理论应用和心得体会。


文 | 索 信 达 JM Crystal

在可解释机器学习领域,获取可解释性最简单的方法是使用传统的可解释统计模型,如线性回归、逻辑回归、决策树模型等。然而,传统统计模型却往往有精度低的弊端,为了追求更高的精度,人们往往选用现在比较流行的一些机器学习模型,其中包含黑箱模型。


黑箱模型精度很高,但可解释性差,人们无法知道为什么模型给出了这个结果,更不清楚如何判断结果的合理性。为了解决这个问题,科学家们提出一种与模型无关的可解释方法(model-agnostic interpretable)。它能够在模型训练完成后解析出部分可解释性质,从而摆脱了模型本身的限制。


我们分三部分介绍几个与模型无关的可解释方法:

(一)VI、PDP、ICE、ALE;

(二)LIME;

(三)SHAP。


本文将介绍第一部分。除了阐明相关原理,为了演示与模型无关的解释方法,我们从Blockchain.org下载了一组比特币近两三年的价格数据,以及与之相关的货币交易统计、区块信息、采矿信息、网络活动等多维度特征数据。作为一种典型的黑箱模型,Xgboost被用来建立多维特征与比特币价格之间的预测关系,训练得出的R方高达0.9795。下面我们将用不同方法对模型结果进行归因解析。




一、变量重要性(Variable Importance)


变量重要性 (Variable Importance) 也被称为特征重要性(Feature Importance),指的是当某个变量的值发生改变后,模型在预测表现上的差异。如果一个特征非常重要,它对模型精度的影响会很大。通过对一个模型的特征重要性进行排序,就可获得对模型的全局解释。


在与模型无关的几种特征重要性判断方法中,我们举例介绍一种“置换特征重要性“算法(Permutation Feature Importance Algorithm),其基本过程如下:

1. 已知:训练出的模型,特征矩阵,目标变量以及损失函数; 2. 通过损失函数计算出原始模型误差; 3. 对于特征矩阵中的每一个特征: i)随机置换该特征的取值,从而得到了一个新的置换特征矩阵。由于改变了原有特征的值,该特征与目标变量之间的原有关系被打破; ii)使用新的置换特征矩阵得到模型预测值,并计算出置换模型的误差; iii)通过计算置换模型误差和原始模型误差的差异来反映特征重要性。

4. 将计算出的每个特征的特征重要性按降序排列。


下图显示的是基于比特币数据集的Xgboost模型变量重要性排行榜。从中我们能清晰地看出区块信息“区块大小”这个变量极其重要,其次是交易量、平均区块大小等等。



这种方法能够针对模型提供一种高度浓缩、全局化的解释。但是,当特征变量之间高度相关时,置换特征重要性的方法会产生一定程度的偏差。




二、部分依赖图(Partial Dependence Plot)


部分依赖图简称PDP图,能够展现出一个或两个特征变量对模型预测结果影响的函数关系:近似线性关系、单调关系或者更复杂的关系。


给定任意黑箱模型,定义一个部分依赖函数:

其中,X_s表示我们所感兴趣的特征变量,X_c表示所有其他变量。通过对X_c进行积分,我们得到一个只依赖于X_s的函数f(X_s)。该函数即为部分依赖函数,它能够实现对单一变量X_s的解释。


在实际操作中,我们通常使用蒙特卡洛方法,通过计算训练集的平均值,来得到部分依赖函数,具体公式如下图所示,其中n表示样本容量。



单一变量PDP图的具体实施步骤如下:

  1. 挑选一个我们感兴趣的特征变量,并定义搜索网格;
  2. 将搜索网格中的每一个数值代入上述PDP函数中的X_s,使用黑箱模型进行预测,并将得到的预测值取平均;
  3. 画出特征变量的不同取值与预测值之间的关系,该图即为部分依赖图。

以比特币数据集为例,我们使用PDP方法对Xgboost模型结果进行解析。下图刻画的是单变量“区块大小”与比特币价格之间的函数关系。这是一个典型的非线性关系:当“区块大小”在12000-15000范围内增长时,比特币价格逐渐上涨;随着“区块大小”的进一步增长,会对比特币价格产生负向影响,直到区块大小高于20000时,又会对比特币价格产生正向影响。



PDP图的优点在于易实施,缺点在于不能反映特征变量本身的分布情况,且拥有苛刻的假设条件——变量之间严格独立。若变量之间存在相关关系,会导致计算过程中产生过多的无效样本,估计出的值比实际偏高。另一个缺点是样本整体的非均匀效应(Heterogeneous effect):PDP只能反映特征变量的平均水平,忽视了数据异质对结果产生的影响。




三、个体条件期望图(Individual Conditional Expectation Plot)


个体条件期望图(ICE Plot)计算方法与PDP类似,它刻画的是每个个体的预测值与单一变量之间的关系。个体条件期望图消除了非均匀效应的影响,它的原理和实现方法如下:对某一个体,保持其他变量不变,随机置换我们选定的特征变量的取值,放入黑箱模型输出预测结果,最后绘制出针对这个个体的单一特征变量与预测值之间的关系图。


继续以上述比特币数据为例,下图反映的是“区块大小”对比特币价格影响的ICE图,其中浅蓝色线反映的是每个个体的条件期望图,深蓝色线反映所有个体的平均水平。从图中可看出所有个体并不一定遵循相同的变化趋势,因此相较于PDP的一概而论,ICE图能够更准确地反映特征变量与目标之间的关系。



ICE图的优点在于易于理解,能够避免数据异质的问题。在ICE图提出之后,人们又提出了衍生ICE图,能够进一步检测变量之间的交互关系并在ICE图中反映出来。


ICE图的缺点在于只能反映单一特征变量与目标之间的关系,仍然受制于变量独立假设的要求,同时ICE图像往往由于个体过多导致图像看起来过于冗杂,不容易获取解释信息。




四、累积局部效应图(Accumulated Local Effects Plot)


累积局部效应图(ALE plot),用于描述特征变量对预测目标的平均影响。ALE最大的特点是摆脱了变量独立性假设的约束,使其在实际环境中获得了更广泛的应用。


若两个特征变量之间存在相关性,要剥离出单一特征变量对目标的纯粹影响,有一种方法是利用条件分布计算出预测值并对结果取平均。举例来说,若X1与X2相关,想要计算出X2对于目标的影响,需先固定X1并构建一个条件分布函数,如

,再放入模型计算出预测值,从而达到”固定X1仅反映X2与预测值之间关系“的目标。这种方法看似有效,却存在问题,它真正反映出的是X1与X2的联合效应与预测值之间的关系。


下面介绍ALE的原理:通过计算局部效应来消除相关性的干扰。仍以比特币数据为例,探究“区块大小“与比特币价格之间的关系。由于“区块大小”和“(挖矿)难度”之间存在相关性,我们的目标是找出”区块大小“与预测值之间纯粹的影响关系。首先,计算出特征变量”区块大小“在某一点对预测值的影响:如下图所示将空间划分成一系列区间,假设某一点落在N(7)这个区间内,使用模型计算出该区间左右两个边界点处的预测值,两个预测值相减后的差值即能反映出变量在这一点对目标的影响大小。鉴于每一个小区间内通常会存在多个样本点,此时将他们的预测结果取平均,便能得到该区间内“区块大小“与比特币价格之间的关系。


这种方法之所以能够消除变量相关性的影响是因为在一个小的区间内,相关变量的取值基本不变,两个边界点预测值的差别仅仅来自于变量”区块大小“取值的差别。因此,特征变量的纯粹影响能够被剥离出来。这种方法就叫做局部效应法(Local Effect)。



要获知某单一变量在整个值域上对预测值的影响,需要进一步引出累积局部效应法(Accumulated Local Effects)。累积局部效应法将单个局部效应进行累积,能够反映单一特征变量对预测结果的整体影响情况。


下面举例说明:在比特币数据集中,变量“区块大小”与“(挖矿)难度“之间呈强相关关系,相关系数超过0.95。若研究单变量对目标的影响,则PDP、ICE两种方法不再适用。使用ALE方法处理后能得到如下所示的图。相较于前文的PDP图,ALE图置信区间更窄,精确度更高。





同时,使用ALE方法也可以研究具有强相关性的两个变量对目标的联合效应。以比特币数据为例,“哈希率”和“(挖矿)难度”之间存在强相关性,他们对比特币价格的联合效应如下图所示:红色代表比特币价格高于平均值,蓝色代表比特币价格低于平均值。这张图反映出了“哈希率“与”(挖矿)难度“之间的交互关系:挖矿难度较大且哈希率较高(大于0.8)会提高比特币价格。当哈希率处于0.5-0.7的范围且挖矿难度较大时,会降低比特币价格。



综上,ALE方法有三个优点:第一,ALE方法做出的图是无偏的。由于ALE能够处理特征之间的相关关系,做出的图像不会受到联合效应的影响。第二,ALE的计算速度比PDP快,需要计算的次数少于PDP。第三,ALE图的解释非常清晰明了。由于剥离了相关变量的影响,人们可以很容易对特征变量和模型结果进行解释。


ALE方法同样存在一些问题,比如如何去确定区间,到底确定多少个区间比较合适等等,都是需要进一步的研究与探讨。




五、总结


上述介绍的VI、PDP、ICE和ALE是“与模型无关的解释方法”中最基本、最常用的四种。四种方法各有利弊,人们可以针对实际的业务场景有选择性地应用这些方法,从而达到技术与业务相结合,一加一大于二的效果。

发布于 2019-11-05 14:30