A Simple Python Testbed for Federated Learning Algorithms

TL;DR

This paper introduces a Python-based testbed designed to facilitate the development and testing of federated learning algorithms, especially for IoT edge systems, supporting both centralized and decentralized approaches.

Contribution

It provides a simple, pure Python framework for federated learning, validated with example algorithms, addressing the gap in tools for IoT edge system development.

Findings

Supports both centralized and decentralized algorithms

Validated with three example algorithms

Facilitates development for IoT edge systems

Abstract

Nowadays many researchers are developing various distributed and decentralized frameworks for federated learning algorithms. However, development of such a framework targeting smart Internet of Things in edge systems is still an open challenge. In this paper, we present our solution to that challenge called Python Testbed for Federated Learning Algorithms. The solution is written in pure Python, and it supports both centralized and decentralized algorithms. The usage of the presented solution is both validated and illustrated by three simple algorithm examples.