基于 WH 转化的协同联邦双重随机森林算法 为了在保障数据隐私的前提下解决传统联邦随机森林在非独立同...
تم الإنشاء في: ٦ سبتمبر ٢٠٢٦
تمت الإجابة باستخدام GPT-5.6 Thinking بواسطة Chat01
تم الإنشاء في: ٦ سبتمبر ٢٠٢٦
تمت الإجابة باستخدام GPT-5.6 Thinking بواسطة Chat01
基于 WH 转化的协同联邦双重随机森林算法
为了在保障数据隐私的前提下解决传统联邦随机森林在非独立同分布数据下面面临的特征
选择偏误与节点样本多样化不足导致的过拟合问题,本文提出了一种基于 Wilson-Hilferty 转化
的协同联邦双重随机森林算法。此算法中,模型不采用在各个客户端独立构建树模型后简单平
均的传统方法,而是采用协同联邦构建的策略。算法主要分为三个阶段:初始化阶段、树生长阶
段与树调节阶段。
3.1 初始化阶段
首先,由中央服务器生成包含 M 棵空决策树的集合 E = {t1, t2, . . . , tM}。其次,为森林中
的每一棵决策树 tj 生成一个客户端 C 的随机序列 Pj = {C1, C2, . . . , CK}。该客户端的随机序
列决定了决策树在树生长阶段,各个客户端之间迭代的先后顺序。通过为不同决策树生成相互
独立的客户端序列,增强了集成模型的泛化性能。
3.2 树生长阶段
树生长阶段是此算法的核心。服务器根据客户端序列 Pj,在每一轮迭代中将树发给对应的
客户端。客户端接收到待生长决策树集合后,进行以下三个步骤:
√
ν
[(
χ
2
ν
ν
)1/3
− 1 +
2
9ν
])3
. (1)
2. 若 X 为连续变量
本文遵循 GUIDE 算法提出的连续变量处理方式 [34],首先将连续变量离散化,再利用
列联表完成统计检验, 具体步骤如下:
(a) 计算当前节点 t 对应的特征子集 Dt
∗ 中特征 X 取值的均值 x¯ 和标准差 s。
(b) 若样本量满足 N(t) ≥ 20Jt,则以 x¯ 和 x¯ ± s
√
3/2 作为边界值,将 X 的连续取值
范围自适应划分为四个区间。否则,若 N(t) < 20Jt,则以 x¯ ± s
√
3/3 作为边界值,
将 X 的取值范围划分为三个区间。
(c) 构建一个交叉列联表,以类别标签作为行,以本步骤(b)中划分的离散区间作为列。
(d) 遵循步骤 1(b) 的数理转化映射方法,计算得到对齐量纲后的无偏得分 WM(X)。
最终,算法通过比较 Dt
∗ 中所有特征的 WM(X) 值,选择该得分最大的特征作为当前节点
的最优分裂特征。Wilson–Hilferty 转化仅用于确定最优分裂特征,而对于选中的特征,在
节点样本 Dt
∗ 上采用 CART 基尼不纯度准则确定最优分裂阈值。随后,再利用该分裂规则
对节点数据 Dt 进行真正划分,并生成左右子节点。
3. 寻找最优分裂值
在步骤 2 于节点 t 处选定最佳分裂变量 X 后,我们需要找到 X 的一个取值来进行分裂,
形成节点分裂 t = tL ∪ tR,其中 tL 和 tR 分别表示节点 t 的左子节点和右子节点。最优分
裂阈值的选择是在节点数据集 Dt
∗ 上进行的。设 pL 和 pR 分别表示进入左、右子节点的样
本比例。我们的目标是寻找能够使基尼不纯度加权和最小化的分裂值:
min pLg(tL) + pRg(tR) (2)
若 X 为连续变量,则存在 (nt − 1) 种可能的最优分裂值,先对连续变量进行从小到大的
排序,取每个分裂值为两个变量间的均值;若 X 为离散变量,则遍历这些所有的分裂值,
找出使基尼不纯度加权和最小的最优分裂值。
值得注意的是,节点划分完成后,仍采用原节点样本 Dt 按照所得分裂规则生成左右子节点。
3.3 树调节阶段
在树调节阶段中,每个客户端都会从服务器接收一组决策树集合 E。随后,各客户端利用
本地数据调整 E 中各决策树的叶节点信息,并将本地的完整数据集遍历整个决策树结构,也就
是将全部数据从根节点按照分裂条件传递至叶节点。对于每个叶节点,计算到达该节点的数据
子集 Dt 中各类别出现的频率。在 Dt 中出现频率最高的类别被定义为多数类别标签 l,并存储
于该叶节点中。
随后,每个客户端都将树模型返回给中央服务器。中央服务器随后汇总所有客户端的树模
型中叶节点的信息。对于每棵树中的每个叶节点,中央服务器会将各客户端提供的类别标签 l ,
汇总为多数类标签列表 L ,记 L = {l1 , l2 , . . . , lM },并将该列表存储在最终集成模型 E 中对
应树结构的叶节点内。因此,E 中的每个叶节点都将包含各客户端提供的类别标签列表。
4
3.4 预测阶段
在预测阶段,对于给定的测试样本 x,它将由最终的联邦森林模型 E = {t1, t2, . . . , tM} 进
行处理。
当样本 x 遍历第 k 棵决策树 tM 时,最终会到达特定的叶节点。在每个叶节点处,都包含一
个标签列表 Lk。该列表包含了在树调节阶段到达该节点的不同客户端所确定的多数类别标签。
对于所有的决策树,我们将它们输出的标签列表进行汇总,并计算每个类别标签的出现频
率,从而构建一个类别频率向量 F。其中,F = [f1, f2, . . . , fC] 表示类别 c 在所有列表中出现的
总频率。
最终的测试样本的结果 yˆ 其数学表达式如下:
yˆ = argmax
c
F[c] (3)
这种多数投票的机制能够确保预测结果具有可靠性,从而准确反映不同客户端数据分布的
异质性特征。若出现多个类别获得相同的最高票数,则通过随机选取其中一个类别作为最终的
预测结果。这是我的算法 如果我想加入证明 可以从哪些方面加入呢 同时首先给我证明的完整步骤 其次给我证明的解释 我可以从联邦和随机森林的哪些方面加入证明呢 如果我想加入随机森林收敛性的证明 要怎么加入呢
可以,而且你的算法其实非常适合加一章“理论性质与收敛性分析”。但要把几种不同性质分开证明,否则很容易出现“证明了树数收敛,却写成算法 Bayes 一致性”的逻辑跨越。
我建议你把证明体系分成 5 个核心定理 + 2 个辅助命题。其中最值得作为主定理的是:
定理 1:固定联邦训练数据时,当森林树数 ,你的协同联邦双重随机森林类别投票分数几乎必然收敛,并且在不存在极限投票平局时,分类结果与泛化误差也收敛。
这个证明最完整、最稳健,也最符合经典随机森林理论。Breiman 对随机森林的经典结果也是从“树数趋于无穷时,利用强大数定律证明森林泛化误差趋于极限”出发。(Department of Statistics)
我建议最终形成下面这一套。
| 理论部分 | 能证明什么 | 推荐程度 |
|---|---|---|
| WH 变换的自由度校准性质 | 不同自由度卡方统计量可以在统一尺度比较 | ★★★★★ |
| WH 节点特征选择一致性 | 大样本下有信息特征压过无信息特征 | ★★★★★ |
| 客户端随机排列公平性 | 每个客户端处于各生长位置的概率渐近相等 | ★★★★★ |
| 双重 Bootstrap 性质 | 二次抽样不改变经验分布的一阶期望,同时增加随机扰动 | ★★★★☆ |
| 森林树数 收敛 | 森林投票、预测以及风险稳定 | ★★★★★ |
| 有限树数的收敛速度 | 用 Hoeffding 给出指数型误差界 | ★★★★★ |
| 联邦叶节点共识一致性 | 各客户端样本量增加时叶节点标签趋于稳定 | ★★★★★ |
| Bayes 一致性 | 在较强正则条件下趋于 Bayes 分类器 | ★★★★☆ |
| 强度—相关性分析 | 解释双 Bootstrap 为什么可能降低过拟合 | ★★★★☆ |
| 隐私证明 | 当前算法只能证明“不传原始数据”,不能证明 DP | ★★★☆☆ |
| 通信复杂度 | 分析服务器与客户端模型传输量 | ★★★☆☆ |
其中我最推荐你在论文正文放 前 7 个,而把严格 Bayes 一致性写成“在一定正则条件下的进一步结果”。
设有 个客户端,
第 个客户端拥有
类别数为 。
森林包含 棵树。
对于第 棵树,把所有随机性统一记为
这里 包括:
在给定联邦数据
后,第 棵树的结构完全由 决定。
给一个测试样本 ,它在第 棵树中到达叶节点
经过你的树调节阶段,客户端 给这个叶节点一个标签
于是定义第 棵树对类别 的“联邦投票比例”
注意:
你的最终森林实际上计算的是
最终分类器为
这就是后面所有证明的起点。
这是我最建议你放进论文的主定理。
假设:
A1. 在给定联邦数据集 的条件下,不同树的随机变量
相互独立且服从相同分布。
也就是说不同树使用独立的客户端排列和 Bootstrap 随机数。
A2. 客户端数量 和类别数 有限。
定义无限森林的类别得分
那么对于任意固定测试样本 和类别 ,
由于
显然
所以
根据假设 A1,在给定 后,
为独立同分布随机变量。
因此根据强大数定律,
即
证毕。
仅仅证明 收敛还不够,我们继续证明 收敛。
定义无限森林预测:
假设最大值唯一。
令
定义无限森林的投票间隔
假设
因为对于所有 ,
取
当 足够大时,对于所有类别 ,都有
于是
而任意 ,
因此
所以最终一定有
当 足够大时成立。
因此
只要极限分类不存在平局。
定义测试分布 ,泛化风险为
无限森林风险:
如果对于 -几乎所有 ,
那么上面已经证明
几乎处处成立。
因此
而指标函数始终满足
根据控制收敛定理,
这就是你的随机森林树数收敛定理。
它与 Breiman 随机森林经典证明中的思想是一致的:把每棵树看成独立随机分类器,用强大数定律证明无限森林投票稳定。Breiman 进一步指出森林误差与单树强度、树间相关性有关。(Department of Statistics)
这个会明显增强论文理论部分。
因为
由 Hoeffding 不等式,
由于一共有 个类别,利用并集界:
现在令
只要所有类别得分误差都小于 ,有限森林就与无限森林给出相同预测。
因此
这个结论特别好。
它告诉你:
当无限森林的类别投票存在正间隔 时,有限森林与无限森林预测不一致的概率随树数 指数衰减。
这比单纯说“随机森林会收敛”更有理论含量。
它不是在证明
其中 是 Bayes 风险。
它证明的是:
也就是:
增加树的数量不会让模型无限震荡;你的协同联邦森林最终趋向一个稳定的无限森林分类器。
这是“森林规模收敛”。
而所谓“统计一致性”要求证明
这是另一个更难的问题。
所以你的总收敛框架最好写成:
第一步你现在可以严格证明。
第二步需要额外条件。
你的初始化中有一个很好的理论点:
是客户端随机排列。
假设每棵树都独立地从所有 个排列中均匀选择一个。
对于客户端 ,定义
显然
设在 棵树中客户端 位于第 个位置的次数为
那么
因此
由强大数定律,
还可以用 Hoeffding:
决策树高层节点通常比深层节点影响更多样本。
如果永远按照
训练,就可能导致某些客户端长期控制更靠近根节点的位置。
而随机客户端排列保证:
因此随着森林规模增加,各客户端在不同树深位置上的参与比例趋于一致。
所以你的论文可以写:
随机客户端序列并不能保证任意 non-IID 场景下预测误差一定降低,但它能够从概率意义上消除固定客户端迭代顺序造成的系统性位置偏差。
这个表述非常稳。
你现在的公式其实非常适合推导。
设特征 的 Pearson 卡方统计量为
Wilson–Hilferty 变换指出:
近似服从标准正态分布:
这是 Wilson 和 Hilferty 1931 年提出的经典立方根近似。(USGS Water Resources)
对于一个自由度的卡方变量 ,反过来有
因此
将 代入:
于是得到你的公式:
这正是“双重 Wilson–Hilferty 转化”的来源。分类树文献中确实有使用双重 WH 变换把不同自由度卡方统计量转换到统一的一自由度尺度,再进行变量选择的做法。(UW Computer Sciences)
你现在写:
“无偏卡方值 ”
我建议改成:
“经自由度校准后的近似一自由度卡方得分”
或者:
“Wilson–Hilferty 自由度校准得分”
因为 WH 本质上是近似变换。
严格意义上不能证明:
完全精确成立。
比较安全的是:
GUIDE 相关方法的目标也是降低或使变量选择偏差变得很小,而不是说有限样本下数学意义上绝对零偏差。(UW Computer Sciences)
还有一个算法细节建议你补上:
否则审稿人可能会问 怎么处理。
这个很适合支撑你的创新点。
假设节点中有 个候选变量。
先考虑无关变量 。
如果
则在标准卡方检验条件成立时,
因此
相应地
如果特征 和类别存在真正关联,定义总体 Pearson 距离
如果
则根据经验频率的大数定律,
所以
而 WH 转换关于 是单调非减的,并且
另一方面,无关特征满足
于是,如果候选特征数量有限:
因此:
它证明的是:
当节点样本量充分大、真正相关特征产生非零类别关联时,WH 得分对有效变量不断增大,而无效变量保持随机的 水平,所以错误地选取纯噪声变量的概率趋于零。
这比简单写“WH 可以降低特征选择偏差”更强。
你的连续变量先离散化。
那么增加两个条件即可:
因此你的三个或四个区间边界也收敛到总体边界。
只要离散化以后仍然存在
就可以重复上面的证明。
不过这里必须加一句:
如果某个连续特征虽然与 有关系,但三/四区间离散化恰好完全消除了这种关系,那么上述一致性结论不一定成立。
这是理论上严谨的写法。
这一块要特别小心。
不要证明:
“第二次 Bootstrap 增加了样本数量/独立信息。”
这是不成立的。
你可以证明的是:
二次 Bootstrap 在经验分布意义下不产生一阶系统性偏移,同时增加了一个独立随机扰动源。
设节点数据
从其中再次有放回抽取 个样本:
对于任意函数 ,定义
以及二次 Bootstrap 后
条件于原节点数据 :
因此
同时
所以:
这正好可以用来解释你为什么要做第二层随机化。
可以结合随机森林经典的“树强度—相关性”观点。
假设单棵树某个预测得分 有
且任意两棵树相关系数近似为 。
森林平均得分为
那么
等价于
当
有
所以即使树非常多,如果
很高,森林方差仍然不会很小。
Breiman 的理论同样把森林性能与单树强度以及树间相关性联系起来。(Department of Statistics)
于是你的逻辑应该写成:
注意“有潜力”。
不能直接数学证明:
对任意数据分布都成立。
这个最好通过实验验证,例如统计不同树预测向量间平均 Pearson correlation。
这样就形成:
理论:增加独立随机扰动;
实验:观察到树间相关性下降;
经典 RF 理论:相关性下降有利于集成方差。
逻辑会非常完整。
这其实是你算法非常有特色的一块。
固定一个叶节点区域 。
客户端 的总体类别概率定义为
客户端使用本地数据计算
假设该叶节点内的样本数量
根据大数定律,
假设客户端 在这个叶节点存在唯一多数类别
并且类别间隔为
由于经验频率收敛,所以最终
将与总体多数类一致:
服务器得到
于是最终服务器类别投票比例趋于
因此服务器叶节点预测收敛到
这个可以称为:
客户端多数类共识一致性(client-majority consensus consistency)
非常适合你的 non-IID 场景。
你的最终规则收敛的不是:
而是:
也就是:
“客户端多数类的多数”
这和:
“所有客户端数据合在一起后的多数类”
不是同一个东西。
因此你的算法理论上最准确的定位不是“等价于集中式随机森林”,而是:
建立面向异构客户端的决策级共识分类器。
这个其实更符合你的创新点。
如果某个客户端在叶节点 中:
那么你现在算法中的“多数类别标签”实际上没有定义。
建议明确规定:
服务器聚合时忽略 。
于是改为:
这在 non-IID 情况尤其重要,因为一个客户端完全可能没有样本进入某些树叶。
把这个细节补上以后,算法和理论都会严谨很多。
定义一个虚拟随机客户端
假设某棵树只随机选择一个客户端的叶标签,则类别 的随机投票为
而你的算法不是随机挑一个客户端,而是使用全部客户端:
注意:
根据全方差公式或者 Rao–Blackwell 原理,
也就是说:
与每棵树随机采用单客户端叶节点标签相比,你的“保存所有客户端多数标签列表”的方式,在不改变平均客户端投票含义的情况下,可以降低客户端选择造成的额外随机方差。
这个证明我很推荐。
因为它是专门针对你的树调节阶段设计出来的理论结论,不是直接套随机森林经典定理。
这部分要进入
经典随机森林一致性理论里,一个很常见的路线是证明:
以及
直观上:
这样局部类别比例才能逼近真实条件概率。Biau、Devroye 和 Lugosi 的随机森林一致性研究明确使用了这样的分割分类器条件;后续随机森林理论也大量沿用类似思想。(ResearchGate) Scornet、Biau 与 Vert 后来还对更接近 Breiman 随机森林的模型证明了统计一致性,但需要额外模型和正则条件。(Project Euclid)
设
为了允许 non-IID,不要求
甚至可以允许条件概率不同。
但是先加入一个相对合理的条件:
对于几乎所有 ,存在唯一类别 ,使得所有有效客户端都满足
也就是说,不同客户端可以:
甚至
但它们在分类边界上保持一致。
这是比完全 IID 弱很多的条件。
设测试点 所在终端节点为
要求
对于参与这个区域投票的客户端,
比如
连续。
定义
要求
几乎处处成立。
首先由于
叶节点 中所有点逐渐靠近 。
根据 的连续性,
记
于是
另一方面,由
以及大数定律,
因此
由 Bayes 类别间隔
最终客户端叶节点多数标签满足
概率趋于 1。
即
因为 有限,所以
因此所有客户端的叶节点投票最终都会集中在
于是单棵协同树:
进而无限森林:
最后:
因此
这就是 Bayes 一致性。
最终形成两层极限:
其中
也就是说:
利用强大数定律:
利用
和
证明
这个结构非常适合作为你整篇论文理论部分的核心图景。
你现在没有明确写停止生长规则。
如果树无限分裂到每个叶子只有 1 个样本,那么
就无法成立。
所以建议算法增加类似:
并要求理论上:
直观解释:
这正是很多树分类器一致性证明中非常典型的“叶节点越来越小但样本越来越多”的平衡思想。相关随机森林一致性结果也采用类似条件。(ResearchGate)
也就是必须证明
这个对你的 WH 选变量并不是天然成立的。
因为如果 WH 总喜欢选择 ,树可能一直:
分裂,却从来不处理其他相关方向。
所以如果你真的想做严格 universal consistency,证明会很困难。
有两个办法。
把它写成一个正则条件:
假设 WH-CART 分裂机制所生成的随机分割满足终端节点直径趋于零和终端节点样本数量趋于无穷。
然后证明:
这是完全正常的理论写法。
增加一个很小的随机探索概率 。
例如:
这样每个变量都有正概率被不断选择:
就更容易证明各维度不断被切分,从而:
近年来也有随机森林理论通过“概率充分不纯度下降”等条件证明更一般随机树算法的一致性,思路就是给各种分裂方式留出足够的概率性下降条件。(arXiv)
不过这个会改变你的算法,所以如果你不想改,采用方案 A 就好。
假设:
即不同客户端在同一个 上连最优类别都不同。
那么不存在一个天然的“所有客户端共同 Bayes 分类器”。
这时候不要硬证明
你应该定义:
然后定义联邦共识目标:
再证明:
这个反而非常符合你的算法。
然后给一个推论:
如果所有客户端具有共同 Bayes 最优类别,即
则
从而得到 Bayes 一致性。
这样逻辑非常干净:
然后
我按推荐顺序给你排:
证明
解释协同生长不会长期偏向某个客户端的高层节点。
证明
进一步:
这直接对应 non-IID。
证明
这能说明为什么你不是只选择一个客户端标签。
你当前算法可以严格写:
原始训练记录
从不被传输至中央服务器。
服务器获得的是:
但这里要特别注意:
你现在不能证明 differential privacy。
因为一个客户端改变一条记录,可能导致叶节点多数标签从 0 变为 1,因此服务器仍可能从模型输出来推断一定信息。
类似的协同联邦随机森林工作也采用“只传叶节点多数类别标签、原始数据保留在本地”的设计来减少数据暴露。(ScienceDirect) 但如果你论文中要写“严格保障隐私”,则需要 DP、安全聚合或安全多方计算。
所以建议把:
“在保障数据隐私的前提下”
改为更严谨的:
“在无需交换客户端原始数据的联邦学习框架下”
除非你后面加入 DP。
我最推荐下面五个:
第一:树数收敛
这是主定理。
第二:有限森林误差界
理论含量很高。
第三:双 Bootstrap 一阶无偏扰动
解释第二重随机化。
第四:树间相关性—森林方差关系
解释为什么需要增加随机性。
第五:统计一致性
在
等条件下证明
经典 RF 理论从 Breiman 的森林规模收敛,逐渐发展到各种随机森林模型的一致性分析;最近的综述也把一致性及其收敛速度作为 RF 理论的重要主线。(Annual Reviews)
你目前是:
3.1 初始化阶段
3.2 树生长阶段
3.3 树调节阶段
3.4 预测阶段
后面直接增加:
定理 1:
并推导你的公式。
证明:
而
所以
证明:
再结合树相关性公式进行解释。
证明:
这是正文主定理:
以及
再给有限树数界:
证明:
以及
最后写成条件定理:
如果
并且不同客户端具有共同 Bayes 最优类别,则
于是最终:
如果是硕士论文或篇幅有限,不必全证明。
优先放:
然后把 Bayes 一致性写成一个“推论”:
在叶节点直径收缩、叶样本数发散以及客户端具有共同最优分类决策的条件下,该算法进一步具有 Bayes 一致性。
这是风险最低、理论完整度最高的组合。
另外,你现在如果真的准备加入“随机森林收敛性证明”,一定把“ 的森林收敛”和“ 的统计一致性”分开写。这是整部分最关键的理论区分。Breiman 的经典结果首先解决的是前者;更完整的统计一致性需要额外的分区正则条件。(Department of Statistics)
如果按你的算法创新程度来排序,**“WH 特征选择一致性 + 客户端顺序公平性 + 联邦投票 收敛 + 客户端共识一致性”**这四个定理最能体现这是“你的算法的证明”,而不是简单把传统随机森林定理搬过来。
Algorithm 1 基于 WH 转化的协同联邦双重随机森林
输入: 客户端集合 C = {C1, C2, . . . , CK};各客户端本地数据集 D = {(xi
, yi)}
Nt
i=1,样本量为 n;森林中
决策树的总数 M。
输出: 全局联邦双重随机森林模型 E = {t1, t2, . . . , tM}。
=== 1. 初始化过程 ===
1: for 每棵树 tj ∈ E do
2: 服务器生成一个客户端随机序列 Pj = {C1, C2, . . . , CK},该序列决定了决策树 tj 在各客户端间
的迭代顺序
3: end for
=== 2. 树生长过程 ===
4: for 生长迭代轮次 h = 1 to K do
5: 服务器根据序列 Pj 在第 h 轮迭代中将树发给对应的客户端
6: for 每个客户端 Ci 并行执行 do
7: Ghi ← 当前客户端 Ci 接收到的待生长决策树子集
8: for 每棵树 tj ∈ Ghi do
对于当前树 tj 中待分裂的当前节点 t,此节点的本地样本子集为 Dt,样本量为 nt:
9: (1) 双重 Bootstrap 抽样机制:
10: if 当前节点为根节点 then
11: 初始化根节点数据集 Dt ← Bootstrap(D)
12: else
13: if nt > n × 0.1 then
14: 从 Dt 中再次进行第二次 Bootstrap 抽样生成样本子集: Dt
∗ ← Bootstrap(Dt)
15: else
16: 直接令当前计算数据集为: Dt
∗ ← Dt
17: end if
18: end if
19: 从 Dt
∗ 的特征空间中无放回随机抽取 m ≈
√p 个特征组成候选子集 V
7
Algorithm 2 基于 WH 转化的协同联邦双重随机森林
20: (2) 基于无偏卡方检验与 Wilson-Hilferty 转化的特征选择
21: 初始化 best_wm_score ← −1, best_feat ← None
22: for 每个候选特征 X ∈ V do
23: if 特征 X 是分类变量 then
24: 以数据集 Dt
∗ 的类别标签为行,特征 X 的不同分类取值为列,构建交叉列联表
25: 计算该列联表删除无观测值行和列后的自由度 ν 及独立性检验卡方统计量 χ
2
ν
26: if ν > 1 then
27: 计算无偏得分 WM(X) ← max {
0,
(
7
9 +
√
ν
[(
χ
2
ν
ν
)1/3
− 1 + 9
2
ν
])3
}
28: else
29: WM(X) ← χ
2
ν
30: end if
31: else 特征 X 是连续变量
32: 计算当前节点子集 Dt
∗ 中特征 X 取值的均值 x¯ 和标准差 s
33: if 样本量满足 nt ≥ 20Jt then ▷ J 为 D 的类别数,Jt 为 Dt 的类别数
34: 依据边界 x¯ 和 x¯ ± s
√
3/2 将连续值划分为四个区间
35: else
36: 依据边界 x¯ ± s
√
3/3 将连续值自适应划分为三个区间
37: end if
38: 以类别标签为行,划分后的离散区间为列构建交叉列联表
39: 遵循步骤 1 的转换方法,计算得到的无偏得分 WM(X)
40: end if
41: if WM(X) > best_wm_score then
42: best_wm_score ← WM(X), best_feat ← X
43: end if
44: end for
8
Algorithm 3 基于 WH 转化的协同联邦双重随机森林
45: (3) 寻找最优分裂值
46: if best_feat = None then
47: 设 m 为数据集 Dt
∗ 中变量 X 对应的取值数
48: if X 为连续变量 then
49: 对 X 的取值进行升序排序,取相邻两变量间的均值共 (m − 1) 种作为候选分裂值
50: elseX 为离散变量
51: 则共有 (2m−1 − 1) 种候选分裂值
52: end if
53: 遍历所有候选分裂值,分别计算模拟分裂 t = tL ∪ tR 后的指标值
54: 找出使基尼不纯度加权和 min (pLg(tL) + pRg(tR)) 的最优分裂值
55: end if
56: 将 Dt 按照最优分裂规则进行传递,按照最优分裂规则将 Dt 划分为 DtL 和 DtR
57: (4) 传递:
58: 当前客户端完成本轮节点生长
59: 将当前决策树发送服务器
60: end for
61: end for
62: 服务器将树传递给序列 Pj 中的下一个客户端进行下一轮迭代
63: end for
Algorithm 4 基于 WH 转化的协同联邦双重随机森林
=== 3. 树调节过程 ===
1: for 每个客户端 Ci 并行执行 do
2: 从服务器接收决策树集合 E
3: 利用本地数据集遍历 E 中各决策树,每棵树都形成一个形成多数类标签 l
4: end for
5: for 每棵树 do
6: 服务器汇总在客户端 Ci 出现的多数类标签 l, 记为多数类标签列表 L = {l1 , l2 , . . . , lM }
=== 4. 预测过程 ===
7: 计算在 L 中出现的各类别的频数,记为 F = [f1, f2, . . . , fC ]
8: 令 yˆ = argmaxc F[c],yˆ 为最终预测结果 这是我的伪代码 请你按照这个符号 为我重新给出这七个证明 以及解释:WH 变换的自由度校准性质、不同自由度卡方统计量可以在统一尺度比较
WH 节点特征选择一致性 大样本下有信息特征压过无信息特征 ★★★★★
客户端随机排列公平性 每个客户端处于各生长位置的概率渐近相等 ★★★★★
双重 Bootstrap 性质 二次抽样不改变经验分布的一阶期望,同时增加随机扰动 ★★★★☆
森林树数
M
→
∞
M→∞ 收敛 森林投票、预测以及风险稳定 ★★★★★
有限树数的收敛速度 用 Hoeffding 给出指数型误差界 ★★★★★
联邦叶节点共识一致性
可以。按照你现在的 Algorithm 1–4,我建议先做几个符号上的统一,然后再给出 7 个理论结果。下面这套基本可以直接改写成论文中的“算法理论性质分析”。
先说明:你当前的算法中,Wilson–Hilferty 部分与 GUIDE 分类树中的做法是一致的,即先把不同自由度的 Pearson 卡方统计量经 WH 近似转换到标准正态尺度,再反变换到 1 自由度卡方尺度,从而进行变量比较。(UW Computer Sciences) 随机森林的 收敛则可以沿 Breiman 的强大数定律框架证明。(Department of Statistics)
不过正式写证明前,你的伪代码建议先修正 4 个地方:
if best_feat ≠ None,否则没有选出特征时反而进入“寻找最优分裂值”,逻辑相反。你伪代码中的符号尽量保持不变。
共有 个客户端:
为了证明时区分客户端,把客户端 的数据记为
如果你的实验中所有客户端样本量统一写为 ,则可以令
森林为
对于树 ,服务器产生一个随机客户端排列:
在客户端 当前节点 上:
当
时:
否则:
从 个特征中随机无放回选择
个候选特征,记为
总类别数记为 ,当前节点存在的类别数为 。
下面所有证明都按照这些符号进行。
定理 1(Wilson–Hilferty 自由度校准性质)
设当前节点 中候选特征 ,其 Pearson 独立性检验统计量为
其中 为删除零观测行列后的自由度。
当零假设
成立时,对于 ,定义
则在 Wilson–Hilferty 近似下,
因此,对于具有不同自由度
的候选特征,经过 WH 转换以后,其零假设统计量均近似位于统一的
尺度上。
若
Wilson–Hilferty 立方根变换给出
原始 Wilson–Hilferty 结果就是利用卡方变量立方根近似正态分布。(USGS Water Resources)
因此定义标准化变量
则
另一方面,若
再次应用 Wilson–Hilferty 变换:
其中
因此,
现在将 代入:
注意:
因此
即
由于 WH 近似在极小统计量区域可能产生小于 0 的近似值,所以算法取
因此定理成立。证毕。
GUIDE 分类树正是使用“两次 Wilson–Hilferty 变换”,把自由度大于 1 的 Pearson 卡方统计量转换成 1 自由度卡方值后进行变量比较。(UW Computer Sciences)
假设两个特征:
直接比较
并不公平,因为在零假设下
自由度越大的变量,其原始卡方值天然倾向于更大。
WH 变换之后:
于是两者被放在统一尺度比较。
因此论文里建议不要说:
得到了“严格无偏卡方值”。
而写:
通过 Wilson–Hilferty 转换将不同自由度的 Pearson 卡方统计量近似映射到统一的一自由度卡方尺度,从而削弱由变量取值数和自由度差异导致的特征选择偏差。
因为 WH 是近似,而不是严格等分布。
这是最能体现你 WH 创新的理论之一。
在当前节点 上,将候选特征分成两类。
有关特征集合:
无关特征集合:
由于 Algorithm 1 第 19 行只随机选择 个变量,所以实际参与比较的是
因此严格定理必须写成:
即候选子集 中至少包含一个真正有信息的变量。
定理 2(WH 节点特征选择的一致性)
假设当前节点样本数量满足
候选变量数量 固定,并满足标准 Pearson 卡方渐近条件。
进一步假设:
对于连续变量,假定 Algorithm 2 第 32–38 行离散化以后仍保持与类别 的非零关联。
则有
即大样本条件下,只要随机候选特征集合中存在至少一个有效变量,WH 选择无信息变量的概率趋于 0。
对于
有
因此 Pearson 卡方统计量满足:
所以
由于 WH 变换为连续函数,因此
因此所有无关特征得分不会随着
增加而无限增长。
若候选无关特征数量有限,则
设
对于离散后的列联表,记总体联合概率为
边缘概率:
定义 Pearson 关联距离:
由于
所以存在至少一个 使得
因此
由大数定律,
所以
因此
再看 WH 公式:
当
时,其主导项为
于是
因此:
对于无信息特征:
对于信息特征:
因此
而你的 Algorithm 2 第 41–43 行执行
因此:
条件是
证毕。
你不能直接写:
无条件成立。
为什么?
因为你 Algorithm 1 第 19 行有
随机特征抽样。
假设总共有 个有效特征,那么一个有效特征都没有进入 的概率为
因此
所以严格而言:
因此你的论文应该写:
条件于候选特征集合 至少包含一个与类别相关的特征,随着节点有效样本量增加,WH 选择规则选择无关变量的概率趋于零。
这比无条件“趋于 1”严谨得多。
这对应 Algorithm 1 第 1–3 行。
假设对于每棵树
服务器从 个客户端排列中均匀、独立地产生随机序列
则对任意客户端 及任意生长位置
客户端 出现在位置 的频率满足
定义指示变量:
因为 是 个客户端的均匀随机排列,所以:
且
在 棵树中,客户端 位于位置 的总次数为
因此:
所以
因为不同树的随机排列相互独立,所以根据强大数定律:
即:
证毕。
例如有 4 个客户端:
如果永远按照
生长,那么:
而树靠近根部的分裂通常对更多样本产生影响,因此固定顺序可能带来“位置偏差”。
你的随机排列保证:
并且:
后,各位置实际频率也趋于
因此论文中可以写:
随机客户端排列在概率意义上消除了固定迭代顺序造成的系统性客户端位置偏差,使各客户端获得渐近相等的不同树生长位置参与机会。
注意它证明的是参与位置公平性,不是直接证明“所有客户端预测贡献完全相等”。
这对应 Algorithm 1 第 9–18 行。
当
时,对当前节点样本
执行第二次 Bootstrap:
其中
则:
第二次 Bootstrap 不改变当前节点经验分布的一阶条件期望:
除非节点中所有样本在所考察统计量上完全相同,否则:
因此第二次 Bootstrap 引入了额外的节点级随机扰动。
节点 的经验分布为
Bootstrap 相当于从这些样本中有放回抽取 次。
设原样本 被抽中的次数为
则:
第二次 Bootstrap 的经验分布:
因为:
所以:
于是:
因此:
性质一得证。
设考察任意函数
节点原经验均值:
Bootstrap 后:
那么:
且
若
并非对所有样本都相同,则:
所以:
以
为中心,同时产生新的随机波动。
你的机制为:
因此不同树即使来到相似节点,由于:
不同,也可能获得:
进而产生不同:
或不同分裂阈值。
所以可以写:
第二重 Bootstrap 在条件期望意义上保持当前节点经验分布的一阶中心不变,同时为 WH 特征评估及 CART 分裂值搜索引入额外随机扰动,从而增强不同决策树之间的结构多样性。
但不要写:
“二次 Bootstrap 一定降低树间相关性。”
你目前只能严格证明:
要证明树间相关性确实下降,应当实验比较:
Breiman 的随机森林分析指出,森林性能与单树强度和树间依赖/相关性密切相关。(Department of Statistics)
这是我认为你论文中最重要的主定理。
首先需要把 Algorithm 4 的投票数学化。
对于测试样本 ,它在树
中到达某个叶节点,记为
客户端 在树调节阶段为这个叶节点产生多数类标签:
于是对类别
定义第 棵树的联邦类别投票比例:
因此:
整个 棵森林的归一化类别频率:
等价地:
你的 Algorithm 4 中原来的频数 与这里仅相差常数 ,因此:
将树 中的随机因素统一写为:
包括:
假设在给定所有客户端训练数据
之后:
相互独立且同分布。
这一假设与经典随机森林理论中“每棵树由独立同分布随机向量 驱动”的框架一致。(Department of Statistics)
定义无限森林类别得分:
则:
因为:
所以:
在给定
条件下,由于不同树的随机机制独立同分布,因此:
为独立同分布有界随机变量。
根据强大数定律:
即:
证毕。
这与 Breiman 随机森林的核心收敛思路相同:独立随机树数量趋于无穷时,类别投票比例由强大数定律趋于稳定极限。(Department of Statistics)
定义无限森林的最终分类类别:
假定其唯一。
定义无限森林类别间隔:
假设:
前面已经证明对所有有限类别 :
因此存在充分大的 ,使得:
此时:
对任意
有:
所以:
即:
于是当 足够大:
所以:
定义测试分布下风险:
无限森林风险:
若对于 -几乎所有 :
则:
且:
因此根据控制收敛定理:
这就是你 Algorithm 1–4 的森林规模收敛性。
Breiman 原始随机森林理论也证明了随着树数增加,泛化误差趋于一个稳定极限。(Department of Statistics)
它证明的是:
也就是:
但是它没有证明
其中 为 Bayes 风险。
所以论文中一定要称:
森林规模收敛性
而不能直接称:
“Bayes 一致性”。
随机森林的统计一致性是另外一个更强的问题;相关理论文献专门研究了什么随机森林结构能够达到一致性。(Journal of Machine Learning Research)
这个定理我非常建议加,因为它使上一节从“极限结果”变成了“有限样本定量结果”。
由于:
并且在给定训练数据后不同树独立,所以由 Hoeffding 不等式:
一共有 个类别,根据并集界:
因此:
增加时,有限森林类别频率偏离无限森林类别频率的概率呈指数下降。
令无限森林预测类别为:
对于任意竞争类别
定义:
由于:
所以:
定义其期望:
并且:
如果有限森林错误地让类别 战胜 ,则:
因此:
Hoeffding 不等式给出:
由于:
有:
一共有 个竞争类别,所以:
这就是非常漂亮的有限森林指数收敛界。
如果你希望:
只需要:
取对数:
因此:
即可保证有限森林与无限森林预测不一致的概率至多为 。
这个定理意味着两个因素决定有限森林需要多少棵树:
越大:
越小。
如果:
很大,说明无限森林非常确定,所以少量树就可以稳定。
如果:
说明样本本身就在困难区域/决策边界附近,需要更多树。
因此你可以写:
在无限森林具有正投票间隔的条件下,有限协同联邦森林偏离无限森林预测结果的概率随森林规模 指数衰减。
这个结论比简单说“随着树增加模型会稳定”更强。
这是最具有“联邦”特色的证明。
对应 Algorithm 4。
固定某棵树:
固定其中某个叶节点:
客户端:
利用自己的全部本地数据遍历树。
到达叶节点 的样本数:
类别 的样本数:
于是客户端估计叶节点类别比例:
客户端产生多数类标签:
服务器收到:
定义:
定义客户端 在叶节点 的真实多数类别:
假设这个最大类别唯一。
定义类别间隔:
要求:
假设:
并且:
则:
进一步,如果服务器端客户端多数标签存在唯一多数类别,则服务器叶节点共识标签也几乎必然趋于该总体客户端共识类别。
因为:
根据大数定律:
因此:
根据大数定律:
同时:
因此:
由连续映射定理:
即:
真实多数类为:
其类别间隔:
因为所有:
所以当 足够大时:
对所有类别同时成立。
于是:
而:
因此:
于是:
最终几乎必然成立。
所以:
定义总体客户端多数投票数:
定义服务器目标共识类别:
假设最大值唯一。
因为:
有限,且对于每个客户端都有:
所以最终所有客户端标签同时稳定。
服务器实际票数:
最终满足:
因此:
证毕。
你的算法最终叶节点学习的不是“把所有客户端样本放在一起以后最大的类别”。
它学习的是:
数学上:
而集中式 pooled-data 分类器通常是:
两者并不相同。
这实际上很好地体现了你的 non-IID 思路:
每一个客户端首先形成自己的局部决策,再由服务器进行决策级共识,而不是直接让大样本客户端在样本数量上压倒小样本客户端。
所以你可以把这一性质称为:
联邦叶节点客户端共识一致性
或者英文:
Client-level Consensus Consistency of Federated Leaf Nodes
如果某客户端 在叶节点 中:
那么:
不存在。
在 non-IID 情况下这是完全可能出现的。
所以建议 Algorithm 4 明确增加:
服务器聚合时忽略:
定义活动客户端集合:
则:
预测票数改为:
这样你的算法和证明都会完整很多。
你的七个理论结果不是彼此孤立的,可以组织成下面这样:
↓
不同类别数、不同特征取值数造成的自由度差异被校准。
然后:
↓
如果候选集合中包含有效变量,大样本下有效变量压过纯噪声变量。
与此同时:
↓
不同客户端在不同树生长位置上的参与比例渐近达到:
同时:
↓
保持节点经验分布的一阶中心,同时增加节点级随机扰动。
因此形成具有多重随机性的单棵协同树。
之后:
↓
由强大数定律:
进一步:
以及:
同时 Hoeffding 给出:
最后树调节阶段满足:
进而:
这七个证明正好分别对应你的三类创新。
| 你的算法创新 | 对应理论证明 |
|---|---|
| WH 特征选择 | 定理 1:WH 自由度校准 |
| WH 特征选择 | 定理 2:节点特征选择一致性 |
| 协同联邦生长 | 定理 3:客户端随机排列公平性 |
| 双重随机森林 | 定理 4:Double Bootstrap 一阶保持与随机扰动 |
| 双重随机森林 | 定理 5: 森林收敛 |
| 双重随机森林 | 定理 6:有限 的 Hoeffding 指数界 |
| 联邦树调节 | 定理 7:联邦叶节点共识一致性 |
这样你的理论部分就不是简单套用传统 RF,而是每一个算法模块都有对应的理论解释。
你可以直接写成:
定理 1.
若 与类别标签 在节点 内独立,则经过式(1)的两阶段 Wilson–Hilferty 转换后:
因此不同自由度的候选特征能够在统一统计尺度下比较。
定理 2.
若随机候选集合 至少包含一个与 有关的变量,并且:
则:
定理 3.
定理 4.
同时:
定理 5.
且在唯一最大投票类别条件下:
以及:
定理 6.
定理 7.
以及:
从理论严谨性排序,我会这样评价:
定理 3 客户端排列公平性
最严格,基本没有争议。
定理 4 Bootstrap 一阶性质
严格成立。
定理 5
在树随机性条件独立同分布下严格成立,而且与经典 RF 理论高度一致。Breiman 正是用强大数定律证明随机森林随树数增加趋于稳定。(Department of Statistics)
定理 6 Hoeffding 收敛速度
非常严格,而且很适合论文。
定理 7 联邦叶共识一致性
在叶节点区域固定、有效客户端叶节点样本量趋于无穷、无类别平局条件下严格成立。
定理 1 WH 自由度校准
属于近似统计性质,因为 Wilson–Hilferty 本身是渐近近似。GUIDE 也把它明确作为 approximation 使用。(UW Computer Sciences)
定理 2 WH 特征选择一致性
可以成立,但要明确写“候选集合包含有效变量”和“离散化后仍保持关联”等正则条件;不要写成无条件 universal consistency。
你现在这 7 个证明中的定理 5 和定理 6解决的是:
也就是森林中树数量增加时的收敛性。
而真正意义上的统计一致性要求研究:
并最终证明:
第二个箭头需要进一步证明“叶节点直径趋于 0、叶节点样本量趋于无穷”等性质。随机森林一致性研究中这属于另一层更强的理论问题;已有文献专门研究随机森林及 averaging classifiers 的一致性,而并非所有随机森林构造天然都具有 universal consistency。(Journal of Machine Learning Research)
所以如果你下一步要把论文理论部分做到比较完整,最自然的第 8 个定理 就是:
而这个第 8 个证明,需要专门结合你 Algorithm 1–4 的停止生长条件、、WH 分裂机制、non-IID 假设和叶节点调节机制来推,不能直接照搬普通随机森林定理。