预测模型训练方法、信息预测方法及相应装置

    公开(公告)号:CN115907056A

    公开(公告)日:2023-04-04

    申请号:CN202111154399.5

    申请日:2021-09-29

    Abstract: 提供一种预测模型训练方法及装置、一种信息预测方法及装置。所述预测模型训练方法包括:向训练装置发送待训练模型,其中,所述待训练模型包括用于提取用户特征的特征提取层和用于进行信息预测的预测层;基于各训练装置提取的用户特征,将各训练装置分类到至少一个群组;接收各训练装置对所述待训练模型进行训练获得所述特征提取层对应的第一参数和所述预测层对应的第二参数;基于各训练装置的第一参数进行全局联邦聚合,得到全局联邦聚合结果,并且在所述至少一个群组的每个群组中,将群组中的各训练装置的第二参数进行群组内联邦聚合,得到群组内联邦聚合结果;向各训练装置发送全局联邦聚合结果以及对应的群组内联邦聚合结果。

Patent Agency Ranking