[发明专利]多方联合决策树构建方法、设备及可读存储介质

权利要求书

1.一种多方联合决策树构建方法，其特征在于，所述多方联合决策树构建方法应用于第一设备，所述多方联合决策树构建方法包括：

获取第一样本数据和待构建树信息，并基于所述第一样本数据和所述待构建树信息，通过与各第二设备进行联邦交互，计算特征分裂增益数据；

基于所述特征分裂增益数据，确定符合预设特征分裂增益条件的目标增益分裂点集合，以构建所述待构建树信息对应的联合构建决策树；

其中，所述特征分裂增益数据包括第一特征分裂增益直方图和第二特征分裂增益直方图，所述目标增益分裂点集合包括参与方最大增益分裂点和主导方最大增益分裂点，

所述基于所述特征分裂增益数据，确定符合预设特征分裂增益条件的目标增益分裂点集合，以构建所述待构建树信息对应的联合构建决策树的步骤包括：

确定所述待构建树信息中的待构建树类型，若所述待构建树类型为节点混合类型，则基于所述第一特征分裂增益直方图和各所述第二特征分裂增益直方图，确定所述目标增益分裂点集合；

若所述待构建树类型为分层类型，则基于所述第一特征分裂增益直方图，确定所述主导方最大增益分裂点，并基于各所述第二特征分裂增益直方图，确定所述参与方最大增益分裂点；

若所述待构建树类型为参与方类型，则基于各所述第二特征分裂增益直方图，确定所述目标增益分裂点集合；

若所述待构建树类型为主导方类型，则基于所述第一特征分裂增益直方图，确定所述目标增益分裂点集合；

基于所述目标增益分裂点集合，确定当前决策树；

若所述当前决策树达到预设联邦结束条件，则将所述当前决策树作为所述联合构建决策树。

2.如权利要求1所述多方联合决策树构建方法，其特征在于，所述基于所述第一样本数据和所述待构建树信息，通过与各第二设备进行联邦交互，计算特征分裂增益数据的步骤包括：

对所述第一样本数据进行特征分箱，获得第一特征分箱数据；

获取模型残差和所述第一特征分箱数据对应的数据标签，并基于所述模型残差和所述数据标签，计算所述第一特征分箱数据对应的一阶梯度集合和二阶梯度集合；

基于所述一阶梯度集合、所述二阶梯度集合和所述待构建树信息，通过与各所述第二设备进行联邦交互以进行纵向联邦学习，计算所述特征分裂增益数据。

3.如权利要求2所述多方联合决策树构建方法，其特征在于，所述待构建树信息包括待构建树类型，

所述基于所述一阶梯度集合、所述二阶梯度集合和所述待构建树信息，通过与各所述第二设备进行联邦交互以进行纵向联邦学习，计算所述特征分裂增益数据的步骤包括：

基于所述一阶梯度集合和所述二阶梯度集合，计算所述第一特征分裂增益直方图；

若所述待构建树类型为主导方树，则将所述第一特征分裂增益直方图作为所述特征分裂增益数据；

若所述待构建树不为所述主导方树，则将所述一阶梯度集合和所述二阶梯度集合加密发送至各所述第二设备，并接收各所述第二设备基于所述一阶梯度集合和所述二阶梯度集合分别加密反馈的第二特征分裂增益直方图；

将所述第一特征分裂增益直方图和各所述第二特征分裂增益直方图作为所述特征分裂增益数据。

4.如权利要求1所述多方联合决策树构建方法，其特征在于，所述目标增益分裂点集合包括第一最大增益分裂点和第二最大增益分裂点，

所述基于所述第一特征分裂增益直方图和各所述第二特征分裂增益直方图，确定所述目标增益分裂点集合的步骤包括：

基于所述第一特征分裂增益直方图和各所述第二特征分裂增益直方图，通过预设增益计算公式确定所述第一样本数据对应的根节点的所述第一最大增益分裂点；

基于所述第一最大增益分裂点，分裂所述根节点，获得所述根节点对应的初始孩子节点；

判断所述初始孩子节点是否达到预设停止分裂条件，若所述初始孩子节点达到所述预设停止分裂条件，则判定所述初始孩子节点属于预设叶子节点类型，并将所述第一最大增益分裂点作为所述目标增益分裂点集合；

若所述初始孩子节点未达到所述预设停止分裂条件，基于所述预设增益计算公式，确定所述初始孩子节点对应的所述第二最大增益分裂点；

基于所述第二最大增益分裂点，对所述初始孩子节点进行再分裂，直至所述初始孩子节点对应的各分裂节点达到所述预设停止分裂条件，获得所述目标增益分裂点集合。