Federated Learning for Attack Prediction on UNSW-NB15 Training Data

Year 2024, Volume: 5 Issue: 2, 1 - 7, 30.12.2024

Hayriye Tanyıldız , Canan Batur Şahin , Özlem Batur Dinler

https://doi.org/10.46572/naturengs.1524394

Abstract

With the increasing prevalence of cyber threats, robust network security has become imperative. Traditional centralized machine learning models for network intrusion detection systems (NIDS) face significant challenges related to data confidentiality, scalability, and centralization risks. Federated Learning (FL) offers a decentralized approach that allows multiple clients to train a shared model while keeping their data local collaboratively. This technique is particularly relevant to attack prediction in cybersecurity, where organizations may be hesitant to share sensitive data due to privacy and security concerns. This paper investigates the application of FL on the UNSW-NB15 dataset to predict network attacks. The study demonstrates the potential of FL to achieve high accuracy and minimal false negatives in attack detection while preserving data confidentiality. The results visualized in the confusion matrix highlight the effectiveness of FL in distinguishing between normal and malicious network traffic, making it a promising approach for real-world cybersecurity applications. By leveraging FL, organizations can improve their network security infrastructure while reducing the risks associated with centralized data processing.

Keywords

Federated Learning, Anaomaly Detection, Cyber Attack

References

• Shaoqin Li, Zhendong Wang, Shuxin Yang, Xiao Luo, Daojing He, Sammy Chan, (2024), Internet of Things intrusion detection: Research and practice of NSENet and LSTM fusion models, Egyptian Informatics Journal, 26, 100476.
• Sharma, J., Giri, C., Granmo, O. C., & Goodwin, M. (2019). Multi-layer intrusion detection system with ExtraTrees feature selection, extreme learning machine ensemble, and softmax aggregation. EURASIP Journal on Information Security, 2019(1), 1-16.
• H. Gharaee, H. Hosseinvand, in 2016 8th International Symposium on Telecommunications (IST). A new feature selection is based on genetic algorithm and SVM, (2016), pp. 139–144.
• T. Salman, D. Bhamare, A. Erbad, R. Jain and M. Samaka, (2017), "Machine Learning for Anomaly Detection and Categorization in Multi-Cloud Environments," 2017 IEEE 4th International Conference on Cyber Security and Cloud Computing (CSCloud), New York, NY, USA, 2017, pp. 97-103.
• Zhang, C., Xie, Y., Bai, H., Yu, B., Li, W., & Gao, Y. (2021). A survey on federated learning. Knowledge- Based Systems, 216, 106775.
• Li Li, Yuxi Fan, Mike Tse, Kuo-Yi Lin, (2020), A review of applications in federated learning, Computers & Industrial Engineering, 149,106854.
• Konečný, J., McMahan, H. B., Yu, F. X., Richtárik, P., Suresh, A. T., & Bacon, D. (2016). Federated learning: Strategies for improving communication efficiency.
• Tanyıldız, H., Batur Şahin, C., & Batur Dinler, Ö. (2024). Disrupting Downtime: Different Deep Learning Journeys into Predictive Maintenance Anomaly Detection. NATURENGS, 5(1), 47-53.
• Tanyıldız, H., Batur Şahin, C., & Batur Dinler, Ö. (2024). Enhancing Cybersecurity through GAN-Augmented and Hybrid Feature Selection Machine Learning Models: A Case Study on EVSE Data. NATURENGS, 5(1), 61-70.
• C. B. Şahin, "DCW-RNN: Improving Class Level Metrics for Software Vulnerability Detection Using Artificial Immune System with Clock-Work Recurrent Neural Network," 2021 International Conference on Innovations in Intelligent Systems and Applications (INISTA), Kocaeli, Turkey, 2021, pp. 1-8.
• Ulah, A., Aznaoui, H., Batur S¸ahin, C., Sadie, M., Dinler, O.: Cloud computing and 5G challenges and open issues. Int. J. Adv. Appl. Sci. (2022).
• Ullah, A., Şahin, C. B., Dinler, O. B., Khan, M. H., & Aznaoui, H. (2021). Heart disease prediction using various machine learning approaches. Journal of Cardiovascular Disease Research, 12(3), 379–391.
• Ozlem Batur Dinler, Canan Batur Şahin, & Hanane Aznaoui. (2024). HYBRID MODEL USED FOR REDUCING LATENCY IN SMART HEALTHCARE SYSTEMS. Journal of Advancement in Computing, 2(1), 10–20.
• Mikail Mohammed Salim, Abir El Azzaoui, Xianjun Deng, Jong Hyuk Park, (2024). FL-CTIF: A federated learning based CTI framework based on information fusion for secure IIoT, Information Fusion, 102, 102074.
• Parimala Boobalan, Swarna Priya Ramu, Quoc-Viet Pham, Kapal Dev, Sharnil Pandya, Praveen Kumar Reddy Maddikunta, Thippa Reddy Gadekallu, Thien Huynh-The, (2022). Fusion of Federated Learning and Industrial Internet of Things: A survey, Computer Networks, 212,109048.