AOP: Towards Adaptive Offloading Point Approach in a Federated Learning Framework for Edge AI Applications
Tran Anh Khoa, Do-Van Nguyen, Minh-Son Dao, Koji Zettsu · 2023
The study discusses challenges in deploying Artificial Intelligence (AI) on Internet-of-Things (IoT) devices and introduces a solution called Adaptive Offloading Point (AOP) for Federated Learning (FL) in Edge AI. AOP accelerates local training on resource-constrained devices and uses reinforcement learning-based clustering (RL) to determine which deep neural network (DNN) layers to offload to the server. Test results show that, in the VIT transformer model, AOP significantly reduces training time compared to classical FL and the baseline method called FedAdapt, making it a promising solution for Edge AI applications.