-
公开(公告)号:CN118133945A
公开(公告)日:2024-06-04
申请号:CN202410341649.3
申请日:2024-03-25
Applicant: 杭州电子科技大学
IPC: G06N5/02 , G06N3/042 , G06N3/096 , G06N3/0464 , G06N3/098
Abstract: 本发明公开了一种基于知识蒸馏的联邦学习场景链接预测方法,该方法首先获取真实的图数据集并进行预处理。其次利用客户端内的图神经网络和数据,进行预训练并进行模型的蒸馏,将GNN模型蒸馏到MLP中。然后模型的联邦学习训练,各个客户端上传本地训练模型参数到服务器,随后由服务器进行聚合后分发回客户端,直至全局模型收敛。最后进行客户端间及客户端内链接的预测。本发明解决各客户端模型异质性、独立客户端中训练数据不足、跨客户端进行链接预测的问题。