A Federated Learning Approach for Graph Convolutional Neural Networks

In this paper we cast the problem of training a graph neural network based on labeled graph data in a "federated learning" scenario where different agents have access to data from a subset of the network nodes. The learning problem is not decomposable, therefore it does not lend itself to...

Full description

Saved in:
Bibliographic Details
Published inProceedings of the IEEE Sensor Array and Multichannel Signal Processing Workshop pp. 1 - 5
Main Authors Campbell, Andrew, Liu, Hang, Scaglione, Anna, Wu, Tong
Format Conference Proceeding
LanguageEnglish
Published IEEE 08.07.2024
Subjects
Computational modeling
Data models
Distributed databases
Federated learning
Graph convolutional neural network
multi-agent federated learning
Numerical models
Signal processing algorithms
Training
Online AccessGet full text

Cover

More Information
Summary:In this paper we cast the problem of training a graph neural network based on labeled graph data in a "federated learning" scenario where different agents have access to data from a subset of the network nodes. The learning problem is not decomposable, therefore it does not lend itself to a straightforward mapping onto a distributed multi-agent protocol. We propose a multi-agent federated learning scheme which leverages the local and sparse structure of graph filters to limit the information sharing while emulating the performance of centralized training. Even though we preserve data locality and agent communication is restricted to the neighborhood level, the proposed method still converges in simulation.
ISSN:2151-870X
DOI:10.1109/SAM60225.2024.10636596