用图神经网络构建动态攻击路径模拟环境,助力自动化网络安全分析。
SubstratumGraphEnv: Reinforcement Learning Environment (RLE) for Modeling System Attack Paths
- 基于Sysmon日志构建系统事件图,动态建模进程父子关系。
- 设计Gymnasium环境与PyTorch接口,实现图数据向DRL的转化。
- 结合GCN与A2C模型,识别关键系统事件属性以指导攻防训练。
自动化网络安全部署,特别是潜在攻击路径的识别,面临重大挑战。由于系统事件具有序列性、关联性和演化性,大多数人工智能技术难以有效建模。本文提出一种强化学习(RL)环境生成框架,模拟Windows操作系统上执行的进程序列,实现对恶意进程的动态建模。该方法采用图表示法刻画操作系统状态及其转移,图结构来源于开源System Monitor(Sysmon)日志。为应对系统事件类型、字段和日志格式的多样性,开发了捕获并建模父-子进程关系的机制。构建了名为SubstratumGraphEnv的Gymnasium环境,作为强化学习感知基础,并开发了定制化的PyTorch接口(SubstratumBridge),将Gymnasium图转换为深度强化学习(DRL)的观测值和离散动作。图卷积网络(GCNs)用于提取图的局部与全局状态,输入到优势演员-评论家(A2C)模型的策略头与价值头中。本工作的核心贡献在于设计了一种新型深度图形化强化学习环境,可自动翻译用户与系统事件序列,为网络安全分析提供关键上下文。该工作为未来研究训练参数设计与高级奖励设计奠定基础,同时揭示了哪些系统事件属性对训练自主强化学习代理至关重要。
原文摘要 · Abstract (English)
Automating network security analysis, particularly the identification of potential attack paths, presents significant challenges. Due in part to the sequential, interconnected, and evolutionary nature of system events which most artificial intelligence (AI) techniques struggle to model effectively. This paper proposes a Reinforcement Learning (RL) environment generation framework that simulates the sequence of processes executed on a Windows operating system, enabling dynamic modeling of malicious processes on a system. This methodology models operating system state and transitions using a graph representation. This graph is derived from open-source System Monitor (Sysmon) logs. To address the variety in system event types, fields, and log formats, a mechanism was developed to capture and model parent-child processes from Sysmon logs. A Gymnasium environment (SubstratumGraphEnv) was constructed to establish the perceptible basis for an RL environment, and a customized PyTorch interface was also built (SubstratumBridge) to translate Gymnasium graphs into Deep Reinforcement Learning (DRL) observations and discrete actions. Graph Convolutional Networks (GCNs) concretize the graph's local and global state, which feed the distinct policy and critic heads of an Advantage Actor-Critic (A2C) model. This work's central contribution lies in the design of a novel deep graphical RL environment that automates translation of sequential user and system events, furnishing crucial context for cybersecurity analysis. This work provides a foundation for future research into shaping training parameters and advanced reward shaping, while also offering insight into which system events attributes are critical to training autonomous RL agents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。