TinyFed: Lightweight Federated Learning for Constrained IoT Devices
Year: 2024
Type: article
Abstract: Tiny Machine Learning (TinyML) is a paradigm that directly enables the inference of machine learning (ML) algorithms on microcontrollers — devices with limited computational resources, including energy, processing power, and storage constraints. Extending TinyML to allow inference and model training in a collaborative setting has led to the development of Tiny Federated Learning (TinyFL). This proposal introduces TinyFed, a lightweight framework designed to support the entire Federated Learning cycle, including training a neural network on the device, transmitting the model to a server, aggregating multiple models, and delivering the aggregated model back to resource-constrained microcontrollers engaged in the federated process. TinyFed has been validated by deploying a neural network with four inputs, three hidden layers (16, 8, and 4 neurons), and two outputs on ESP32 devices to detect temperature, humidity, luminosity, and voltage measurement anomalies. Our results demonstrate the feasibility of this approach with successful local training.
Authors: Silva, Claudio, Prazeres, Cássio
Language: English
Subfield: Artificial Intelligence
Field: Computer Science
Domain: Physical Sciences
Open Access status: green
OpenAlex ID: W6911854359
FWCI:
Cites:
Cited by:
Related to: