1.简述
GNN已经在业界有了很广泛的应用且很多领域已有SOTA的效果,但是在隐私安全日渐趋严的背景下,如何通过FL实现gnn的跨域联邦学习是一个有益的结合,本文通过联邦环境下对于Graph数据的如何在不同domain集群下进行训练,包括邻居采样、聚合和梯度回传等提出了一套完整的方案,并开源相关实现框架
2.两阶段的联邦GNN训练方式
阶段一:消息传递,下图k表示编号为k的客户端
阶段二:readout,下图中S可以表示节点或者图,根据任务类型而定
标题3.框架功能和架构
内核功能:框架在标准的分布式引擎功能基础上,通过分布式通信组件实现了多client联合训练的交互需要
ML-API