arXiv:2412.16144cs.LGcs.DC2024-12被引 1

提出联邦图注意力网络,低通信开销下实现隐私保护的图模型训练

FedGAT: A Privacy-Preserving Federated Approximation Algorithm for Graph Attention Networks

  • 通过预训练通信一轮,近似图注意力机制,避免每轮传递敏感信息
  • 实验显示准确率接近中心化训练,且在多客户端和不同数据分布下稳定
  • 适合需要隐私保护的社交网络或电商图学习场景

联邦学习在社交网络和电商平台的图学习中广泛应用,但隐私法规要求本地数据不得跨客户端共享。图结构自然被分割,跨客户端边的存在带来挑战:单个客户端训练需获取另一端节点的特征信息。直接保留这些边会带来巨大通信开销,而完全丢弃则降低模型性能。简单模型如图卷积网络可通过预先通信少量特征解决,但图注意力网络(GAT)的注意力权重随训练轮次动态变化,无法预先传递。本文提出联邦图注意力网络(FedGAT),用于半监督节点分类,通过可证明误差边界的方式近似GAT行为。只需一次预训练通信轮次,显著降低联邦GAT训练的通信开销。我们分析了近似误差,并研究了算法的通信开销与计算复杂度。实验表明,FedGAT在准确性上几乎达到集中式设置下的GAT表现,且对客户端数量和数据分布具有鲁棒性。

原文摘要 · Abstract (English)

Federated training methods have gained popularity for graph learning with applications including friendship graphs of social media sites and customer-merchant interaction graphs of huge online marketplaces. However, privacy regulations often require locally generated data to be stored on local clients. The graph is then naturally partitioned across clients, with no client permitted access to information stored on another. Cross-client edges arise naturally in such cases and present an interesting challenge to federated training methods, as training a graph model at one client requires feature information of nodes on the other end of cross-client edges. Attempting to retain such edges often incurs significant communication overhead, and dropping them altogether reduces model performance. In simpler models such as Graph Convolutional Networks, this can be fixed by communicating a limited amount of feature information across clients before training, but GATs (Graph Attention Networks) require additional information that cannot be pre-communicated, as it changes from training round to round. We introduce the Federated Graph Attention Network (FedGAT) algorithm for semi-supervised node classification, which approximates the behavior of GATs with provable bounds on the approximation error. FedGAT requires only one pre-training communication round, significantly reducing the communication overhead for federated GAT training. We then analyze the error in the approximation and examine the communication overhead and computational complexity of the algorithm. Experiments show that FedGAT achieves nearly the same accuracy as a GAT model in a centralised setting, and its performance is robust to the number of clients as well as data distribution.

联邦学习图神经网络隐私保护

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。