用大模型实现自动网络故障响应,无需手动建模。
In-Context Autonomous Network Incident Response: An End-to-End Large Language Model Agent Approach
- 用140亿参数大模型整合感知、推理、规划、执行四功能。
- 在真实日志上比顶尖大模型快23%恢复故障。
- 可直接运行在普通硬件,支持按上下文动态调整策略。
快速演变的网络攻击要求响应系统具备自主学习和适应新威胁的能力。以往研究多采用强化学习方法,通过大量模拟来学习应对策略,但需人工构建仿真器,且忽略了原始系统日志与告警中的语义信息。为此,我们提出利用大语言模型(LLM)预训练的安全知识和上下文学习能力,构建端到端的智能体式响应方案。具体而言,该智能体将感知、推理、规划、行动四大功能集成于一个轻量级14B模型中。通过微调与思维链推理,模型能够处理系统日志,推断网络状态(感知),更新攻击模型假设(推理),模拟不同响应策略的后果(规划),并生成有效响应动作(行动)。通过将大模型模拟结果与实际观测对比,智能体可反复修正其攻击推测与对应响应,实现上下文自适应。该方法无需建模,可在普通硬件上运行。在文献报告的事件日志上评估,本方案比前沿大模型最快恢复时间提升23%。
原文摘要 · Abstract (English)
Rapidly evolving cyberattacks demand incident response systems that can autonomously learn and adapt to changing threats. Prior work has extensively explored the reinforcement learning approach, which involves learning response strategies through extensive simulation of the incident. While this approach can be effective, it requires handcrafted modeling of the simulator and suppresses useful semantics from raw system logs and alerts. To address these limitations, we propose to leverage large language models' (LLM) pre-trained security knowledge and in-context learning to create an end-to-end agentic solution for incident response planning. Specifically, our agent integrates four functionalities, perception, reasoning, planning, and action, into one lightweight LLM (14b model). Through fine-tuning and chain-of-thought reasoning, our LLM agent is capable of processing system logs and inferring the underlying network state (perception), updating its conjecture of attack models (reasoning), simulating consequences under different response strategies (planning), and generating an effective response (action). By comparing LLM-simulated outcomes with actual observations, the LLM agent repeatedly refines its attack conjecture and corresponding response, thereby demonstrating in-context adaptation. Our agentic approach is free of modeling and can run on commodity hardware. When evaluated on incident logs reported in the literature, our agent achieves recovery up to 23% faster than those of frontier LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。