Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing

Modern cybersecurity best practices and standards require continuous Vulnerability Assessment (VA) and Penetration Test (PT). These activities are human- and time-expensive. The research community is trying to propose autonomous or semi-autonomous solutions based on Deep Reinforcement Learning (DRL)...

Full description

Saved in:
Bibliographic Details
Main Authors: Andrea Venturi, Mauro Andreolini, Mirco Marchetti, Michele Colajanni
Format: Article
Language:English
Published: Elsevier 2024-12-01
Series:Array
Subjects:
Online Access:http://www.sciencedirect.com/science/article/pii/S2590005624000316
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1846119443217776640
author Andrea Venturi
Mauro Andreolini
Mirco Marchetti
Michele Colajanni
author_facet Andrea Venturi
Mauro Andreolini
Mirco Marchetti
Michele Colajanni
author_sort Andrea Venturi
collection DOAJ
description Modern cybersecurity best practices and standards require continuous Vulnerability Assessment (VA) and Penetration Test (PT). These activities are human- and time-expensive. The research community is trying to propose autonomous or semi-autonomous solutions based on Deep Reinforcement Learning (DRL) agents, but current proposals require further investigations. We observe that related literature reports performance tests of the proposed agents against a limited subset of the hosts used to train the models, thus raising questions on their applicability in realistic scenarios. The main contribution of this paper is to fill this gap by investigating the generalization capabilities of existing DRL agents to extend their VAPT operations to hosts that were not used in the training phase. To this purpose, we define a novel VAPT environment through which we devise multiple evaluation scenarios. While evidencing the limited capabilities of shallow RL approaches, we consider three state-of-the-art deep RL agents, namely Deep Q-Network (DQN), Proximal Policy Optimization (PPO), and Advantage Actor–Critic (A2C), and use them as bases for VAPT operations. The results show that the algorithm using A2C DRL agent outperforms the others because it is more adaptable to unknown hosts and converges faster. Our methodology can guide future researchers and practitioners in designing a new generation of semi-autonomous VAPT tools that are suitable for real-world contexts.
format Article
id doaj-art-eb1f9eae75b2488c91ce56d34e40bee4
institution Kabale University
issn 2590-0056
language English
publishDate 2024-12-01
publisher Elsevier
record_format Article
series Array
spelling doaj-art-eb1f9eae75b2488c91ce56d34e40bee42024-12-17T05:00:22ZengElsevierArray2590-00562024-12-0124100365Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration TestingAndrea Venturi0Mauro Andreolini1Mirco Marchetti2Michele Colajanni3Department of Engineering “Enzo Ferrari”, University of Modena and Reggio Emilia, Italy; Corresponding author.Department of Physics, Computer Science and Mathematics, University of Modena and Reggio Emilia, ItalyDepartment of Engineering “Enzo Ferrari”, University of Modena and Reggio Emilia, ItalyDepartment of Computer Science and Engineering, University of Bologna, ItalyModern cybersecurity best practices and standards require continuous Vulnerability Assessment (VA) and Penetration Test (PT). These activities are human- and time-expensive. The research community is trying to propose autonomous or semi-autonomous solutions based on Deep Reinforcement Learning (DRL) agents, but current proposals require further investigations. We observe that related literature reports performance tests of the proposed agents against a limited subset of the hosts used to train the models, thus raising questions on their applicability in realistic scenarios. The main contribution of this paper is to fill this gap by investigating the generalization capabilities of existing DRL agents to extend their VAPT operations to hosts that were not used in the training phase. To this purpose, we define a novel VAPT environment through which we devise multiple evaluation scenarios. While evidencing the limited capabilities of shallow RL approaches, we consider three state-of-the-art deep RL agents, namely Deep Q-Network (DQN), Proximal Policy Optimization (PPO), and Advantage Actor–Critic (A2C), and use them as bases for VAPT operations. The results show that the algorithm using A2C DRL agent outperforms the others because it is more adaptable to unknown hosts and converges faster. Our methodology can guide future researchers and practitioners in designing a new generation of semi-autonomous VAPT tools that are suitable for real-world contexts.http://www.sciencedirect.com/science/article/pii/S2590005624000316Vulnerability assessmentPenetration testingDeep reinforcement learningDeep Q-networkAdversarial actor–criticProximal policy optimization
spellingShingle Andrea Venturi
Mauro Andreolini
Mirco Marchetti
Michele Colajanni
Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing
Array
Vulnerability assessment
Penetration testing
Deep reinforcement learning
Deep Q-network
Adversarial actor–critic
Proximal policy optimization
title Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing
title_full Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing
title_fullStr Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing
title_full_unstemmed Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing
title_short Assessing generalizability of Deep Reinforcement Learning algorithms for Automated Vulnerability Assessment and Penetration Testing
title_sort assessing generalizability of deep reinforcement learning algorithms for automated vulnerability assessment and penetration testing
topic Vulnerability assessment
Penetration testing
Deep reinforcement learning
Deep Q-network
Adversarial actor–critic
Proximal policy optimization
url http://www.sciencedirect.com/science/article/pii/S2590005624000316
work_keys_str_mv AT andreaventuri assessinggeneralizabilityofdeepreinforcementlearningalgorithmsforautomatedvulnerabilityassessmentandpenetrationtesting
AT mauroandreolini assessinggeneralizabilityofdeepreinforcementlearningalgorithmsforautomatedvulnerabilityassessmentandpenetrationtesting
AT mircomarchetti assessinggeneralizabilityofdeepreinforcementlearningalgorithmsforautomatedvulnerabilityassessmentandpenetrationtesting
AT michelecolajanni assessinggeneralizabilityofdeepreinforcementlearningalgorithmsforautomatedvulnerabilityassessmentandpenetrationtesting