arrow
Return

Autonomous Model Aggregation for Decentralized Learning on Edge Devices

delete2025-10-14
delete0
PRE
AI
J
Jinru Chen
J
Jingke Tu
杨磊 (Lei Yang)
J
Jiannong Cao
DOI:10.1109/TPDS.2025.3621058delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Edge AI applications enable edge devices to collaboratively learn a model via repeated model aggregations, aiming to utilize the distributed data on the devices for achieving high model accuracy. Existing methods either leverage a centralized server to directly aggregate the model updates from edge devices or need a central coordinator to group the edge devices for localized model aggregations. The centralized server (or coordinator) has a performance bottleneck and a high cost of collecting the global state needed for making the grouping decision in large-scale networks. In this paper, we propose an Autonomous Model Aggregation (AMA) method for large-scale decentralized learning on edge devices. Instead of needing a central coordinator to group the edge devices, AMA allows the edge devices to autonomously form groups using a highly efficient protocol, according to model functional similarity and historical grouping information. Moreover, AMA adopts a reinforcement learning approach to optimize the size of each group. Evaluation results on our self-developed edge computing testbed demonstrate that AMA outperforms the benchmark approaches by up to 20.71% in accuracy and reduced the convergence time by 75.58%.
Keywords:
Edge intelligence
distributed edge learning
model aggregation

Journal

IEEE Transactions on Parallel and Distributed Systems cover
IEEE Transactions on Parallel and Distributed Systems
IF:
6
Papers:
5.2K
Citations:
1.1W

Organization

T
The Hong Kong Polytechnic University
Scholars:
5.1K
Papers: 3.0K
Citations: 17
S
south china university of technology
Scholars:
6.8W
Papers: 5.1W
Citations: 85