摘要
本发明提出一种基于深度强化学习与大语言模型分层决策的智能化渗透测试系统,属于渗透测试技术领域。该系统包括:宏观动作决策模型、动作库、状态信息编码器、目标选择器、下游任务处理模型、工具库;其中:目标选择器根据目标网络环境的探测结果选择待测试目标;宏观动作决策模型使用深度强化学习算法,接收目标环境的状态向量,并根据ATT&CK框架从动作库中选择并输出下一步要完成的渗透测试任务,输出的动作作为下游任务目标发布给下游任务处理模型;下游任务处理模型为大语言模型,其根据任务描述自主生成解决方案并调用工具库中的渗透测试工具进行执行,任务执行后的反馈结果由大语言模型观察并记录,并由其判断该下游任务是否完成。
技术关键词
渗透测试系统
渗透测试工具
大语言模型
决策
深度强化学习算法
信息编码器
主机
分层
队列
网络
深度强化学习模型
渗透测试技术
策略更新
系统启动阶段
ICMP协议
深度优先搜索
在线
传播算法