Fine-Tuning Large Language Models for Network Traffic Analysis in Cyber Security

Ortal Lavi, Ofir Manor, Tomer Schwartz, Andres F. Murillo, Ayoub Messous, Motoyoshi Sekiya, Junichi Suga, Kenji Hikichi, Yuki Unno

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Language modelling has demonstrated the exceptional interpretation and analysis capabilities of Large Language Models (LLMs) in addressing a wide range of Natural Language Processings (NLPs) tasks. However, LLMs still face challenges in technical network text analysis due to its special terminology, syntax, and protocols which are not consistent with conventional human language. In this paper, we introduce a method for fine-tuning NLP-based LLMs on technical network text intended for machine comprehension. We apply our method to detect and classify different types of attacks in network traffic flows. To validate our method, we used the Kitsune Network Attack Dataset and our results show that our method is able to correctly classify the network attacks, even in adverse conditions, when the model receives irrelevant context for classification. This study highlights how the capabilities of LLMs can be extended across different unknown domains, specifically cyber security, thereby enhancing their ability to detect anomalies, classify them, and distinguish between various types of attacks. Our experiments show that fine-tuned models significantly outperform non-fine-tuned models, underlining the effectiveness of fine-tuning LLMs for cyber security applications involving the analysis of technical text data.

Original languageEnglish
Title of host publicationProceedings - 2024 IEEE Conference on Dependable and Secure Computing, DSC 2024
PublisherInstitute of Electrical and Electronics Engineers
Pages45-50
Number of pages6
ISBN (Electronic)9798331540289
DOIs
StatePublished - 1 Jan 2024
Externally publishedYes
Event7th IEEE Conference on Dependable and Secure Computing, DSC 2024 - Virtual, Tokyo, Japan
Duration: 6 Nov 20248 Nov 2024

Publication series

NameProceedings - 2024 IEEE Conference on Dependable and Secure Computing, DSC 2024

Conference

Conference7th IEEE Conference on Dependable and Secure Computing, DSC 2024
Country/TerritoryJapan
CityVirtual, Tokyo
Period6/11/248/11/24

Keywords

  • cyber security
  • large language models
  • network anomaly classification

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Networks and Communications
  • Computer Science Applications
  • Computer Vision and Pattern Recognition
  • Software
  • Safety, Risk, Reliability and Quality

Fingerprint

Dive into the research topics of 'Fine-Tuning Large Language Models for Network Traffic Analysis in Cyber Security'. Together they form a unique fingerprint.

Cite this