학술논문

Joint Edge Server Selection and Data Set Management for Federated-Learning-Enabled Mobile Traffic Prediction
Document Type
Periodical
Source
IEEE Internet of Things Journal IEEE Internet Things J. Internet of Things Journal, IEEE. 11(3):4971-4986 Feb, 2024
Subject
Computing and Processing
Communication, Networking and Broadcast Technologies
Servers
Training
Predictive models
Data models
3GPP
Estimation
6G mobile communication
Federated learning (FL)
genetic algorithm
mobile edge computing (MEC)
mobile traffic prediction
Language
ISSN
2327-4662
2372-2541
Abstract
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.