Efficient Federated Intrusion Detection in 5G ecosystem using optimized BERT-based model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Adjewa, Frederic, Esseghir, Moez, Merghem-Boulahia, Leila
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916414111088640
author Adjewa, Frederic
Esseghir, Moez
Merghem-Boulahia, Leila
author_facet Adjewa, Frederic
Esseghir, Moez
Merghem-Boulahia, Leila
contents The fifth-generation (5G) offers advanced services, supporting applications such as intelligent transportation, connected healthcare, and smart cities within the Internet of Things (IoT). However, these advancements introduce significant security challenges, with increasingly sophisticated cyber-attacks. This paper proposes a robust intrusion detection system (IDS) using federated learning and large language models (LLMs). The core of our IDS is based on BERT, a transformer model adapted to identify malicious network flows. We modified this transformer to optimize performance on edge devices with limited resources. Experiments were conducted in both centralized and federated learning contexts. In the centralized setup, the model achieved an inference accuracy of 97.79%. In a federated learning context, the model was trained across multiple devices using both IID (Independent and Identically Distributed) and non-IID data, based on various scenarios, ensuring data privacy and compliance with regulations. We also leveraged linear quantization to compress the model for deployment on edge devices. This reduction resulted in a slight decrease of 0.02% in accuracy for a model size reduction of 28.74%. The results underscore the viability of LLMs for deployment in IoT ecosystems, highlighting their ability to operate on devices with constrained computational and storage resources.
format Preprint
id arxiv_https___arxiv_org_abs_2409_19390
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient Federated Intrusion Detection in 5G ecosystem using optimized BERT-based model
Adjewa, Frederic
Esseghir, Moez
Merghem-Boulahia, Leila
Cryptography and Security
Artificial Intelligence
The fifth-generation (5G) offers advanced services, supporting applications such as intelligent transportation, connected healthcare, and smart cities within the Internet of Things (IoT). However, these advancements introduce significant security challenges, with increasingly sophisticated cyber-attacks. This paper proposes a robust intrusion detection system (IDS) using federated learning and large language models (LLMs). The core of our IDS is based on BERT, a transformer model adapted to identify malicious network flows. We modified this transformer to optimize performance on edge devices with limited resources. Experiments were conducted in both centralized and federated learning contexts. In the centralized setup, the model achieved an inference accuracy of 97.79%. In a federated learning context, the model was trained across multiple devices using both IID (Independent and Identically Distributed) and non-IID data, based on various scenarios, ensuring data privacy and compliance with regulations. We also leveraged linear quantization to compress the model for deployment on edge devices. This reduction resulted in a slight decrease of 0.02% in accuracy for a model size reduction of 28.74%. The results underscore the viability of LLMs for deployment in IoT ecosystems, highlighting their ability to operate on devices with constrained computational and storage resources.
title Efficient Federated Intrusion Detection in 5G ecosystem using optimized BERT-based model
topic Cryptography and Security
Artificial Intelligence
url https://arxiv.org/abs/2409.19390