FedLoG

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

★ 17 Open GitHub ↗

The official source code for "Subgraph Federated Learning for Local Generalization (FedLoG)" at ICLR 2025 (Oral).

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Subgraph Federated Learning for Local Generalizatioin (FedLoG)

The official source code for [**Subgraph Federated Learning for Local Generalization**](https://www.arxiv.org/abs/2503.03995) at ICLR 2025 (Oral).

## Abstract 
Federated Learning (FL) on graphs enables collaborative model training to enhance performance without compromising the privacy of each client. However, previous methods often overlook the mutable nature of graph data, which frequently introduces new nodes and leads to shifts in label distribution. Unlike prior methods that struggle to generalize to unseen nodes with diverse label distributions, our proposed method FedLoG effectively addresses this issue by alleviating the problem of local overfitting. Our model generates global synthetic data by condensing the reliable information from each class representation and its structural information across clients. Using these synthetic data as a training set, we alleviate the local overfitting problem by adaptively generalizing the absent knowledge within each local dataset. This enhances the generalization capabilities of local models, enabling them to handle unseen data effectively. Our model outperforms the baselines in proposed experimental settings, which are designed to measure generalization power to unseen data in practical scenarios.

## Requirements
- python=3.10
- pytorch=2.0.1
- torch-geometric=2.4.0

## Datasets
You can download the datasets using the following link:  
[Download Dataset](https://drive.google.com/file/d/12u40AJMXeeplxfSeOuhU29rv8YtBzWMl/view?usp=share_link)

After extracting `data.zip`, put the `data` folder in the same directory as the `main.py` file.

## How to run
```
python main.py --dataset cora --n_silos 3 --unseen_setting closeset
```
### Flags
`dataset` : cora, citeseer, pubmed, photo, computers  
`n_silos` : 3, 5, 10  
`unseen_setting` : closeset, openset

### Unseen Settings
<img width="500" alt="image" src="https://github.com/user-attachments/assets/e8ddc755-0ecd-4395-9294-cabb2fcec582" />

### Citation

```BibTex
@article{kim2025subgraph,
  title={Subgraph Federated Learning for Local Generalization},
  author={Kim, Sungwon and Lee, Yoonho and Oh, Yunhak and Lee, Namkyeong and Yun, Sukwon and Lee, Junseok and Kim, Sein and Yang, Carl and Park, Chanyoung},
  journal={arXiv preprint arXiv:2503.03995},
  year={2025}
}
```

FedLoG

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README