The implementation of CDW_FedAvg in the paper "Blockchain-Based Federated Learning for Device Failure Detection in Industrial IoT"
git clone https://github.com/upcbdipt/CDW_FedAvg.git && cd CDW_FedAvg
Install dependencies using python 3.6+ (recommend using a virtualenv):
pip install -r requirements.txt
python 0_prepare_data.py
python 1_reduce_dimension.py
python 2_run.py
python 3_plot.py
If you use this work, please cite:
@ARTICLE{9233457,
author={Zhang, Weishan and Lu, Qinghua and Yu, Qiuyu and Li, Zhaotong and Liu, Yue and Lo, Sin Kit and Chen, Shiping and Xu, Xiwei and Zhu, Liming},
journal={IEEE Internet of Things Journal},
title={Blockchain-Based Federated Learning for Device Failure Detection in Industrial IoT},
year={2021},
volume={8},
number={7},
pages={5926-5937},
doi={10.1109/JIOT.2020.3032544}}
CDW_FedAvg is distributed under Apache 2/lic.0 license.
Contact: Weishan Zhang ([email protected])