Title: Efficient edge AI implementation for IoT device identification for hierarchical federated learning

Authors: Sumitra Budania; Jeevan Kittur; Meetha V. Shenoy

Addresses: Department of Electrical and Electronics Engineering, Birla Institute of Technology and Science, Pilani, 333031, India ' Department of Electrical and Electronics Engineering, Birla Institute of Technology and Science, Pilani, 333031, India ' Department of Electrical and Electronics Engineering, Birla Institute of Technology and Science, Pilani, 333031, India

Abstract: As IoT devices proliferate, efficient IoT device identification is crucial for resource management, planning, and detecting anomalous traffic. Traditional ML-based identification relies on centralised training, but federated learning (FL) offers a privacy-preserving alternative, enabling collaborative model training without sharing raw data. FL enhances edge devices' ability to identify previously unconnected devices. However, resource constraints like limited computation, power, and communication capabilities may prevent some edge devices from actively participating in FL. We propose a solution where resource-limited IoT devices benefit from FL by subscribing to server-based services. This work presents an efficient AI model implementation for IoT device identification on embedded edge devices, detailing the toolflow from model generation to hardware implementation. We apply and evaluate various model optimisation techniques to balance performance and resource trade-offs, offering insights to advance edge-AI and scalable FL-based ML applications for IoT networks.

Keywords: edge AI; model compression; inference optimisation; deep learning; internet of things; IoT; device identification.

DOI: 10.1504/IJES.2025.144935

International Journal of Embedded Systems, 2025 Vol.18 No.1, pp.61 - 73

Received: 20 Mar 2024
Accepted: 04 Jan 2025
Published online: 11 Mar 2025 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article