AutoSF: Adaptive Distributed Model Training in Dynamic Edge Computing

Lei Yang, Yingqi Gan, Jinru Chen, Jiannong Cao

Abstract: 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.3× speedups in convergence speed and 31.0$\%$ increases in training accuracy.

External IDs:dblp:journals/tmc/YangGCC24