Skip to content

Commit

Permalink
20241205 update
Browse files Browse the repository at this point in the history
  • Loading branch information
Noploop authored Dec 5, 2024
1 parent 1e51de6 commit a70e1ab
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 2 deletions.
32 changes: 30 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,2 +1,30 @@
# SlaugFL
Code for SlaugFL
# SlaugFL: Efficient Edge Federated Learning With Selective GAN-Based Data Augmentation

This is the pytorch implementation code of the paper "[SlaugFL: Efficient Edge Federated Learning with Selective GAN-based Data Augmentation](https://ieeexplore.ieee.org/abstract/document/10521828)"

## Overview
The following figure shows an overview of our SlaugFL. The SlaugFL consists of two parts. The left part, in the dotted box, is the preparation phase which aims to obtain a trained ACGAN model. The right part shows the process of our FL collaborative training. In the collaborative training phase, we propose a dual-calibration method on the device side to improve FL performance: during the local training phase of selected devices, the local model is calibrated on the augmented data and further corrected through aligning its local class prototypes with the received p-f global class prototypes.

![SlaugFL](./SlaugFL_Overview.png)

We divide the implementation code into two parts: The Preparation Phase and The Collaborative Training Phase.

## The Preparation Phase


## The Collaborative Training Phase


## Citation

If you find this code is useful to your research, please consider to cite our paper.

```
@article{liu2024slaugfl,
title={SlaugFL: Efficient Edge Federated Learning with Selective GAN-based Data Augmentation},
author={Liu, Jianqi and Zhao, Zhiwei and Luo, Xiangyang and Li, Pan and Min, Geyong and Li, Huiyong},
journal={IEEE Transactions on Mobile Computing},
year={2024},
publisher={IEEE}
}
```
Binary file added SlaugFL_Overview.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit a70e1ab

Please sign in to comment.