arrow
Return

Automatic penetration testing model based on reinforcement learning for complex network environments

delete2025-07-18
delete0
PRE
AI
Y
Yang Chen
J
Junjiang He
W
Wenbo Fang *
S
Shenwen Yang
J
Jiangchuan Chen
李
李涛 (Tao Li)
兰小龙 cover
兰小龙 (Xiaolong Lan)
DOI:10.1007/s11227-025-07656-2delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Penetration testing is an effective method for assessing cybersecurity posture. However, as cyber attacks become increasingly sophisticated and network environments grow in complexity, traditional intelligent penetration testing models encounter significant challenges in efficiently identifying optimal strategies and achieving rapid convergence. To address these issues, this paper proposes an automatic penetration testing model based on reinforcement learning, which abstracts the testing process as a Markov Decision Process (MDP) and incorporates a curiosity mechanism to enhance exploration efficiency. The core of this approach lies in training an agent using a reinforcement learning algorithm equipped with a curiosity mechanism, enabling it to automatically explore and select optimal penetration testing strategies based on reward feedback from the environment, and to persist until sensitive privileges are obtained. Experiments conducted in a simulated network environment with 50 hosts and millions of states demonstrate that the proposed model reduces convergence time by approximately 50% compared to the standard Proximal Policy Optimization (PPO) algorithm. By integrating GPU acceleration within a high-performance computing (HPC) framework and employing distributed parallel processing, the model effectively satisfies the real-time and scalability requirements of penetration testing in complex network environments.
Keywords:
Penetration testing
Reinforcement learning
Curiosity mechanism
HPC

Journal

Journal of Supercomputing cover
Journal of Supercomputing
IF:
2.7
Papers:
1.2K
Citations:
1.0W

Organization

S
School of Cyber Science and Engineering
Scholars:
200
Papers: 81
Citations: 0
Cited Papers

Cited Papers

Extending Attack Graphs to Represent Cyber-Attacks in Communication Protocols and Modern IT Networks
err2022-05-01
err26
errOAAI
errStan, Orly; Bitton, Ron; Ezrets, Michal; Dadon, Moran; Inokuchi, Masaki; Ohta, Yoshinobu; Yagyu, Tomohiko; Elovici, Yuval; Shabtai, Asaf
errShare
errSave
A review of attack graph and attack tree visual syntax in cyber security
err2020-02-01
err121
errOAAI
errLallie, Harjinder Singh; Debattista, Kurt; Bal, Jay
errShare
errSave
A hierarchical deep reinforcement learning model with expert prior knowledge for intelligent penetration testing
err2023-09-01
err11
PREAI
errLi, Qianyu; Zhang, Min; Shen, Yi; Wang, Ruipeng; Hu, Miao; Li, Yang; Hao, Hao
errShare
errSave
GAIL-PT: An intelligent penetration testing framework with generative adversarial imitation learning
err2023-03-01
err13
PREAI
errChen, Jinyin; Hu, Shulong; Zheng, Haibin; Xing, Changyou; Zhang, Guomin
errShare
errSave
An Efficient Algorithm for Finding Modules in Fault Trees
err2021-09-01
err12
PREAI
errChen, Mei; Xiao, Ning-Cong; Zuo, Ming J.; Ding, Yi
errShare
errSave
Cyborg Intelligence: Towards Bio-Machine Intelligent Systems INTRODUCTION
err2014-11-01
err6
errOAAI
errWu, Zhaohui; Pan, Gang; Principe, Jose C.; Cichocki, Andrzej
errShare
errSave
errShare
errSave
Automated generation and analysis of attack graphs
err2024-11-19
err0
PREAI
errO. Sheyner; J. Haines; S. Jha; R. Lippmann; J.M. Wing
errShare
errSave
researcher View more