README.md

FedAA: A Reinforcement Learning Perspective on Adaptive Aggregation for Fair and Robust Federated Learning (AAAI 2025)

This repository contains the code for the paper titled FedAA: A Reinforcement Learning Perspective on Adaptive Aggregation for Fair and Robust Federated Learning, authored by Jialuo He, Wei Chen, Xiaojin Zhang.

Link to Arxiv: TODO

Installation

1. Clone the repository:

   git clone https://github.com/Gp1g/FedAA.git
   cd FedAA

2. Create a new environment:

   conda create -n FedAA python==3.8
   conda activate FedAA

3. Install the necessary dependencies:

   pip install -r requirements.txt

Usage

We provide the initial data for MNIST in ./dataset, you can run the main.py as follows:

python main.py

Citation (TODO)