Learn What You Need in Personalized Federated Learning

Codes for the paper Learn What You Need in Personalized Federated Learning submitted to NeurIPS 2023.

Table of Content

• 1. File descriptions
• 2. Train Learn2pFed

1. File descriptions

A brief description for the files in this repo:

• Layers.py constructions of Learn2pFed network for regression in synthetic data and forecasting in ELD dataset
• demo.sh & demo.py scripts on training Learn2pFed on synthetic data and real-world data

2. Run

Regression in synthetic data

To see the performance of Learn2pFed on the synthetic data, run

sh demo.sh

This repo is keeping on updating.