Autonomous Network Defence using Reinforcement Learning

Myles Foley,Chris Hicks,Kate Highnam,Vasilios Mavroudis
DOI: https://doi.org/10.1145/3488932.3527286
2024-09-27
Abstract:In the network security arms race, the defender is significantly disadvantaged as they need to successfully detect and counter every malicious attack. In contrast, the attacker needs to succeed only once. To level the playing field, we investigate the effectiveness of autonomous agents in a realistic network defence scenario. We first outline the problem, provide the background on reinforcement learning and detail our proposed agent design. Using a network environment simulation, with 13 hosts spanning 3 subnets, we train a novel reinforcement learning agent and show that it can reliably defend continual attacks by two advanced persistent threat (APT) red agents: one with complete knowledge of the network layout and another which must discover resources through exploration but is more general.
Artificial Intelligence,Cryptography and Security,Machine Learning
What problem does this paper attempt to address?