Generalized SplitFed learning, which takes advantage of both federated learning and split learning

This is for releasing the source code of our work "Design and Evaluation of Federated Learning, Split Learning and SplitFed for Internet of Things".

If you find it is useful and used for publication. Please kindly cite our work as:

@inproceedings{gao2021desgin,
title={Evaluation and Optimization of Distributed Machine Learning Techniques for Internet of Things},
author={Gao, Yansong and Kim, Minki and Thapa, Chandra and Abuadbba, Sharif and Zhang, Zhi and Camtepe, Seyit A and Kim, Hyoungshick and Nepal, Surya},
year={2021}}

Helpful Link

• github project for SRDS 2020 paper "End-to-End Evaluation of Federated Learning and Split Learning for Internet of Things"

Description

This repository contains the implementations of splitfed learning and performance evaluations under IID, imbalanced and non-IID data distribution settings. It also has the code used for Raspberry Pi implementation. For the split learning and federated learning implementations, refer to above link "github project for SRDS 2020".