STFL: A Temporal-Spatial Federated Learning Framework for Graph Neural Networks
arXiv:2111.06750
Abstract
We present a spatial-temporal federated learning framework for graph neural networks, namely STFL. The framework explores the underlying correlation of the input spatial-temporal data and transform it to both node features and adjacency matrix. The federated learning setting in the framework ensures data privacy while achieving a good model generalization. Experiments results on the sleep stage dataset, ISRUC_S3, illustrate the effectiveness of STFL on graph prediction tasks.
References in corpus (8)
- Adam: A Method for Stochastic Optimization
- Semi-Supervised Classification with Graph Convolutional Networks
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- Inductive Representation Learning on Large Graphs
- FedGraphNN: A Federated Learning System and Benchmark for Graph Neural Networks
- Federated Graph Learning -- A Position Paper
- Cascade-BGNN: Toward Efficient Self-supervised Representation Learning on Large-scale Bipartite Graphs
- Cross-Node Federated Graph Neural Network for Spatio-Temporal Data Modeling