Add Flower Baseline: FedBABU
Paper
FedBABU: Towards Enhanced Representation for Federated Image Classification
Link
https://arxiv.org/abs/2106.06042
Maybe give motivations about why the paper should be implemented as a baseline.
FedBABU (200+ citations) is quite famous in pFL field and is considered as one of the most common baselines that introduced by many pFL papers.
The plan is reproducing FedBABU and FedAvg experiment results in Table 5 in the paper.
Is there something else you want to add?
I've reviewed the PR and issue list and found no ongoing work on FedBABU. I've already begun the reproduction process.
Implementation
To implement this baseline, it is recommended to do the following items in that order:
For first time contributors
- [ ] Read the
first contributiondoc - [ ] Complete the Flower tutorial
- [ ] Read the Flower Baselines docs to get an overview:
Prepare - understand the scope
- [X] Read the paper linked above
- [X] Decide which experiments you'd like to reproduce. The more the better!
- [X] Follow the steps outlined in Add a new Flower Baseline.
- [X] You can use as reference other baselines that the community merged following those steps.
Verify your implementation
- [X] Follow the steps indicated in the
EXTENDED_README.mdthat was created in your baseline directory - [X] Ensure your code reproduces the results for the experiments you chose
- [X] Ensure your
README.mdis ready to be run by someone that is no familiar with your code. Are all step-by-step instructions clear? - [X] Ensure running the formatting and typing tests for your baseline runs without errors.
- [X] Clone your repo on a new directory, follow the guide on your own
README.mdand verify everything runs.
Hey @KarhouTam, great to see you back with another baseline! Having the FedBABU and FedAvg columns in Table 5 looks good!.
Just note that we have updated the process of contributing a baseline. Now is different from your recently merged FedRep and the template is based on the flwr new/run project structure. Ping us if you have some doubts.
Migrating this to Flower Discuss. There is an open PR but people can pick up from there if there is interest.