A Semi-Supervised Federated Learning Scheme via Knowledge Distillation for Intrusion Detection

Ruijie Zhao, Linbo Yang, Yijun Wang, Zhi Xue, Guan Gui, Tomoaki Ohtsuki

研究成果: Conference contribution

抄録

Federated learning (FL) has become an increasingly popular solution for intrusion detection to avoid data privacy leakage in Internet of Things (IoT) edge devices. However, most of the current FL-based intrusion detection methods still suffer from three limitations: (1) model parameters transmitted in each round may be used to recover private data which leads to security risks, (2) not independent and identically distributed (non-IID) private data seriously adversely affects the training of FL (especially distillation-based FL), and (3) high communication overhead caused by the large model size greatly hinders the actual deployment of the solution. To address these problems, this paper develops an intrusion detection method based on semi-supervised FL scheme via knowledge distillation. First, our proposed method leverages unlabeled data via distillation method to enhance the classifier performance. Second, we build a CNN-based model for extracting deep features of the traffic packets, and take this model as both the classifier network and discriminator network. Third, discriminator is designed to improve the quality of each client's predicted labels, to avoid the failure of distillation training caused by a large number of incorrect predictions under private non-IID data. Moreover, the combination of hard-label strategy and voting mechanism further reduces communication overhead. Experimental results on the real-world traffic dataset show that our proposed method can achieve better classification performance as well as lower communication overhead than state-of-the-art methods.

本文言語English
ホスト出版物のタイトルICC 2022 - IEEE International Conference on Communications
出版社Institute of Electrical and Electronics Engineers Inc.
ページ2688-2693
ページ数6
ISBN(電子版)9781538683477
DOI
出版ステータスPublished - 2022
イベント2022 IEEE International Conference on Communications, ICC 2022 - Seoul, Korea, Republic of
継続期間: 2022 5月 162022 5月 20

出版物シリーズ

名前IEEE International Conference on Communications
2022-May
ISSN(印刷版)1550-3607

Conference

Conference2022 IEEE International Conference on Communications, ICC 2022
国/地域Korea, Republic of
CitySeoul
Period22/5/1622/5/20

ASJC Scopus subject areas

  • コンピュータ ネットワークおよび通信
  • 電子工学および電気工学

フィンガープリント

「A Semi-Supervised Federated Learning Scheme via Knowledge Distillation for Intrusion Detection」の研究トピックを掘り下げます。これらがまとまってユニークなフィンガープリントを構成します。

引用スタイル