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)...
Saved in:
Main Authors: | , , , |
---|---|
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 |