Federated Learning Empowered End-Edge-Cloud Cooperation for 5G HetNet Security
Yunkai Wei, Sipei Zhou, Supeng Leng, Sabita Maharjan, Yan Zhang
Abstract
The distributed and heterogeneous framework in the 5G heterogeneous network (Het-Net) makes it vulnerable to attacks of different kinds. Nodes for improving the network security are therefore important to eliminate such critical threats. Without cooperation or with limited cooperation, these nodes are substantially restricted in their protecting capacity due to specific characteristics such as heterogeneity, hierarchy, and wide range in the 5G HetNet. In this article, we propose a federated learning empowered end-edge-cloud cooperation-based framework for enhancing 5G HetNet security. In this framework, nodes equipped with attack detection mechanisms are distributed in the end, edge, and cloud of the 5G HetNet. We then design cooperative training schemes to realize the full potential of these nodes in detecting attacks. Illustrative results demonstrate the superior performance of our proposed scheme compared to three different benchmark schemes.