This repository contains the experimental code for the "Privacy-Preserving Collaborative Split Learning Framework for Smart Grid Load Forecasting" paper currently under review at TIFS.
- Install Python 3.6, PyTorch 1.9.0 to run the code.
- FEDformer is used as the central model
- SplitGSSP is the SplitGlobal model discussed in the paper
- SplitPerson is the SplitPersonal model discussed in the paper
- To run the individual tests, run
- run_Central.py to train a Central model
- run_SplitFramework to train the SplitGlobal model
- run_SplitFrameworkPersonal to train the SplitPersonal model
We appreciate the following github repos a lot for their valuable code base or datasets:
https://github.com/MAZiqing/FEDformer
https://github.com/thuml/Autoformer