arXiv:2410.12772cs.DCcs.AI2024-10

通过引入可控噪声提升联邦学习在无线调制识别中的抗噪能力

Vaccinating Federated Learning for Robust Modulation Classification in Distributed Wireless Networks

  • 用谐波噪声鲁棒性机制动态调节训练过程,增强模型抗噪性
  • 在非独立同分布数据下实现更优收敛,较现有方法准确率提升5.2%
  • 适合需要隐私保护的分布式无线网络场景,尤其对抗干扰敏感系统

自动调制分类(AMC)在分布式无线网络中对保障高效可靠通信至关重要。近年来,基于深度神经网络(DNN)的AMC模型发展迅速,联邦学习(FL)成为有前景的框架。然而,信号中的各类噪声给模型优化带来挑战,且现有基于FL的AMC模型多依赖线性聚合策略,在非独立同分布(non-IID)环境下难以有效整合局部微调参数,阻碍最优学习收敛。为此,我们提出FedVaccine,一种新型联邦学习模型,通过有意识引入平衡噪声水平,提升模型在不同噪声条件下的泛化能力。该方法基于谐波噪声鲁棒性设计,识别DNN模型的最佳噪声容忍度,调控训练过程并缓解过拟合。同时,为克服传统线性聚合的局限,采用结构聚类拓扑与本地队列数据结构的分层学习策略,支持自适应累积更新。实验结果表明,无论在IID与non-IID数据集上,或进行消融分析,FedVaccine均展现出更强鲁棒性与优越性能,优于现有基于FL的AMC方法。这表明其在实际无线网络环境中具有提升AMC系统可靠性与性能的潜力。

原文摘要 · Abstract (English)

Automatic modulation classification (AMC) serves a vital role in ensuring efficient and reliable communication services within distributed wireless networks. Recent developments have seen a surge in interest in deep neural network (DNN)-based AMC models, with Federated Learning (FL) emerging as a promising framework. Despite these advancements, the presence of various noises within the signal exerts significant challenges while optimizing models to capture salient features. Furthermore, existing FL-based AMC models commonly rely on linear aggregation strategies, which face notable difficulties in integrating locally fine-tuned parameters within practical non-IID (Independent and Identically Distributed) environments, thereby hindering optimal learning convergence. To address these challenges, we propose FedVaccine, a novel FL model aimed at improving generalizability across signals with varying noise levels by deliberately introducing a balanced level of noise. This is accomplished through our proposed harmonic noise resilience approach, which identifies an optimal noise tolerance for DNN models, thereby regulating the training process and mitigating overfitting. Additionally, FedVaccine overcomes the limitations of existing FL-based AMC models' linear aggregation by employing a split-learning strategy using structural clustering topology and local queue data structure, enabling adaptive and cumulative updates to local models. Our experimental results, including IID and non-IID datasets as well as ablation studies, confirm FedVaccine's robust performance and superiority over existing FL-based AMC approaches across different noise levels. These findings highlight FedVaccine's potential to enhance the reliability and performance of AMC systems in practical wireless network environments.

联邦学习调制识别抗噪设计无线网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。