CN114169537A - 一种纵向xgboost决策树的联邦学习方法和*** - Google Patents
一种纵向xgboost决策树的联邦学习方法和*** Download PDFInfo
- Publication number
- CN114169537A CN114169537A CN202210127193.1A CN202210127193A CN114169537A CN 114169537 A CN114169537 A CN 114169537A CN 202210127193 A CN202210127193 A CN 202210127193A CN 114169537 A CN114169537 A CN 114169537A
- Authority
- CN
- China
- Prior art keywords
- participant
- node
- decision tree
- ciphertext
- participants
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Granted
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N20/00—Machine learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F21/00—Security arrangements for protecting computers, components thereof, programs or data against unauthorised activity
- G06F21/60—Protecting data
- G06F21/602—Providing cryptographic facilities or services
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04L—TRANSMISSION OF DIGITAL INFORMATION, e.g. TELEGRAPHIC COMMUNICATION
- H04L9/00—Cryptographic mechanisms or cryptographic arrangements for secret or secure communications; Network security protocols
- H04L9/008—Cryptographic mechanisms or cryptographic arrangements for secret or secure communications; Network security protocols involving homomorphic encryption
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Software Systems (AREA)
- Physics & Mathematics (AREA)
- Computer Security & Cryptography (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- Medical Informatics (AREA)
- Computer Networks & Wireless Communication (AREA)
- Evolutionary Computation (AREA)
- Data Mining & Analysis (AREA)
- Mathematical Physics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Artificial Intelligence (AREA)
- Computing Systems (AREA)
- Signal Processing (AREA)
- Health & Medical Sciences (AREA)
- Bioethics (AREA)
- General Health & Medical Sciences (AREA)
- Computer Hardware Design (AREA)
- Management, Administration, Business Operations System, And Electronic Commerce (AREA)
Abstract
本发明公开一种纵向xgboost决策树的联邦学习方法和***,涉及联邦学习和机器学习决策树技术领域,能够保证各方所持有样本的特征数据的隐私性,增强对各方特征数据的安全性。本发明的主要技术方案为:本发明提供了纵向xgboost决策树的联合训练过程和联合推理过程,在联合训练过程中计算***点以及在联合推理过程中对每个节点进行判别,联合训练过程公开的信息是每个参与方的最大***值,而不会直接泄露每个参与方的特征信息,联合推理过程的安全依赖门限同态加密方案。本发明主要应用于决策树的联合训练和联合推理过程。
Description
技术领域
本发明涉及联邦学习和机器学习决策树技术领域,尤其涉及一种纵向xgboost决策树的联邦学习方法和***。
背景技术
纵向xgboost决策树联邦学习是指当每个样本的特征信息和标签信息掌握在不同所有者处,各方联合进行xgboost决策树的训练,并且各方在训练和推理过程中不希望向其他任意方泄露关于样本的特征信息或者标签信息。
例如xgboost决策树的训练所需要的完整样本信息如下,例举表一:
ID | x1 | x2 | x3 | x4 | x5 | x6 | y |
u1 | ... | ... | ... | ... | ... | ... | ... |
u2 | ... | ... | ... | ... | ... | ... | ... |
u3 | ... | ... | ... | ... | ... | ... | ... |
其中,ID表示能唯一确定所描述对象的标识,x1到x6表示该对象的属性信息(特征信息),y表示标签信息。其中的属性信息有可能分布在不同的机构,例如机构A具有前三个特征信息和标签信息;机构B具有后三个特征信息。比如,现实中银行和保险机构可以有一批共同的客户,而银行和保险具有关于其共同客户的不同属性信息,因此不同机构间具有联合进行xgboost决策树训练的条件。联合训练的障碍来自银行和保险不愿/不能向对方透漏各自客户的信息。
目前,对于使用paillier同态加密方案设计纵向xgboost决策树联邦学习,例如SecureBoost方案,它是一种新的基于联邦学习的无损隐私保护树增强方案,具体阐述如下:
在SecureBoost方案中,拥有标签信息的参与方称为“主动方”(Active party);只拥有特征信息的参与方称为“被动方”(Passive parties);“主动方”一般在一个***中唯一存在,“被动方”可以有多个参与方,各参与方联合训练得到xgboost决策树模型。
具体的,训练过程为如下:
根据前t-1轮生成的决策树模型计算每个样本的一阶梯度gi和二阶梯度hi,例如对逻辑回归的损失函数:
这些信息包含主动方的标签信息,主动方不愿向被动方透露标签信息。
主动方生成Paillier加法同态加密方案的公钥和私钥,安全保存私钥,将公钥发送给其他被动方。主动方使用公钥加密一阶梯度gi和二阶梯度hi,并将加密后的密文<gi>和<hi>发送其他被动方,根据加法同态加密的特性,被动方可以在不知密文对应明文的情况下计算样本空间的梯度累积和。
被动方根据密文梯度信息分别计算所有特征对应的所有可能***空间的密文梯度累加值,并将该密文值发送给主动方。主动方收到被动方的密文梯度累加值后,使用私钥解密。主动方根据解密后的梯度累加值计算全体的最优***点。各被动方使用加法同态在本地计算可能的***点的梯度和。
以上,主动方根据解密后的梯度和寻找最优的***点。被动方只能被动接收主动方所发送的最佳***点的特征和阈值。相应的被动方根据接收的***特征和阈值进行样本空间的划分。从整个过程看被动方的作用是辅助主动方寻找最优的***点。由算法描述知,主动方可获取被动方的所有可能***点的梯度累加值,主动方可获取决策树模型的所有消息,包括决策树的结构,每个节点的判断条件,以及叶子节点的权重消息。
除了以上训练过程,相应的联合推理过程为:每一棵树的推理过程均是在主动方的导引下由多个参与方协助完成路径的推理。从一棵树的根节点开始,对每个需要进行决策的节点,主动方询问相应特征的所有者进行路径的判别,直到获得叶子节点的权重值。
在SecureBoost方案的训练过程中,具体的,每轮树的训练过程中,每个被动方需要向主动方发送所有可能***样本空间的梯度和,主动方可获取大量数据。例如:对一个节点,样本集合中有5个元素,那么主动方可能获得如下信息:
表一
上面例子可得:o1<1;1=<o2,o3<2;2=<o4,o5<3。主动方可获取大量对同一个集合的划分信息,其实就是对集合中每个元素的取值的判别信息,从中泄露每个样本的每个特征的取值范围,划分的粒度越小,信息泄露越严重。
在SecureBoost方案的联合推理过程中,被动方对一个节点的判别会直接向主动方泄露该特征的范围信息,由于xgboost由多棵树组成,每棵树中对一个特征的判别可以分布在多层,因此该方案的推理过程对被动方的特征信息泄露严重。
例如,一个节点的对年龄的判别结果是小于30,一个节点对年龄的判别是大于或等于20,那么主动方可直接获得的信息是被推理的对象的年龄是20至30岁之间的,随着树的层数和数量的增加被推理对象的信息泄露愈发严重,当一个特征的取值范围较小时可能完全被泄露。
以上,在SecureBoost方案的训练过程和联合推理过程中,都是存在被动方的特征信息泄露严重的情况,那么在纵向xgboost决策树联邦学习过程中,是难以保护各方样本的特征信息的隐私安全性的。
发明内容
有鉴于此,本发明提供一种纵向xgboost决策树的联邦学习方法和***,主要目的在于保证各方所持有样本的特征数据的隐私性,增强对各方特征数据的安全性。
为了达到上述目的,本发明主要提供如下技术方案:
本发明第一方面提供了一种纵向xgboost决策树的联邦学习方法,该方法包括:
根据前t-1轮生成的决策树模型,参与方计算每个样本的一阶梯度和二阶梯度;
一个所述参与方将每个样本的一阶梯度和二阶梯度发送给其他多个参与方;
每个所述参与方在本地计算其拥有的所有特征对应的多个***值并从中选择最大***值,得到每个所述参与方自身对应的最大***值;
通过比较每个所述参与方自身对应的最大***值,选择一个全局最大***值;
根据所述全局最大***值确定目标***点,完成对节点样本空间进行划分,将划分后的节点样本空间对应的集合信息同步给每个所述参与方,以使得每个所述参与方共同获知每棵决策树的全局结构信息,用于完成对第t轮生成的k棵决策树模型的训练过程;
每个所述参与方分别对所述k棵决策树模型中所有需要其负责判别的节点进行本地判别,利用门限同态加密方案保护各个所述参与方的交互过程,以完成纵向联邦决策树的联合推理过程。
在本申请的一些变更实施方式中,所述每棵决策树的全局结构信息,包括:每个节点对应的属主、不同节点之间的关联,所述属主为对节点进行判别的参与方;每个节点具有唯一编号,所述编号为由树的序号和节点所述树中的位置进行表征。
在本申请的一些变更实施方式中,所述通过比较每个所述参与方自身对应的最大***值,选择一个全局最大***值,包括:
在每个所述参与方确定自身对应的最大***值之后,每个所述参与方将自身对应的最大***值发送给其他参与方;
所有的参与方经比较多个所述最大***值,从中选择一个全局最大***值。
在本申请的一些变更实施方式中,所述根据所述全局最大***值确定目标***点,完成对节点样本空间进行划分,包括:
根据所述全局最大***值,确定对应的属主;
根据所述全局最大***值和所述属主,确定目标***点的特征和阈值;
根据所述特征和阈值对节点样本空间进行划分。
在本申请的一些变更实施方式中,所述每个所述参与方分别对所述k棵决策树模型中所有需要其负责判别的节点进行本地判别,利用门限同态加密方案保护各个所述参与方的交互过程,以完成纵向联邦决策树的联合推理过程,包括:
每个所述参与方获取对应的判别输出表;
根据所述判别输出表,每个所述参与方计算对应所负责节点的判别边长,得到所负责节点对应的目标边长;
每个所述参与方对所负责节点的所述目标边长使用门限同态加密方案的公钥进行加密,得到所负责节点对应的密文边长;
每个所述参与方向其他参与方公开包含有所负责节点的节点编号和密文边长的对照表;
根据所述对照表,每个叶子节点对应的属主收集从根节点达到叶子节点对应途径的所有节点的密文边长,使用同态加法计算对应的密文边长和;
对于每个叶子节点,所述叶子节点的属主会生成一个标志密文和隐藏权重密文的二元组;
每个所述参与方对其所负责的所有叶子节点的标志密文和隐藏权重密文的二元组进行随机置换后公开;
每个所述参与方对公开的二元组中的标志密文进行分布式解密,得到解密结果;
每个所述参与方将标志密文解密结果为0的元组中的隐藏权重密文使用同态加法计算隐藏权重密文之和,得到推理结果密文并公开;
根据公开的所述推理结果密文,每个所述参与方获得样本对应的推理结果,以完成纵向联邦决策树的联合推理过程。
在本申请的一些变更实施方式中,所述根据公开的所述推理结果密文,每个所述参与方获得样本对应的推理结果,包括:
每个所述参与方对所述推理结果密文进行分布式解密,将解密的中间结果发送需要获得样本推理结果的参与方;
根据所述中间结果,所述需要获得样本推理结果的参与方获得样本的推理结果。
在本申请的一些变更实施方式中,使用的所述门限同态加密方案为可使用门限somewhat同态加密的技术。
本发明第二方面提供了一种纵向xgboost决策树的联邦学习***,应用于执行如上所述的方法,所述***包括:多个参与方,每个所述参与方拥有特征信息和标签信息;多个所述参与方联合进行xgboost决策树的训练和推理过程。
本发明第三方面提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如上所述的纵向xgboost决策树的联邦学习方法
本发明第四方面提供了一种电子设备,包括:存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如上所述的纵向xgboost决策树的联邦学习方法。
借由上述技术方案,本发明提供的技术方案至少具有下列优点:
本发明提供的一种纵向xgboost决策树的联邦学习方法和***。本发明提供了纵向xgboost决策树的联合训练过程和联合推理过程,在联合训练过程中计算***点以及在联合推理过程中对每个节点进行判别,联合训练过程公开的信息是每个参与方的最大***值,而不会直接泄露每个参与方的特征信息,联合推理过程的安全依赖门限同态加密方案。相较于现有技术,解决了因现有的纵向联邦决策树方案难以保护各方样本的特征信息的隐私安全性的问题,本发明能够在联合训练和联合推理过程中保证各方所持有样本的特征数据的隐私性,增强对各方特征数据的安全性。
上述说明仅是本发明技术方案的概述,为了能够更清楚了解本发明的技术手段,而可依照说明书的内容予以实施,并且为了让本发明的上述和其它目的、特征和优点能够更明显易懂,以下特举本发明的具体实施方式。
附图说明
通过阅读下文优选实施方式的详细描述,各种其他的优点和益处对于本领域普通技术人员将变得清楚明了。附图仅用于示出优选实施方式的目的,而并不认为是对本发明的限制。而且在整个附图中,用相同的参考符号表示相同的部件。在附图中:
图1为本发明实施例提供的一种纵向xgboost决策树的联邦学习方法流程图;
图2为本发明实施例提供的对纵向xgboost决策树的联邦学习的联合推理过程的流程图。
具体实施方式
下面将参照附图更详细地描述本发明的示例性实施例。虽然附图中显示了本发明的示例性实施例,然而应当理解,可以以各种形式实现本发明而不应被这里阐述的实施例所限制。相反,提供这些实施例是为了能够更透彻地理解本发明,并且能够将本发明的范围完整的传达给本领域的技术人员。
本发明实施例提供了一种纵向xgboost决策树的联邦学习方法,该方法由多个参与方联合训练和联合推理,这些多个参与方组成了一个实现该方法对应的***,即一种纵向xgboost决策树的联邦学习***。具体的,该***包括:多个参与方,每个参与方拥有特征信息和标签信息,多个参与方联合进行xgboost决策树的训练和推理过程。
对于本发明实施例提供的纵向xgboost决策树的联邦学习方法,如图1所示,对此本发明实施例提供以下具体步骤:
101、根据前t-1轮生成的决策树模型,参与方计算每个样本的一阶梯度和二阶梯度。
其中,参与方为拥有标签信息和特征信息。
在本发明实施例中,对于第t轮训练进行解释说明,假定标签信息的所有者(即标识为参与方)通过安全的联合推理过程获得所有样本的推理结果,标签信息的所有者通过所有样本的推理结果和标签信息计算得到每个样本的一阶梯度和二阶梯度。
102、一个参与方将每个样本的一阶梯度和二阶梯度发送给其他多个参与方。
103、对于一个决策树节点的训练,每个参与方在本地计算其拥有的所有特征对应的多个***值并从中选择最大***值,得到每个参与方自身对应的最大***值。
其中,Lsplit为***值,IL为该节点样本空间***后的左孩子集合,IR为该节点样本空间***后的右孩子集合,I为该节点样本空间集合。每个参与方从自身得到多个***值中选取最大***值Lsplit,max。
104、通过比较所有参与方的最大***值,选择其中的最大值作为上述决策树节点的全局最大***值。
在本发明实施例中,进一步的,本步骤细化解释说明为:在每个参与方确定自身对应的最大***值之后,每个参与方将自身对应的最大***值发送给其他参与方;所有的参与方经比较多个最大***值,选择其中的最大值作为一个全局最大***值。
105、根据全局最大***值确定决策树节点的***信息,完成对节点样本空间的划分,将划分后的节点样本空间对应的集合信息同步给每个参与方,以使得每个参与方共同获知每棵决策树的全局结构信息,用于完成对第t轮生成的k棵决策树模型的训练过程。
在本发明实施例中,进一步的,本步骤细化解释说明为:根据全局最大***值,确定对应的属主,即提供该节点的全局最大***的参与方;根据全局最大***值和属主,确定决策树节点的的***特征和阈值;根据特征和阈值对节点样本空间进行划分。***点指对样本集合进行划分的特征和阈值。
进一步的,对于节点***后下层节点成为叶子节点的情况,该参与方不发送***后的集合信息,而是计算叶子节点的权重值,仅告知其他参与方分支已达到叶子节点。
需要说明的是,将划分后的节点样本空间对应的集合信息同步给每个参与方,以使得每个参与方共同获知每棵决策树的全局结构信息。具体的,每棵决策树的全局结构信息,包括:每个节点对应的属主、不同节点之间的关联,该属主为对节点进行判别的参与方;每个节点具有唯一编号,编号为由树的序号和节点树中的位置进行表征。
对于本发明实施例,以上步骤101-105为决策树的训练过程,每个参与方共同知悉每棵决策树的全局结构信息,每个参与方在每个决策树节点***信息的寻找过程中仅发送一次本地可能***点的最大***值,用于比较出全局最大***值,而没有发送任何与本地特征相关的信息,如:每个参与方秘密保存各自的节点判别信息,包括所负责的节点的判别特征和阈值。
106、每个参与方分别对k棵决策树模型中所有需要其负责判别的节点进行本地判别,利用门限同态加密方案保护各个参与方的交互过程,以完成纵向联邦决策树的联合推理过程。
对于本发明实施例,采用的门限同态加密方案,是指多个参与方共同使用各自的私钥对同一个密文进行解密的同态加密方案。同态加密方案一般由:密钥生成,加密,解密,密文加法,密文乘法等算法组成;门限同态加密方案则要求密钥生成和解密算法具有分布式特性,而加密、密文加法和密文乘法保持不变。本发明实施例仅使用门限somewhat同态加密。
在本发明实施例中,联合推理过程的安全依赖门限同态加密方案,每个参与方分别对k棵决策树模型中所有需要其负责判别的节点进行本地判别,利用门限同态加密方案保护各个参与方的交互过程,以保证各方所持有样本的特征数据的隐私性。
以上,本发明实施例提供的一种纵向xgboost决策树的联邦学习方法。本发明实施例提供了纵向xgboost决策树的联合训练过程和联合推理过程,在联合训练过程中计算***点以及在联合推理过程中对每个节点进行判别,联合训练过程公开的信息是每个参与方的最大***值,而不会直接泄露每个参与方的特征信息,联合推理过程的安全依赖门限同态加密方案。相较于现有技术,解决了因现有的纵向联邦决策树方案难以保护各方样本的特征信息的隐私安全性的问题,本发明实施例能够在联合训练和联合推理过程中保证各方所持有样本的特征数据的隐私性,增强对各方特征数据的安全性。
为了对上述实施例的步骤106做出更加详细的说明,本发明实施例还进一步地对纵向xgboost决策树的联邦学习的联合推理过程进行解释说明,如图2所示,对于一个样本的联合推理,本发明实施例提供以下具体步骤:
首先,各参与方联合训练获得k棵决策树模型,所有参与方联合进行门限同态加密方案的实例的初始化,各自安全生成并保存私钥,联合生成门限同态加密方案公钥。然后,利用门限同态加密方案保护各个参与方的交互过程,执行如下步骤201-210。
201、每个参与方获取对应的判别输出表。
在本发明实施例中,决策树模型中的每个节点的属主(即对该节点进行判别的参与方)在本地使用相应的判别特征和阈值对该样本的特征取值进行判别,特征取值小于阈值输出0,否则输出1。
每个参与方分别对k棵决策树模型中所有需要其负责判别的节点进行本地判别。每个参与方均可得到一张判别输出表,如下表二。
表二
202、根据判别输出表,每个参与方计算对应所负责节点的判别边长,得到所负责节点对应的目标边长。
在本发明实施例中,结合表二进行解释说明如下:
每个参与方分别计算每个所负责节点的“判别边长”得到对应的目标边长,一个判别节点的判别边长等于判别结果乘以节点权重,节点权重等于以2为基数的决策树的深度减去节点的层数次幂。决策树中节点的层数的设置为从根节点开始递增,即根节点层数设置为第0层,一个父节点***后子节点的层数等于父节点层数加1。节点权重的设置为上层节点的权重是下层节点的2倍,最底层判别节点的权重为1;在深度为d决策树中,层数为l的节点的权重等于2d-l。决策树中父节点到左孩子节点的边长设置为0,到右孩子节点的边长设置为子节点的节点权重;那么对于每个叶子节点,“路程”定义为从叶子节点到根节点的路径上的所有边长之和。一棵树中每个叶子节点的路程是固定且唯一的。
203、每个参与方对所负责节点的目标边长使用门限同态加密方案的公钥进行加密,得到所负责节点对应的密文边长。
在本发明实施例中,使用的门限同态加密方案为可使用门限somewhat同态加密。每个参与方将所负责的每个节点的“目标边长”使用门限同态加密方案的公钥进行加密得到“密文边长”。
204、每个参与方向其他参与方公开包含有所负责节点的节点编号和密文边长的对照表。
对于本发明实施例,例举每个参与方公开所负责节点的节点编号和密文边长的对照表,如下表三。
表三
205、根据对照表,每个叶子节点对应的属主收集从根节点达到叶子节点对应途径的所有节点的密文边长,使用同态加法计算对应的密文边长和。
结合每个参与方所公开的对照表,每个叶子节点的属主收集从根节点到达该叶子节点途经的所有节点的密文边长,然后使用同态加法计算途经节点的密文边长和。
206、对于每个叶子节点,叶子节点的属主会生成一个标志密文和隐藏权重密文的二元组。
在本发明实施例中,对于一个叶子节点,该叶子节点的属主生成一个标志密文和隐藏权重密文二元组,具体实施方法包括如下:
随机生成两个随机数,使用同态操作计算得到途经节点边长和减去叶子节点路程再乘以一个随机数的密文,使用同态操作计算得到途经节点边长和减去叶子节点路程再乘以另一个随机数的结果再加上叶子节点的权重的密文。该过程中涉及到密文的加减法和乘法均为同态加密方案定义的操作算法。
例如,设一个叶子节点的路程为c,路程的密文为<c>,途经节点的密文边长和为<d>,所选取的随机数分别为R1和R2,随机数的密文分别为<R1>和<R2>,该叶子节点的权重为w,密文权重为<w>;
通过同态计算可得:
标志密文=(<c>-<d>)*<R1>=<(c-d)*R1>,
隐藏权重密文=(<c>-<d>)*<R2>+<w>=<(c-d)*R2+w>。
这个过程相当于所有参与方以密文的形式联合遍历决策树的所有路径。
需要说明的是,根据以上公式分析,对于一个样本的推理过程中,当一个叶子节点的“路程”与途经节点的边长之和相等时,表明一棵树的推理输出为该叶子节点,此时标志密文的解密结果为0,隐藏权重密文的解密结果等于该叶子节点的权重值。当叶子节点的“路程”与途经节点的边长之和不相等时,标志密文的解密结果为随机值,不会泄露途经节点的边长之和,隐藏权重密文的解密结果也为随机值,不会泄露叶子节点的权重。
207、每个参与方对其所负责的所有叶子节点的标志密文和隐藏权重密文的二元组进行随机置换后公开。
具体的,所有参与方按照一定顺序依次对所有的标志密文和隐藏权重密文元组进行密文更新和元组顺序随机置换。一个参与方获取所有的标志密文和隐藏权重密文元组后将每个密文加上0的密文,也就是使得原密文发生变化,但是解密结果不发生变化。再对元组的顺序进行随机置换后公开,另一个参与方接着对新的元组进行更新和置换。
对所有叶子节点的标志密文和隐藏权重密文元组进行密文更新和元组顺序随机置换的目的是,隐藏每棵树的输出来源,避免泄露输出路径。
208、每个参与方对公开的二元组中的标志密文进行分布式解密,得到解密结果。
具体的,所有参与方对完成更新和置换后的所有元组中的标志密文进行分布式解密,所有参与方或需要样本推理结果的参与方可获得解密结果。
209、每个参与方将标志密文解密结果为0的元组中的隐藏权重密文使用同态加法计算隐藏权重密文之和,得到推理结果密文并公开。
210、根据公开的推理结果密文,每个参与方获得样本对应的推理结果,以完成纵向联邦决策树的联合推理过程。
在本发明实施例中,具体的为,每个参与方对推理结果密文进行分布式解密,将解密的中间结果发送需要获得样本推理结果的参与方,根据中间结果,需要获得样本推理结果的参与方获得样本的推理结果。
以上,本发明实施例提供的对纵向xgboost决策树的联邦学习的联合推理过程,使用门限同态加密方案保护交互过程,仅当所有成员同步解密时才能知悉密文信息,使得联合训练和联合推理过程中均***露决策的中间结果,以此保护各方样本的特征信息。
下面,进一步的,本发明实施例还提供了对决策树的联合推理过程安全性分析,以下以一个更一般的模型,论证树的联合推理过程中树的决策路径应当保密,进一步说明树的每个节点的判别信息或者权重信息应当保密。
假设已知一棵决策树,树的每个节点的判别信息(判别特征和阈值)只由持有该特征的参与方知悉,所有参与方仅知道树的结构,也就是树的每个节点的属主和节点间连接关系。
决策树的推理过程为一个节点的属主直接将判别结果发送下一层分支的属主,最终将叶子节点的权重信息发送给“主动方”(推理结果的获取方)。
该模型在初始阶段保护决策树中的每个节点信息,但在推理过程中没有保护推理路径信息。更通用的模型为以某种方式“主动方”只能获取最终的推理结果,而不知道结果是哪些叶子节点的和。
联合决策树推理模型:各参与方已知森林中每棵树的节点属主和节点间连接关系,拥有标签信息的主动方可获得叶子节点权重(保护标签),推理过程为主动方仅获取推理结果(所有树的输出叶子节点的和)。
对该模型的攻击方法如下:
前提条件,主动方已知一个全部特征的样本集合;该前提条件具有合理性,主动方可以原本具有一个具有全部特征的样本子集,但其拥有的数据大部分具有属性缺失而需要其他参与方联合训练模型;主动方可随着时间的推移不断收集关于样本的其他属性信息;主动方可不断向其他参与方安插其构造的样本信息。
主动方利用一个其已知全部特征的样本集合进行联合决策树的推理,并获得相应的推理结果。
1、恢复推理路径
主动方可根据推理结果和每棵树的叶子节点权重恢复部分树的推理路径;推理结果是所有树的输出叶子节点的累加和,因此某些结果可能具有唯一的组合,因而可获取多棵树的推理路径。主动方自有节点可以用于排除不正确的路径或者确认可能的路径。
2、恢复路径上节点的判别信息
主动方已获得一些样本的推理路径,对于两个通过同一个节点的样本,计算该节点的所有可能特征的差异度。差异度定义为两个样本的同一个特征的样本值的差与取值范围的比例。如果两个样本在共同经过点的判断走向相同,则认为该节点的特征差异度最小的特征为判别特征的概率更大;如果走向相反,则认为在该节点的特征差异度最大的特征为判别特征的概率更大;通过多个经过同一个节点的样本之间的交叉比对,确定一个节点的判别特征。设置一个节点的判别阈值为通过该节点的左向样本的最大值与右向样本的最小值的均值。
由上述两步可逐渐恢复决策树的完整信息;新增的样本可不断完善模型恢复的正确性和准确度。决策树模型一旦泄露,则过往和未来的经过该决策树的样本特征都会通过推理过程被主动方获取,相当于不具备“前向和后向安全性”。
由以上分析可得:决策树的模型信息必须由各参与方各自秘密保存,即使是叶子节点的信息也不能泄露;推理过程中不能泄露决策的中间结果,即不能直接公开每棵树的预测输出和每个节点的判别走向。
综上所述,本发明实施例提供的纵向xgboost决策树的联邦学习方法,使得持有相同对象的不同特征的多个机构可以联合进行xgboost决策的训练和推理。该方案主要保证各方所持有样本的特征数据的隐私性,而不保护样本的标签数据,通过此种设计达到安全性和性能的兼顾,相比于已有的纵向联邦决策树方案如secureboost,本方案增强对各方特征数据的安全性,相比于使用通用安全多方计算的方案,本方案计算和通讯效率更优。
本方案的特性:各方仅知悉各自负责节点的信息,决策树的模型信息必须由各参与方各自秘密保存,即使是叶子节点的信息也不向其他方泄露;训练过程和推理过程的主要计算工作由各方本地完成;使用门限同态加密方案保护交互过程,仅当所有成员同步解密时才能知悉密文信息,使得训练和推理过程中均***露决策的中间结果,以此保护各方样本的特征信息。
本发明实施例提供了一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如上所述的纵向xgboost决策树的联邦学习方法。
本发明实施例提供了一种电子设备,包括:存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如上所述的纵向xgboost决策树的联邦学习方法。
本申请是参照根据本申请实施例的方法、设备(***)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器,使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。
在一个典型的配置中,设备包括一个或多个处理器(CPU)、存储器和总线。设备还可以包括输入/输出接口、网络接口等。
存储器可能包括计算机可读介质中的非永久性存储器,随机存取存储器(RAM)和/或非易失性内存等形式,如只读存储器(ROM)或闪存(flash RAM),存储器包括至少一个存储芯片。存储器是计算机可读介质的示例。
计算机可读介质包括永久性和非永久性、可移动和非可移动媒体可以由任何方法或技术来实现信息存储。信息可以是计算机可读指令、数据结构、程序的模块或其他数据。计算机的存储介质的例子包括,但不限于相变内存 (PRAM)、静态随机存取存储器 (SRAM)、动态随机存取存储器 (DRAM)、其他类型的随机存取存储器 (RAM)、只读存储器 (ROM)、电可擦除可编程只读存储器 (EEPROM)、快闪记忆体或其他内存技术、只读光盘只读存储器(CD-ROM)、数字多功能光盘 (DVD) 或其他光学存储、磁盒式磁带,磁带磁磁盘存储或其他磁性存储设备或任何其他非传输介质,可用于存储可以被计算设备访问的信息。按照本文中的界定,计算机可读介质不包括暂存电脑可读媒体 (transitory media),如调制的数据信号和载波。
本领域技术人员应明白,本申请的实施例可提供为方法、***或计算机程序产品。因此,本申请可采用完全硬件实施例、完全软件实施例或结合软件和硬件方面的实施例的形式。而且,本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。
还需要说明的是,术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含,从而使得包括一系列要素的过程、方法、商品或者设备不仅包括那些要素,而且还包括没有明确列出的其他要素,或者是还包括为这种过程、方法、商品或者设备所固有的要素。在没有更多限制的情况下,由语句“包括一个……”限定的要素,并不排除在包括要素的过程、方法、商品或者设备中还存在另外的相同要素。
以上仅为本申请的实施例而已,并不用于限制本申请。对于本领域技术人员来说,本申请可以有各种更改和变化。凡在本申请的精神和原理之内所作的任何修改、等同替换、改进等,均应包含在本申请的权利要求范围之内。
Claims (10)
1.一种纵向xgboost决策树的联邦学习方法,其特征在于,所述方法包括:
根据前t-1轮生成的决策树模型,参与方计算每个样本的一阶梯度和二阶梯度;
一个所述参与方将每个样本的一阶梯度和二阶梯度发送给其他多个参与方;
每个所述参与方在本地计算其拥有的所有特征对应的多个***值并从中选择最大***值,得到每个所述参与方自身对应的最大***值;
通过比较每个所述参与方自身对应的最大***值,选择一个全局最大***值;
根据所述全局最大***值确定目标***点,完成对节点样本空间进行划分,将划分后的节点样本空间对应的集合信息同步给每个所述参与方,以使得每个所述参与方共同获知每棵决策树的全局结构信息,用于完成对第t轮生成的k棵决策树模型的训练过程;
每个所述参与方分别对所述k棵决策树模型中所有需要其负责判别的节点进行本地判别,利用门限同态加密方案保护各个所述参与方的交互过程,以完成纵向联邦决策树的联合推理过程。
2.根据权利要求1所述的方法,其特征在于,所述每棵决策树的全局结构信息,包括:每个节点对应的属主、不同节点之间的关联,所述属主为对节点进行判别的参与方;每个节点具有唯一编号,所述编号为由树的序号和节点所述树中的位置进行表征。
3.根据权利要求1所述的方法,其特征在于,所述通过比较每个所述参与方自身对应的最大***值,选择一个全局最大***值,包括:
在每个所述参与方确定自身对应的最大***值之后,每个所述参与方将自身对应的最大***值发送给其他参与方;
所有的参与方经比较多个所述最大***值,从中选择一个全局最大***值。
4.根据权利要求1所述的方法,其特征在于,所述根据所述全局最大***值确定目标***点,完成对节点样本空间进行划分,包括:
根据所述全局最大***值,确定对应的属主;
根据所述全局最大***值和所述属主,确定目标***点的特征和阈值;
根据所述特征和阈值对节点样本空间进行划分。
5.根据权利要求1所述的方法,其特征在于,所述每个所述参与方分别对所述k棵决策树模型中所有需要其负责判别的节点进行本地判别,利用门限同态加密方案保护各个所述参与方的交互过程,以完成纵向联邦决策树的联合推理过程,包括:
每个所述参与方获取对应的判别输出表;
根据所述判别输出表,每个所述参与方计算对应所负责节点的判别边长,得到所负责节点对应的目标边长;
每个所述参与方对所负责节点的所述目标边长使用门限同态加密方案的公钥进行加密,得到所负责节点对应的密文边长;
每个所述参与方向其他参与方公开包含有所负责节点的节点编号和密文边长的对照表;
根据所述对照表,每个叶子节点对应的属主收集从根节点达到叶子节点对应途径的所有节点的密文边长,使用同态加法计算对应的密文边长和;
对于每个叶子节点,所述叶子节点的属主会生成一个标志密文和隐藏权重密文的二元组;
每个所述参与方对其所负责的所有叶子节点的标志密文和隐藏权重密文的二元组进行随机置换后公开;
每个所述参与方对公开的二元组中的标志密文进行分布式解密,得到解密结果;
每个所述参与方将标志密文解密结果为0的元组中的隐藏权重密文使用同态加法计算隐藏权重密文之和,得到推理结果密文并公开;
根据公开的所述推理结果密文,每个所述参与方获得样本对应的推理结果,以完成纵向联邦决策树的联合推理过程。
6.根据权利要求5所述的方法,其特征在于,所述根据公开的所述推理结果密文,每个所述参与方获得样本对应的推理结果,包括:
每个所述参与方对所述推理结果密文进行分布式解密,将解密的中间结果发送需要获得样本推理结果的参与方;
根据所述中间结果,所述需要获得样本推理结果的参与方获得样本的推理结果。
7.根据权利要求5或6所述的方法,其特征在于,使用的所述门限同态加密方案为可使用门限somewhat同态加密的技术。
8.一种纵向xgboost决策树的联邦学习***,其特征在于,应用于执行如权利要求1至7中任一项所述的方法,所述***包括:多个参与方,每个所述参与方拥有特征信息和标签信息;多个所述参与方联合进行xgboost决策树的训练和推理过程。
9.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如权利要求1-7中任一项所述的纵向xgboost决策树的联邦学习方法。
10.一种电子设备,其特征在于,包括:存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如权利要求1-7中任一项所述的纵向xgboost决策树的联邦学习方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210127193.1A CN114169537B (zh) | 2022-02-11 | 2022-02-11 | 一种纵向xgboost决策树的联邦学习方法和*** |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210127193.1A CN114169537B (zh) | 2022-02-11 | 2022-02-11 | 一种纵向xgboost决策树的联邦学习方法和*** |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114169537A true CN114169537A (zh) | 2022-03-11 |
CN114169537B CN114169537B (zh) | 2022-05-06 |
Family
ID=80489689
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210127193.1A Active CN114169537B (zh) | 2022-02-11 | 2022-02-11 | 一种纵向xgboost决策树的联邦学习方法和*** |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114169537B (zh) |
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116208316A (zh) * | 2023-04-27 | 2023-06-02 | 蓝象智联(杭州)科技有限公司 | 节约存储空间的半同态加密方法、装置及存储介质 |
WO2023211081A1 (en) * | 2022-04-27 | 2023-11-02 | Samsung Electronics Co., Ltd. | Optimal split federated learning in wireless network |
CN117675411A (zh) * | 2024-01-31 | 2024-03-08 | 智慧眼科技股份有限公司 | 一种基于纵向XGBoost算法的全局模型获取方法及*** |
Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20180322516A1 (en) * | 2017-05-08 | 2018-11-08 | Beijing Baidu Netcom Science And Technology Co., Ltd. | Quality evaluation method, apparatus and device, and computer readable storage medium |
CN111598186A (zh) * | 2020-06-05 | 2020-08-28 | 腾讯科技(深圳)有限公司 | 基于纵向联邦学习的决策模型训练方法、预测方法及装置 |
CN113657471A (zh) * | 2021-07-30 | 2021-11-16 | 深圳前海微众银行股份有限公司 | 多分类梯度提升树的构建方法、装置及电子设备 |
CN113688999A (zh) * | 2021-08-23 | 2021-11-23 | 神州融安科技(北京)有限公司 | 一种横向联邦xgboost决策树的训练方法 |
-
2022
- 2022-02-11 CN CN202210127193.1A patent/CN114169537B/zh active Active
Patent Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20180322516A1 (en) * | 2017-05-08 | 2018-11-08 | Beijing Baidu Netcom Science And Technology Co., Ltd. | Quality evaluation method, apparatus and device, and computer readable storage medium |
CN111598186A (zh) * | 2020-06-05 | 2020-08-28 | 腾讯科技(深圳)有限公司 | 基于纵向联邦学习的决策模型训练方法、预测方法及装置 |
CN113657471A (zh) * | 2021-07-30 | 2021-11-16 | 深圳前海微众银行股份有限公司 | 多分类梯度提升树的构建方法、装置及电子设备 |
CN113688999A (zh) * | 2021-08-23 | 2021-11-23 | 神州融安科技(北京)有限公司 | 一种横向联邦xgboost决策树的训练方法 |
Cited By (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2023211081A1 (en) * | 2022-04-27 | 2023-11-02 | Samsung Electronics Co., Ltd. | Optimal split federated learning in wireless network |
CN116208316A (zh) * | 2023-04-27 | 2023-06-02 | 蓝象智联(杭州)科技有限公司 | 节约存储空间的半同态加密方法、装置及存储介质 |
CN116208316B (zh) * | 2023-04-27 | 2023-07-18 | 蓝象智联(杭州)科技有限公司 | 节约存储空间的半同态加密方法、装置及存储介质 |
CN117675411A (zh) * | 2024-01-31 | 2024-03-08 | 智慧眼科技股份有限公司 | 一种基于纵向XGBoost算法的全局模型获取方法及*** |
CN117675411B (zh) * | 2024-01-31 | 2024-04-26 | 智慧眼科技股份有限公司 | 一种基于纵向XGBoost算法的全局模型获取方法及*** |
Also Published As
Publication number | Publication date |
---|---|
CN114169537B (zh) | 2022-05-06 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114169537B (zh) | 一种纵向xgboost决策树的联邦学习方法和*** | |
EP3506547A1 (en) | Providing security against user collusion in data analytics using random group selection | |
Blanton et al. | Secure and efficient outsourcing of sequence comparisons | |
Huang et al. | Achieving accountable and efficient data sharing in industrial internet of things | |
EP3024169B1 (en) | System and method for matching data sets while maintaining privacy of each data set | |
CN113688999B (zh) | 一种横向联邦xgboost决策树的训练方法 | |
CN111526217A (zh) | 一种区块链中的共识方法和*** | |
CN111639367B (zh) | 基于树模型的两方联合分类方法、装置、设备及介质 | |
Benzekki et al. | A secure cloud computing architecture using homomorphic encryption | |
US20230359631A1 (en) | Updatable private set intersection | |
CN104919754A (zh) | 篡改检测装置、篡改检测方法、以及程序 | |
Erkin et al. | Privacy-preserving distributed clustering | |
EP4000216B1 (en) | Cryptographic pseudonym mapping method, computer system, computer program and computer-readable medium | |
Xue et al. | Secure and privacy-preserving decision tree classification with lower complexity | |
CN115905633A (zh) | 一种隐私保护的图相似性检索方法及*** | |
CN116167868A (zh) | 基于隐私计算的风险识别方法、装置、设备以及存储介质 | |
Erkin et al. | Privacy-preserving user clustering in a social network | |
Mashhadi | How to fairly share multiple secrets stage by stage | |
Rauthan | Fully homomorphic encryption: A case study | |
Trieu et al. | Multiparty Private Set Intersection Cardinality and Its Applications. | |
CN112765570A (zh) | 一个支持数据转移的基于身份可证明数据持有方法 | |
Meraouche et al. | Learning multi-party adversarial encryption and its application to secret sharing | |
Kamal et al. | Searchable encryption using secret-sharing scheme for multiple keyword search using conjunctive and disjunctive searching | |
JP5486519B2 (ja) | 検索システム、判定装置、ベクトル構成装置、その方法及びプログラム | |
Guan et al. | EPGQ: Efficient and private feature-based group nearest neighbor query over road networks |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant | ||
TR01 | Transfer of patent right |
Effective date of registration: 20230608 Address after: Unit 515, floor 5, building 1, No. a 12, Zhongguancun South Street, Haidian District, Beijing 100081 Patentee after: Shenzhou Rongan digital technology (Beijing) Co.,Ltd. Address before: 100081 Room 408, Huantai building, 12 Zhongguancun South Street, Haidian District, Beijing Patentee before: SHENZHOU RONGAN TECHNOLOGY (BEIJING) CO.,LTD. |
|
TR01 | Transfer of patent right |