Deep Transfer Learning: A Novel Collaborative Learning Model for Cyberattack Detection Systems in IoT Networks

IEEE Internet of Things Journal - Tập 10 Số 10 - Trang 8578-8589 - 2023
Tran Viet Khoa1,2, Dinh Thai Hoang1, Nguyen Linh Trung2, Cong T. Nguyen1, Tran Thi Thuy Quynh2, Diep N. Nguyen1, Nguyen Viet Ha2, Eryk Dutkiewicz1
1School of Electrical and Data Engineering, University of Technology Sydney, Sydney, NSW, Australia
2Advanced Institute of Engineering and Technology, University of Engineering and Technology, Vietnam National University, Hanoi, Vietnam

Tóm tắt

Federated learning (FL) has recently become an effective approach for cyberattack detection systems, especially in Internet of Things (IoT) networks. By distributing the learning process across IoT gateways, FL can improve learning efficiency, reduce communication overheads, and enhance privacy for cyberattack detection systems. However, one of the biggest challenges for deploying FL in IoT networks is the unavailability of labeled data and dissimilarity of data features for training. In this article, we propose a novel collaborative learning framework that leverages Transfer Learning (TL) to overcome these challenges. Particularly, we develop a novel collaborative learning approach that enables a target network with unlabeled data to effectively and quickly learn “knowledge” from a source network that possesses abundant labeled data. It is important that the state-of-the-art studies require the participated data sets of networks to have the same features, thus limiting the efficiency, flexibility, as well as scalability of intrusion detection systems. However, our proposed framework can address these problems by exchanging the learning knowledge among various deep learning (DL) models, even when their data sets have different features. Extensive experiments on recent real-world cybersecurity data sets show that the proposed framework can improve more than 40% as compared to the state-of-the-art DL-based approaches.

Từ khóa

#Cyberattack detection #cybersecurity #deep learning (DL) #federated learning (FL) #Internet of Things (IoT) #transfer learning (TL)

Tài liệu tham khảo

2021, KDD Dataset

10.14722/ndss.2018.23204

10.1109/MilCIS.2015.7348942

2021, NSL-KDD Dataset

10.1016/j.patrec.2005.10.010

10.1109/MIS.2020.2988525

10.1109/MPRV.2018.03367731

goodfellow, 2016, Deep Learning

10.1109/ACCESS.2021.3118642

powers, 2011, Evaluation: From precision, recall and F-measure to ROC, informedness, markedness and correlation, J Mach Learn Technol, 2, 37

morgan, 2021, Global Ransomware damage costs predicted to hit $11 5 billion by 2019

keshet, 2020, Half of the malware detected in 2019 was classified as zero-day threats making it the most common malware to date

10.1109/WCNC.2018.8376973

10.1109/ACCESS.2019.2895334

10.1109/TII.2020.3023430

10.1109/MCOM.2018.1700332

10.1109/PIMRC.2015.7343651

10.1109/ACCESS.2020.3000476

10.1109/WoWMoM.2016.7523532

10.1109/ICCNC.2018.8390283

nguyen, 2019, DïoT: A federated self-learning anomaly detection system for IoT, Proc IEEE 39th Int Conf Distrib Comput Syst, 756

10.1109/MIS.2020.2988604

schneble, 2019, Attack detection using federated learning in medical cyber-physical systems, Proc 28th Int Conf Comput Commun Netw, 1

wen, 2019, Time series anomaly detection using convolutional neural networks and transfer learning, arXiv 1905 13628

10.1109/JIOT.2019.2910233

10.1109/WCNC45663.2020.9120761

10.1109/JPROC.2020.3004555

10.1109/TAI.2021.3054609

10.1109/MWC.004.2100542

10.1109/COMST.2020.2986024

khoa, 2022, Collaborative learning for Cyberattack detection in blockchain networks, arXiv 2203 11076

nguyen, 2021, Transfer learning for future wireless networks: A comprehensive survey, arXiv 2102 07572

10.1109/ACCESS.2020.3013541

Thông tin
Thông tin xuất bản

IEEE Internet of Things Journal

Tập 10 Số 10

8578-8589

Thông tin tác giả