Diffusion training in federated learning setting.
Use diffusion_cen.ipynb for centralized training setting.
Use diffusion_fedavg.ipynb for federated learning setting.
Use 20230713_diffusion_env.yml for anaconda environment.
The diffusion implementation is referred from the following colab notebook.
https://colab.research.google.com/drive/1IJkrrV-D7boSCLVKhi7t5docRYqORtm3
Please consider citing our paper if you find this code useful.
@INPROCEEDINGS{10318858,
author={Tun, Ye Lin and Thwal, Chu Myaet and Yoon, Ji Su and Kang, Sun Moo and Zhang, Chaoning and Hong, Choong Seon},
booktitle={2023 International Conference on Advanced Technologies for Communications (ATC)},
title={Federated Learning with Diffusion Models for Privacy-Sensitive Vision Tasks},
year={2023},
volume={},
number={},
pages={305-310},
doi={10.1109/ATC58710.2023.10318858}}