A Knowledge Distillation-Empowered Adaptive Federated Reinforcement Learning Framework for Multi-Domain IoT Applications Scheduling

View at publisher
Authors: Zhiyu Wang, Mohammad Goudarzi, Mingming Gong, Rajkumar Buyya
Cited by: 2
References: 34

Abstract

The rapid proliferation of Internet of Things (IoT) applications across heterogeneous Cloud-Edge-IoT environments presents significant challenges in distributed scheduling optimization. Existing approaches face issues, including fixed neural network architectures that are incompatible with computational heterogeneity, non-Independent and Identically Distributed (non IID) data distributions across IoT scheduling domains, and insufficient cross-domain collaboration mechanisms. This pa per proposes KD-AFRL, a Knowledge Distillation-empowered Adaptive Federated Reinforcement Learning framework that addresses multi-domain IoT application scheduling through three core innovations. First, we develop a resource-aware hybrid architecture generation mechanism that creates dual-zone neural networks enabling heterogeneous devices to participate in collaborative learning while maintaining optimal resource utilization. Second, we propose a privacy-preserving environment-clustered federated learning approach that utilizes differential privacy and K-means clustering to address non-IID challenges and facilitate effective collaboration among compatible domains. Third, we introduce an environment-oriented cross-architecture knowledge distillation mechanism that enables efficient knowledge transfer between heterogeneous models through temperature-regulated soft targets. Comprehensive experiments with real Cloud-Edge IoT infrastructure demonstrate KD-AFRL's effectiveness using diverse IoT applications. Results show significant improvements over the best baseline, with 21% faster convergence and 15.7%, 10.8%, and 13.9% performance gains in completion time, energy consumption, and weighted cost, respectively. Scalability experiments reveal that KD-AFRL achieves 3-5 times better performance retention compared to existing solutions as the number of domains increases.

Fields of Study & Subjects

IoT and Edge/Fog Computing Age of Information Optimization Privacy-Preserving Technologies in Data

Publication details

Journal / source
IEEE Transactions on Mobile Computing
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Volume
25
Issue
8
Pages
12890–12907
Publication date
12 March 2026