arrow
Return

AutoSF: Adaptive Distributed Model Training in Dynamic Edge Computing

delete2024-06-01
delete2
PRE
AI
杨磊 (Lei Yang) *
J
Jinru Chen
J
Jiannong Cao
DOI:10.1109/TMC.2023.3323456delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Distributed learning on edges aims at training the AI model collaboratively in a network of edge devices via frequent model aggregations. Achieving the desired training performance requires the aggregation structure and frequency to fit well with the dynamic edge environment. Existing works often consider the optimization of either aggregation structure or frequency, assuming that the edge environment is stable and deterministic. In this paper, we propose a novel approach, AutoSF, to automatically optimize the aggregation structure and frequency jointly in dynamic edge computing so as to minimize the global loss function. The main idea of AutoSF is that when the edge environment changes, the automated machine learning approach is triggered to find out the near-optimal aggregation structure and frequency that adapt to time-varying edge resources. When the environment keeps unchanged, a heuristic approach is used to tune the aggregation structure and frequency to further tame the heterogeneity caused by data distributions. We validate the effectiveness of AutoSF via numerical experiments with real datasets on our self-developed edge computing testbed. Evaluation results demonstrate that AutoSF outperforms the benchmark approaches by up to 16.3x speedups in convergence speed and 31.0% increases in training accuracy.
Keywords:
Training
Computational modeling
Adaptation models
Performance evaluation
Edge computing
Convergence
Bandwidth
Aggregation frequency
aggregation structure
distributed machine learning
edge intelligence

Journal

IEEE Transactions on Mobile Computing cover
IEEE Transactions on Mobile Computing
IF:
9.2
Papers:
5.6K
Citations:
1.8W

Organization

H
hong kong polytechnic university
Scholars:
3.0W
Papers: 4.1W
Citations: 921
S
south china university of technology
Scholars:
6.7W
Papers: 5.1W
Citations: 85