Intrusion Detection for Wireless Edge Networks Based on Federated Learning

被引:79
作者
Chen, Zhuo [1 ]
Lv, Na [1 ]
Liu, Pengfei [1 ]
Fang, Yu [1 ]
Chen, Kun [1 ]
Pan, Wu [1 ]
机构
[1] Air Force Engn Univ, Informat & Nav Coll, Xian 710077, Peoples R China
基金
中国国家自然科学基金;
关键词
Intrusion detection; Data models; Collaborative work; Training; Image edge detection; Wireless communication; Servers; Wireless edge; intrusion detection; federated learning; gated recurrent unit;
D O I
10.1109/ACCESS.2020.3041793
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Edge computing provides off-load computing and application services close to end-users, greatly reducing cloud pressure and communication overhead. However, wireless edge networks still face the risk of network attacks. To ensure the security of wireless edge networks, we present Federated Learning-based Attention Gated Recurrent Unit (FedAGRU), an intrusion detection algorithm for wireless edge networks. FedAGRU differs from current centralized learning methods by updating universal learning models rather than directly sharing raw data among edge devices and a central server. We also apply the attention mechanism to increase the weight of important devices, by avoiding the upload of unimportant updates to the server, FedAGRU can greatly reduce communication overhead while ensuring learning convergence. Our experimental results show that, compared with other centralized learning algorithms, FedAGRU improves detection accuracy by approximately 8%. In addition, FedAGRU's communication cost is 70% less than other federated learning algorithms, and it exhibits strong robustness against poisoning attacks.
引用
收藏
页码:217463 / 217472
页数:10
相关论文
共 32 条
[1]   A Neural Network Architecture Combining Gated Recurrent Unit (GRU) and Support Vector Machine (SVM) for Intrusion Detection in Network Traffic Data [J].
Agarap, Abien Fred M. .
PROCEEDINGS OF 2018 10TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND COMPUTING (ICMLC 2018), 2018, :26-30
[2]   WSN-DS: A Dataset for Intrusion Detection Systems in Wireless Sensor Networks [J].
Almomani, Iman ;
Al-Kasasbeh, Bassam ;
AL-Akhras, Mousa .
JOURNAL OF SENSORS, 2016, 2016
[3]   Robust Federated Learning With Noisy Communication [J].
Ang, Fan ;
Chen, Li ;
Zhao, Nan ;
Chen, Yunfei ;
Wang, Weidong ;
Yu, F. Richard .
IEEE TRANSACTIONS ON COMMUNICATIONS, 2020, 68 (06) :3452-3464
[4]  
Biggio B., 2012, INT C MACHINE LEARNI
[5]  
Chawla Ashima, 2019, ECML PKDD 2018 Workshops. Nemesis 2018, UrbReas 2018, SoGood 2018 IWAISe 2018, and Green Data Mining 2018. Proceedings: Lecture Notes in Artificial Intelligence (LNAI 11329), P149, DOI 10.1007/978-3-030-13453-2_12
[6]  
Fang MH, 2020, PROCEEDINGS OF THE 29TH USENIX SECURITY SYMPOSIUM, P1623
[7]   Generative Adversarial Networks for Distributed Intrusion Detection in the Internet of Things [J].
Ferdowsi, Aidin ;
Saad, Walid .
2019 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM), 2019,
[8]  
Huang L., 2018, ARXIV181112629
[9]  
Kim G., 2016, arXiv.1611.01726
[10]  
Li T., 2018, arXiv preprint arXiv:1812.06127, P1