Abstract:Real-time machine learning (ML) has recently attracted significant interest due to its potential to support instantaneous learning, adaptation, and decision making in a wide range of application domains, including self-driving vehicles, intelligent transportation, and industry automation. In this paper, we investigate real-time ML in a federated edge intelligence (FEI) system, an edge computing system that implements federated learning (FL) solutions based on data samples collected and uploaded from decentralized data networks, e.g., Internet-of-Things (IoT) and/or wireless sensor networks. FEI systems often exhibit heterogenous communication and computational resource distribution, as well as non-i.i.d. data samples arrived at different edge servers, resulting in long model training time and inefficient resource utilization. Motivated by this fact, we propose a time-sensitive federated learning (TS-FL) framework to minimize the overall run-time for collaboratively training a shared ML model with desirable accuracy. Training acceleration solutions for both TS-FL with synchronous coordination (TS-FL-SC) and asynchronous coordination (TS-FL-ASC) are investigated. To address the straggler effect in TS-FL-SC, we develop an analytical solution to characterize the impact of selecting different subsets of edge servers on the overall model training time. A server dropping-based solution is proposed to allow some slow-performance edge servers to be removed from participating in the model training if their impact on the resulting model accuracy is limited. A joint optimization algorithm is proposed to minimize the overall time consumption of model training by selecting participating edge servers, the local epoch number (the number of model training iterations per coordination), and the data batch size (the number of data samples for each model training iteration). Motivated by the fact that data samples at the slowest edge server may exhibit special characteristics that cannot be removed from model training, we develop an analytical expression to characterize the impact of both staleness effect of asynchronous coordination and straggler effect of FL on the time consumption of TS-FL-ASC. We propose a load forwarding-based solution that allows a slow edge server to offload part of its training samples to trusted edge servers with higher processing capability. We develop a hardware prototype to evaluate the model training time of a heterogeneous FEI system. Experimental results show that our proposed TS-FL-SC and TS-FL-ASC can provide up to 63% and 28% of reduction, in the overall model training time, respectively, compared with traditional FL solutions.

Offloading Federated Learning Task to Edge Computing with Trust Execution Environment

Edge-cloud Collaborative Learning with Federated and Centralized Features

Enabling Execution Assurance of Federated Learning at Untrusted Participants.

Task Offloading in Trusted Execution Environment Empowered Edge Computing.

An Efficient Asynchronous Federated Learning Protocol for Edge Devices

Distributed Learning in Trusted Execution Environment: A Case Study of Federated Learning in SGX

Accelerating Federated Learning with Data and Model Parallelism in Edge Computing

Online Client Scheduling and Resource Allocation for Efficient Federated Edge Learning

Experimental Evaluation and Analysis of Federated Learning in Edge Computing Environments

Computation Offloading for Edge-Assisted Federated Learning

TEA-fed

Accelerating Federated Learning Over Reliability-Agnostic Clients in Mobile Edge Computing Systems

Federated Learning with Workload Reduction through Partial Training of Client Models and Entropy-Based Data Selection

Time-sensitive Learning for Heterogeneous Federated Edge Intelligence

FLEdge: Benchmarking Federated Machine Learning Applications in Edge Computing Systems

Federated Learning With Client Selection and Gradient Compression in Heterogeneous Edge Systems

FedEdge: Accelerating Edge-Assisted Federated Learning

Efficient Coordination of Federated Learning and Inference Offloading at the Edge: A Proactive Optimization Paradigm

Edge-Cloud Collaborative Computation Offloading for Federated Learning in Smart City.

EdgeFed: Optimized Federated Learning Based on Edge Computing

Accelerating Decentralized Federated Learning in Heterogeneous Edge Computing