[发明专利]基于树模型的两方联合分类方法、装置、设备及介质

权利要求书

1.一种基于树模型的两方联合分类方法，各方持有部分树模型，各方持有的部分树模型包括：经双方联合训练得到的完整树模型的树形结构，以及该完整树模型的部分分裂信息以及各叶节点的权值分片；各方持有的目标对象的特征不完全相同，所述方法应用于任一方，包括：

将本方持有的目标对象的特征输入本方的部分树模型，预测得到与所述目标对象匹配的若干第一疑似叶节点；

根据所述若干第一疑似叶节点和对方预测得到的若干第二疑似叶节点，确定所述目标对象预测得到的目标叶节点；

基于本方确定出的所述目标叶节点的本方权值分片的密文、以及对方发送的该目标叶节点的对方权值分片的密文，确定目标叶节点对应的分类权值；其中，所述本方权值分片的密文由本方采用本方公钥对该目标叶节点的本方权值分片进行同态加密得到；

根据所述目标叶节点和所述分类权值确定所述目标对象的分类结果；

其中，所述根据所述若干第一疑似叶节点和对方预测得到的若干第二疑似叶节点，确定所述目标对象预测得到的目标叶节点包括：对各第一疑似叶节点进行哈希运算，得到各第一疑似叶节点的哈希值并发送给对方；对对方发送的各第二疑似叶节点的哈希值进行二次哈希，得到各第二疑似叶节点的二次哈希值并发送给对方；在对方发送的各第一疑似叶节点的二次哈希值、以及本方得到的各第二疑似叶节点的二次哈希值中，确定两方共有的二次哈希值，并将该共有的二次哈希值对应的疑似叶节点作为目标叶节点；

所述基于本方确定出的所述目标叶节点的本方权值分片的密文、以及对方发送的该目标叶节点的对方权值分片的密文，确定目标叶节点对应的分类权值，包括：对对方发送的目标叶节点的对方权值分片的密文进行拆分，得到对方权值分片的第一子分片的密文和第二子分片的密文，并将所述第二子分片的密文发送给对方；其中，本方持有所述对方权值分片的第一子分片；对对方发送的本方权值分片的第二子分片的密文进行解密，得到本方权值分片的第二子分片；基于对方权值分片的第一子分片、本方权值分片的第二子分片、以及预设的本方的部分树模型的权值，确定本方子权值；基于双方子权值，得到所述目标叶节点对应的分类权值；

所述对对方发送的目标叶节点的对方权值分片的密文进行拆分，得到对方权值分片的第一子分片的密文和第二子分片的密文，包括：

生成随机数，作为所述对方权值分片的第一子分片；

采用对方公钥对所述随机数进行同态加密，得到对方权值分片的第一子分片的密文；

基于所述对方权值分片的密文、以及对方权值分片的第一子分片的密文，确定所述对方权值分片的第二子分片的密文。

2.根据权利要求1所述的方法，所述将本方持有的目标对象的特征输入本方的部分树模型，预测得到与所述目标对象匹配的若干第一疑似叶节点，包括：

将本方持有的目标对象的特征输入本方的部分树模型，以由所述本方的部分树模型遍历该部分树模型上的节点，若遍历得到的节点具有分裂信息，则沿着该特征和分裂信息所指示遍历方向继续遍历下层节点，若遍历得到的节点不具有分裂信息，则沿着该遍历得到的节点所连接的所有下层节点的方向继续遍历节点，直至遍历得到叶节点为止。

3.根据权利要求1所述的方法，所述树模型为XGB模型。