Details

Autor(en) / Beteiligte

• Kim, Doyeon
• Shin, Seungjae
• Jeong, Jaewon
• Lee, Joohyung

Titel

Joint Edge Server Selection and Data Set Management for Federated-Learning-Enabled Mobile Traffic Prediction

Ist Teil von

• IEEE internet of things journal, 2024-02, Vol.11 (3), p.4971-4986

Ort / Verlag

Piscataway: The Institute of Electrical and Electronics Engineers, Inc. (IEEE)

Erscheinungsjahr

2024

Link zum Volltext

Quelle

IEEE/IET Electronic Library (IEL)

Beschreibungen/Notizen

• To realize intelligent network management for future 6G-mobile edge computing (MEC) systems, mobile traffic prediction is crucial. Most of the previous machine learning-driven prediction approaches adopt traditional centralized training paradigm wherein mobile traffic data should be transferred to a central server. To exploit the distributed and parallel processing nature of MEC servers for training mobile traffic prediction models in a fast and secure manner, we propose a novel federated learning (FL) framework wherein locally trained prediction models over MEC servers are aggregated into a global model with joint optimization of MEC server selection and data set management for FL participation. From mathematical investigations of the influence of MEC server participation and data set utilization on the global model accuracy and training costs, including both training latency and energy consumption in the FL process, we first formulate an optimization problem for balancing the accuracy-cost tradeoff by considering a linear accuracy estimation model. Here, the optimization problem is designed using mixed-integer nonlinear programming, which is generally known as NP-hard. We then leverage a number of relaxation techniques to develop near-optimal yet the plausible algorithm based on linear programming. Furthermore, for practical concern, the proposed problem is extended by considering a concave accuracy estimation model; a genetic-based heuristic approach to the extension is proposed for determining the suboptimal solution. The numerical and simulation results suggest that our proposed framework can be effective for building mobile traffic prediction models in a more cost-efficient manner while maintaining competitive prediction accuracy.