FBLG: A Local Graph Based Approach for Handling Dual Skewed Non-IID Data in Federated Learning

Abstract

Personalized federated graph neural networks (PFGNN) are an emerging technology that allows multiple graph data owners to collaboratively train personalized models without sharing raw data. However, the Non-IID nature of graph data can cause the coupling of global and local knowledge parameters, which disrupts the optimization in personalized federated learning. Additionally, node neighbors may carry global and local knowledge, and their direct inclusion in training may introduce noise, degrading federated model performance. In this work, we propose the Adaptive Decoupling Personalized Federated Graph Neural Network (ADPFedGNN), which leverages multi-party collaboration to train personalized models for classifying local client graph nodes. We use two automatically updated masks and mutual information minimization to decouple global and local parameters in FGNN. We employ reinforcement learning to adaptively select appropriate neighbors for training global or local knowledge-related parameters while filtering out irrelevant nodes. We also design a personalized federated masked parameter aggregation mechanism that efficiently updates local personalized model parameters and aggregates the masked parameters. Experimental results on three public datasets demonstrate that ADPFedGNN outperforms existing methods, achieving average improvements of 5.66 percent, 5.83 percent, and 12.45 percent in ACC, F1, and Recall, respectively.

Cite

Text

Xu et al. "FBLG: A Local Graph Based Approach for Handling Dual Skewed Non-IID Data in Federated Learning." International Joint Conference on Artificial Intelligence, 2024. doi:10.24963/ijcai.2024/585

Markdown

[Xu et al. "FBLG: A Local Graph Based Approach for Handling Dual Skewed Non-IID Data in Federated Learning." International Joint Conference on Artificial Intelligence, 2024.](https://mlanthology.org/ijcai/2024/xu2024ijcai-fblg/) doi:10.24963/ijcai.2024/585

BibTeX

@inproceedings{xu2024ijcai-fblg,
  title     = {{FBLG: A Local Graph Based Approach for Handling Dual Skewed Non-IID Data in Federated Learning}},
  author    = {Xu, Yi and Li, Ying and Luo, Haoyu and Fan, Xiaoliang and Liu, Xiao},
  booktitle = {International Joint Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {5289-5297},
  doi       = {10.24963/ijcai.2024/585},
  url       = {https://mlanthology.org/ijcai/2024/xu2024ijcai-fblg/}
}