Abstract:In recent years, penetration testing (pen-testing) has emerged as a crucial process for evaluating the security level of network infrastructures by simulating real-world cyber-attacks. Automating pen-testing through reinforcement learning (RL) facilitates more frequent assessments, minimizes human effort, and enhances scalability. However, real-world pen-testing tasks often involve incomplete knowledge of the target network system. Effectively managing the intrinsic uncertainties via partially observable Markov decision processes (POMDPs) constitutes a persistent challenge within the realm of pen-testing. Furthermore, RL agents are compelled to formulate intricate strategies to contend with the challenges posed by partially observable environments, thereby engendering augmented computational and temporal expenditures. To address these issues, this study introduces EPPTA (efficient POMDP-driven penetration testing agent), an agent built on an asynchronous RL framework, designed for conducting pen-testing tasks within partially observable environments. We incorporate an implicit belief module in EPPTA, grounded on the belief update formula of the traditional POMDP model, which represents the agent's probabilistic estimation of the current environment state. Furthermore, by integrating the algorithm with the high-performance RL framework, sample factory, EPPTA significantly reduces convergence time compared to existing pen-testing methods, resulting in an approximately 20-fold acceleration. Empirical results across various pen-testing scenarios validate EPPTA's superior task reward performance and enhanced scalability, providing substantial support for efficient and advanced evaluation of network infrastructure security. The article introduces EPPTA, a reinforcement learning framework designed for penetration testing, and assesses its performance across diverse network configurations. EPPTA incorporates a belief module, augmenting its capacity to handle partially observable security challenges. EPPTA outperforms other methods in terms of convergence times, especially in larger and more complex scenarios, showcasing its scalability and adaptability to evolving security challenges.image

Mind the Gap: Towards Generalizable Autonomous Penetration Testing via Domain Randomization and Meta-Reinforcement Learning

MARNet: Backdoor Attacks Against Cooperative Multi-Agent Reinforcement Learning

GAIL-PT: An Intelligent Penetration Testing Framework with Generative Adversarial Imitation Learning

Behaviour-diverse automatic penetration testing: a coverage-based deep reinforcement learning approach

DynPen: Automated Penetration Testing in Dynamic Network Scenarios Using Deep Reinforcement Learning

EPPTA: Efficient Partially Observable Reinforcement Learning Agent for Penetration Testing Applications

SetTron: Towards Better Generalisation in Penetration Testing with Reinforcement Learning

Knowledge-Informed Auto-Penetration Testing Based on Reinforcement Learning with Reward Machine

Transferable Adversarial Attacks on Deep Reinforcement Learning with Domain Randomization

An Automated Penetration Testing Framework Based on Hierarchical Reinforcement Learning

A Hierarchical Deep Reinforcement Learning Model with Expert Prior Knowledge for Intelligent Penetration Testing

PentestAgent: Incorporating LLM Agents to Automated Penetration Testing

Generalization of Reinforcement Learning with Policy-Aware Adversarial Data Augmentation

Raijū: Reinforcement Learning-Guided Post-Exploitation for Automating Security Assessment of Network Systems

Hierarchical reinforcement learning for efficient and effective automated penetration testing of large networks

Autonomous Penetration Testing Based on Improved Deep Q-Network

Evaluation of Reinforcement Learning for Autonomous Penetration Testing using A3C, Q-learning and DQN

Understanding What Affects the Generalization Gap in Visual Reinforcement Learning: Theory and Empirical Evidence

DoorGym: A Scalable Door Opening Environment And Baseline Agent

Understanding Domain Randomization for Sim-to-real Transfer

Hacking, The Lazy Way: LLM Augmented Pentesting