Smart Computer Security Audit: Reinforcement Learning with a Deep Neural Network Approximator
Pozdniakov, K., Alonso, E. ORCID: 0000-0002-3306-695X, Stankovic, V. ORCID: 0000-0002-8740-6526 , Tam, K. & Jones, K. (2020). Smart Computer Security Audit: Reinforcement Learning with a Deep Neural Network Approximator. In: 2020 International Conference on Cyber Situational Awareness, Data Analytics and Assessment (CyberSA). cyber2020, 15-17 Jun 2020, Dublin, Ireland.
Abstract
A significant challenge in modern computer security is the growing skill gap as intruder capabilities increase, making it necessary to begin automating elements of penetration testing so analysts can contend with the growing number of cyber threats. In this paper, we attempt to assist human analysts by automating a single host penetration attack. To do so, a smart agent performs different attack sequences to find vulnerabilities in a target system. As it does so, it accumulates knowledge, learns new attack sequences and improves its own internal penetration testing logic. As a result, this agent (AgentPen for simplicity) is able to successfully penetrate hosts it has never interacted with before. A computer security administrator using this tool would receive a comprehensive, automated sequence of actions leading to a security breach, highlighting potential vulnerabilities, and reducing the amount of menial tasks a typical penetration tester would need to execute. To achieve autonomy, we apply an unsupervised machine learning algorithm, Q-learning, with an approximator that incorporates a deep neural network architecture. The security audit itself is modelled as a Markov Decision Process in order to test a number of decisionmaking strategies and compare their convergence to optimality. A series of experimental results is presented to show how this approach can be effectively used to automate penetration testing using a scalable, i.e. not exhaustive, and adaptive approach.
Publication Type: | Conference or Workshop Item (Paper) |
---|---|
Additional Information: | © 2020 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works. |
Publisher Keywords: | Pentesting, audit, Q-learning, reinforcement learning, deep neural network |
Subjects: | Q Science > QA Mathematics > QA75 Electronic computers. Computer science R Medicine > RC Internal medicine > RC0321 Neuroscience. Biological psychiatry. Neuropsychiatry |
Departments: | School of Science & Technology > Computer Science |
Download (1MB) | Preview
Export
Downloads
Downloads per month over past year