E-Tree Learning: A Novel Decentralized Model Learning Framework for Edge AI

by   Lei Yang, et al.

Traditionally, AI models are trained on the central cloud with data collected from end devices. This leads to high communication cost, long response time and privacy concerns. Recently Edge empowered AI, namely Edge AI, has been proposed to support AI model learning and deployment at the network edge closer to the data sources. Existing research including federated learning adopts a centralized architecture for model learning where a central server aggregates the model updates from the clients/workers. The centralized architecture has drawbacks such as performance bottleneck, poor scalability and single point of failure. In this paper, we propose a novel decentralized model learning approach, namely E-Tree, which makes use of a well-designed tree structure imposed on the edge devices. The tree structure and the locations and orders of aggregation on the tree are optimally designed to improve the training convergency and model accuracy. In particular, we design an efficient device clustering algorithm, named by KMA, for E-Tree by taking into account the data distribution on the devices as well as the the network distance. Evaluation results show E-Tree significantly outperforms the benchmark approaches such as federated learning and Gossip learning under NonIID data in terms of model accuracy and convergency.


page 1

page 2

page 3

page 4


Blockchain-based Federated Learning for Device Failure Detection in Industrial IoT

Device failure detection is one of most essential problems in industrial...

Graph Federated Learning Based on the Decentralized Framework

Graph learning has a wide range of applications in many scenarios, which...

DisPFL: Towards Communication-Efficient Personalized Federated Learning via Decentralized Sparse Training

Personalized federated learning is proposed to handle the data heterogen...

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

Mobile Edge Computing (MEC), which incorporates the Cloud, edge nodes an...

Collaborative Policy Learning for Dynamic Scheduling Tasks in Cloud-Edge-Terminal IoT Networks Using Federated Reinforcement Learning

In this paper, we examine cloud-edge-terminal IoT networks, where edges ...

EdgeFL: A Lightweight Decentralized Federated Learning Framework

Federated Learning (FL) has emerged as a promising approach for collabor...

Olive Branch Learning: A Topology-Aware Federated Learning Framework for Space-Air-Ground Integrated Network

The space-air-ground integrated network (SAGIN), one of the key technolo...

Please sign up or login with your details

Forgot password? Click here to reset