Give and Take: Federated Transfer Learning for Industrial IoT Network Intrusion Detection
Abstract: The rapid growth in Internet of Things (IoT) technology has become an integral part of today's industries forming the Industrial IoT (IIoT) initiative, where industries are leveraging IoT to improve communication and connectivity via emerging solutions like data analytics and cloud computing. Unfortunately, the rapid use of IoT has made it an attractive target for cybercriminals. Therefore, protecting these systems is of utmost importance. In this paper, we propose a federated transfer learning (FTL) approach to perform IIoT network intrusion detection. As part of the research, we also propose a combinational neural network as the centerpiece for performing FTL. The proposed technique splits IoT data between the client and server devices to generate corresponding models, and the weights of the client models are combined to update the server model. Results showcase high performance for the FTL setup between iterations on both the IIoT clients and the server. Additionally, the proposed FTL setup achieves better overall performance than contemporary machine learning algorithms at performing network intrusion detection.
- U. Cisco, “Cisco annual internet report (2018–2023) white paper,” Cisco: San Jose, CA, USA, vol. 10, no. 1, pp. 1–35, 2020.
- C. Cyrus, “Iot cyberattacks escalate in 2021, according to kaspersky,” IoT World Today, https://www. iotworldtoday. com/2021/09/17/iot-cyberattacks-escalate-in-2021-according-to-kaspersky/, last accessed December, 2021.
- R. Al-amri, R. K. Murugesan, M. Man, A. F. Abdulateef, M. A. Al-Sharafi, and A. A. Alkahtani, “A review of machine learning and deep learning techniques for anomaly detection in iot data,” Applied Sciences, vol. 11, no. 12, p. 5320, 2021.
- N. V. Sharma and N. S. Yadav, “An optimal intrusion detection system using recursive feature elimination and ensemble of classifiers,” Microprocessors and Microsystems, vol. 85, p. 104293, 2021.
- C. Zhang, Y. Xie, H. Bai, B. Yu, W. Li, and Y. Gao, “A survey on federated learning,” Knowledge-Based Systems, vol. 216, p. 106775, 2021.
- F. Zhuang, Z. Qi, K. Duan, D. Xi, Y. Zhu, H. Zhu, H. Xiong, and Q. He, “A comprehensive survey on transfer learning,” Proceedings of the IEEE, vol. 109, no. 1, pp. 43–76, 2020.
- P. Ioulianou, V. Vasilakis, I. Moscholios, and M. Logothetis, “A signature-based intrusion detection system for the internet of things,” Information and Communication Technology Form, 2018.
- W. Li, S. Tug, W. Meng, and Y. Wang, “Designing collaborative blockchained signature-based intrusion detection in iot environments,” Future Generation Computer Systems, vol. 96, pp. 481–489, 2019.
- N. U. Sheikh, H. Rahman, S. Vikram, and H. AlQahtani, “A lightweight signature-based ids for iot environment,” arXiv preprint arXiv:1811.04582, 2018.
- B. A. Tama and K.-H. Rhee, “Attack classification analysis of iot network via deep learning approach,” Res. Briefs Inf. Commun. Technol. Evol.(ReBICTE), vol. 3, pp. 1–9, 2017.
- B. Roy and H. Cheung, “A deep learning approach for intrusion detection in internet of things using bi-directional long short-term memory recurrent neural network,” in 2018 28th international telecommunication networks and applications conference (ITNAC). IEEE, 2018, pp. 1–6.
- N. Chaabouni, M. Mosbah, A. Zemmari, C. Sauvignac, and P. Faruki, “Network intrusion detection for iot security based on learning techniques,” IEEE Communications Surveys & Tutorials, vol. 21, no. 3, pp. 2671–2701, 2019.
- A. Alsaedi, N. Moustafa, Z. Tari, A. Mahmood, and A. Anwar, “Ton_iot telemetry dataset: A new generation dataset of iot and iiot for data-driven intrusion detection systems,” Ieee Access, vol. 8, pp. 165 130–165 150, 2020.
- H. Hindy, E. Bayne, M. Bures, R. Atkinson, C. Tachtatzis, and X. Bellekens, “Machine learning based iot intrusion detection system: An mqtt case study (mqtt-iot-ids2020 dataset),” in Selected Papers from the 12th International Networking Conference: INC 2020. Springer, 2021, pp. 73–84.
- M. A. Ferrag, O. Friha, D. Hamouda, L. Maglaras, and H. Janicke, “Edge-iiotset: A new comprehensive realistic cyber security dataset of iot and iiot applications for centralized and federated learning,” IEEE Access, vol. 10, pp. 40 281–40 306, 2022.
- D. C. Attota, V. Mothukuri, R. M. Parizi, and S. Pouriyeh, “An ensemble multi-view federated learning intrusion detection for iot,” IEEE Access, vol. 9, pp. 117 734–117 745, 2021.
- O. Friha, M. A. Ferrag, L. Shu, L. Maglaras, K.-K. R. Choo, and M. Nafaa, “Felids: Federated learning-based intrusion detection system for agricultural internet of things,” Journal of Parallel and Distributed Computing, vol. 165, pp. 17–31, 2022.
- J. H. Steiger, “Tests for comparing elements of a correlation matrix.” Psychological bulletin, vol. 87, no. 2, p. 245, 1980.
- L. Wen, X. Li, and L. Gao, “A transfer convolutional neural network for fault diagnosis based on resnet-50,” Neural Computing and Applications, vol. 32, pp. 6111–6124, 2020.
- E. Aminanto and K. Kim, “Deep learning in intrusion detection system: An overview,” in 2016 International Research Conference on Engineering and Technology (2016 IRCET). Higher Education Forum, 2016.
- P. Covington, J. Adams, and E. Sargin, “Deep neural networks for youtube recommendations,” in Proceedings of the 10th ACM conference on recommender systems, 2016, pp. 191–198.
- S. Singh, “PPML Series #2 - Federated Optimization Algorithms - FedSGD and FedAvg,” Dec. 2021. [Online]. Available: https://shreyansh26.github.io/post/2021-12-18_federated_optimization_fedavg/
- S. Shitharth, P. R. Kshirsagar, P. K. Balachandran, K. H. Alyoubi, and A. O. Khadidos, “An innovative perceptual pigeon galvanized optimization (ppgo) based likelihood naïve bayes (lnb) classification approach for network intrusion detection system,” IEEE Access, vol. 10, pp. 46 424–46 441, 2022.
- P. A. A. Resende and A. C. Drummond, “A survey of random forest based methods for intrusion detection systems,” ACM Computing Surveys (CSUR), vol. 51, no. 3, pp. 1–36, 2018.
- F. Abbasi, M. Naderan, and S. E. Alavi, “Anomaly detection in internet of things using feature selection and classification based on logistic regression and artificial neural network on n-baiot dataset,” in 2021 5th International Conference on Internet of Things and Applications (IoT). IEEE, 2021, pp. 1–7.
- M. Seol and T. Kim, “Performance enhancement in federated learning by reducing class imbalance of non-iid data,” Sensors, vol. 23, no. 3, p. 1152, 2023.
- B. Thiyam and S. Dey, “Efficient feature evaluation approach for a class-imbalanced dataset using machine learning,” Procedia Computer Science, vol. 218, pp. 2520–2532, 2023.
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.