Out of the Cage: How Stochastic Parrots Win in Cyber Security Environments

23 Aug 2023  ·  Maria Rigaki, Ondřej Lukáš, Carlos A. Catania, Sebastian Garcia ·

Large Language Models (LLMs) have gained widespread popularity across diverse domains involving text generation, summarization, and various natural language processing tasks. Despite their inherent limitations, LLM-based designs have shown promising capabilities in planning and navigating open-world scenarios. This paper introduces a novel application of pre-trained LLMs as agents within cybersecurity network environments, focusing on their utility for sequential decision-making processes. We present an approach wherein pre-trained LLMs are leveraged as attacking agents in two reinforcement learning environments. Our proposed agents demonstrate similar or better performance against state-of-the-art agents trained for thousands of episodes in most scenarios and configurations. In addition, the best LLM agents perform similarly to human testers of the environment without any additional training process. This design highlights the potential of LLMs to efficiently address complex decision-making tasks within cybersecurity. Furthermore, we introduce a new network security environment named NetSecGame. The environment is designed to eventually support complex multi-agent scenarios within the network security domain. The proposed environment mimics real network attacks and is designed to be highly modular and adaptable for various scenarios.

PDF Abstract

Datasets


  Add Datasets introduced or used in this paper
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
CyberBattleSim CyberBattleSim chain scenario DQN Average win rate 100 # 1
CyberBattleSim CyberBattleSim chain scenario GPT-4 Average win rate 100 # 1
NetSecGame NetSecGame full scenario defender GPT-4 Average win rate 50 # 1
NetSecGame NetSecGame full scenario no defender GPT-4 Average win rate 100 # 1
NetSecGame NetSecGame (RL) small scenario defender Q-learning Average win rate 77.96 # 2
NetSecGame NetSecGame (RL) small scenario defender GPT-4 Average win rate 83.33 # 1
NetSecGame NetSecGame (RL) small scenario no defender Q-learning Average win rate 67.41 # 2
NetSecGame NetSecGame (RL) small scenario no defender GPT-4 Average win rate 100 # 1

Methods