Robust Federated Learning for Wireless Networks: A Demonstration with Channel Estimation
Abstract: Federated learning (FL) offers a privacy-preserving collaborative approach for training models in wireless networks, with channel estimation emerging as a promising application. Despite extensive studies on FL-empowered channel estimation, the security concerns associated with FL require meticulous attention. In a scenario where small base stations (SBSs) serve as local models trained on cached data, and a macro base station (MBS) functions as the global model setting, an attacker can exploit the vulnerability of FL, launching attacks with various adversarial attacks or deployment tactics. In this paper, we analyze such vulnerabilities, corresponding solutions were brought forth, and validated through simulation.
- N. H. Tran, W. Bao, A. Zomaya et al., “Federated learning over wireless networks: Optimization model design and analysis,” in IEEE INFOCOM 2019 - IEEE Conference on Computer Communications, 2019, pp. 1387–1395.
- S. Niknam, H. S. Dhillon, and J. H. Reed, “Federated learning for wireless communications: Motivation, opportunities, and challenges,” IEEE Communications Magazine, vol. 58, no. 6, pp. 46–51, 2020.
- S. M. Sheikholeslami, A. Rasti-Meymandi, S. J. Seyed-Mohammadi et al., “Communication-efficient federated learning for hybrid vlc/rf indoor systems,” IEEE Access, vol. 10, pp. 126 479–126 493, 2022.
- M. M. Wadu, S. Samarakoon, and M. Bennis, “Federated learning under channel uncertainty: Joint client scheduling and resource allocation,” in 2020 IEEE Wireless Communications and Networking Conference (WCNC), 2020, pp. 1–6.
- A. M. Elbir and S. Coleri, “Federated learning for hybrid beamforming in mm-wave massive MIMO,” IEEE Communications Letters, vol. 24, no. 12, pp. 2795–2799, 2020.
- S. Elbir, Ahmet M.and Coleri, “Federated learning for channel estimation in conventional and ris-assisted massive MIMO,” IEEE Transactions on Wireless Communications, vol. 21, no. 6, pp. 4255–4268, 2022.
- S. Shahabodini, M. Mansoori, J. Abouei et al., “Recurrent neural network and federated learning based channel estimation approach in mmwave massive MIMO systems,” Transactions on Emerging Telecommunications Technologies, vol. 35, 12 2023.
- Y.-S. Jeon, M. Mohammadi Amiri, and N. Lee, “Communication-efficient federated learning over MIMO multiple access channels,” IEEE Transactions on Communications, vol. 70, no. 10, pp. 6547–6562, 2022.
- M. M. Amiri and D. Gündüz, “Federated learning over wireless fading channels,” IEEE Transactions on Wireless Communications, vol. 19, no. 5, pp. 3546–3557, 2020.
- M. S. H. Abad, E. Ozfatura, D. GUndUz et al., “Hierarchical federated learning across heterogeneous cellular networks,” in ICASSP 2020 - 2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2020, pp. 8866–8870.
- W. Hou, J. Sun, G. Gui et al., “Federated learning for dl-csi prediction in fdd massive MIMO systems,” IEEE Wireless Communications Letters, vol. 10, no. 8, pp. 1810–1814, 2021.
- J.-P. Hong, S. Park, and W. Choi, “Base station dataset-assisted broadband over-the-air aggregation for communication-efficient federated learning,” IEEE Transactions on Wireless Communications, vol. 22, no. 11, pp. 7259–7272, 2023.
- Y. Cao, S. Maghsudi, T. Ohtsuki et al., “Mobility-aware routing and caching in small cell networks using federated learning,” IEEE Transactions on Communications, pp. 1–1, 2023.
- E. M. Campos, A. G. Vidal, J. L. H. Ramos et al., “FedRDF: A robust and dynamic aggregation function against poisoning attacks in federated learning,” 2024, preprint available at arXiv:2402.10082.
- V. Rey, P. M. Sánchez Sánchez, A. Huertas Celdrán et al., “Federated learning for malware detection in IoT devices,” Computer Networks, vol. 204, p. 108693, 2022.
- L. Cao, H. Chen, X. Fan et al., “Bayesian federated learning: A survey,” in IJCAI-23, 8 2023, pp. 7233–7242, survey Track.
- H.-Y. Chen and W.-L. Chao, “FedBE: Making bayesian model ensemble applicable to federated learning,” in International Conference on Learning Representations, 2021.
- B. Han and H. D. Schotten, “A secure and robust approach for distance-based mutual positioning of unmanned aerial vehicles,” 2024, to appear at IEEE WCNC 2024, preprint available at arXiv:2305.12021.
- F. O. Catak, M. Kuzlu, E. Catak et al., “Defensive distillation-based adversarial attack mitigation method for channel estimation using deep learning models in next-generation wireless networks,” IEEE Access, vol. 10, pp. 98 191–98 203, 2022.
- Z. Fang, B. Han, and H. D. Schotten, “A reliable and resilient framework for multi-uav mutual localization,” in 2023 IEEE 98th Vehicular Technology Conference (VTC2023-Fall), 2023, pp. 1–7.
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.