用两阶段联邦学习检测5G干扰,保护隐私还高效。
A Two-Stage CAE-Based Federated Learning Framework for Efficient Jamming Detection in 5G Networks
- 先用自编码器无监督训练,再用分类网络有监督优化。
- 在非独立同分布数据下仍达92%准确率,30轮通信完成训练。
- 适合5G小基站安全防护,兼顾隐私与实时性需求。
5G网络面临复杂干扰攻击的严峻挑战,尤其对异构网络(HetNet)的射频(RF)域构成威胁,导致性能下降。传统机器学习方法依赖集中式训练,存在数据隐私风险。本文提出一种基于两阶段联邦学习(FL)的干扰检测框架,用于5G小型蜂窝网络。第一阶段采用联邦平均(FedAVG)算法训练卷积自编码器(CAE)进行无监督学习;第二阶段在预训练的CAE编码器基础上构建全连接网络(FCN),并使用联邦近端(FedProx)算法进行有监督分类训练。实验表明,该框架在非独立同分布(non-IID)客户端数据上实现高效训练与预测,同时保障数据隐私。具体表现:精度0.94、召回率0.90、F1分数0.92、准确率0.92,通信轮次仅需30轮,最优客户端数量为6,具备良好收敛性与抗干扰能力。
原文摘要 · Abstract (English)
Cyber-security for 5G networks is drawing notable attention due to an increase in complex jamming attacks that could target the critical 5G Radio Frequency (RF) domain. These attacks pose a significant risk to heterogeneous network (HetNet) architectures, leading to degradation in network performance. Conventional machine-learning techniques for jamming detection rely on centralized training while increasing the odds of data privacy. To address these challenges, this paper proposes a decentralized two-stage federated learning (FL) framework for jamming detection in 5G femtocells. Our proposed distributed framework encompasses using the Federated Averaging (FedAVG) algorithm to train a Convolutional Autoencoder (CAE) for unsupervised learning. In the second stage, we use a fully connected network (FCN) built on the pre-trained CAE encoder that is trained using Federated Proximal (FedProx) algorithm to perform supervised classification. Our experimental results depict that our proposed framework (FedAVG and FedProx) accomplishes efficient training and prediction across non-IID client datasets without compromising data privacy. Specifically, our framework achieves a precision of 0.94, recall of 0.90, F1-score of 0.92, and an accuracy of 0.92, while minimizing communication rounds to 30 and achieving robust convergence in detecting jammed signals with an optimal client count of 6.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。