arXiv:2503.03108cs.CRcs.AI2025-03被引 7

用大模型+检索增强,自动识别攻击链并减少误报

OMNISEC: LLM-Driven Provenance-based Intrusion Detection via Retrieval-Augmented Behavior Prompting

  • 通过构建可疑节点和罕见路径定位异常行为
  • 在公开数据集上准确率超现有方法,误报显著降低
  • 适合安全分析师快速定位真实攻击,节省人工排查时间

近年来,基于溯源的入侵检测系统(PIDS)广泛用于终端威胁分析。这类系统可分为基于规则和基于学习两类。由于攻击技术持续演进,规则无法动态建模攻击者所有特征,常导致漏报。基于学习的系统又分为监督学习和异常检测,前者因攻击样本稀缺难以应用,后者则因无法区分正常行为变化与真实攻击而产生大量误报。检测结果的准确性直接影响后续安全分析师的人工处理成本。为降低人工分析负担,本文提出OMNISEC,通过检索增强的行为提示(Retrieval-Augmented Behavior Prompting)将大语言模型(LLM)应用于异常检测系统。OMNISEC能通过构建可疑节点和罕见路径识别异常节点及事件,并结合两个外部知识库,利用检索增强生成(RAG)使LLM判断异常行为是否为真实攻击。最终,系统可重建攻击图,还原攻击者的完整行为链。实验表明,OMNISEC在公开基准数据集上优于当前最优方法。

原文摘要 · Abstract (English)

Recently, Provenance-based Intrusion Detection Systems (PIDSes) have been widely used for endpoint threat analysis. These studies can be broadly categorized into rule-based detection systems and learning-based detection systems. Among these, due to the evolution of attack techniques, rules cannot dynamically model all the characteristics of attackers. As a result, such systems often face false negatives. Learning-based detection systems are further divided into supervised learning and anomaly detection. The scarcity of attack samples hinders the usability and effectiveness of supervised learning-based detection systems in practical applications. Anomaly-based detection systems face a massive false positive problem because they cannot distinguish between changes in normal behavior and real attack behavior. The alert results of detection systems are closely related to the manual labor costs of subsequent security analysts. To reduce manual analysis time, we propose OMNISEC, which applies large language models (LLMs) to anomaly-based intrusion detection systems via retrieval-augmented behavior prompting. OMNISEC can identify abnormal nodes and corresponding abnormal events by constructing suspicious nodes and rare paths. By combining two external knowledge bases, OMNISEC uses Retrieval Augmented Generation (RAG) to enable the LLM to determine whether abnormal behavior is a real attack. Finally, OMNISEC can reconstruct the attack graph and restore the complete attack behavior chain of the attacker's intrusion. Experimental results show that OMNISEC outperforms state-of-the-art methods on public benchmark datasets.

入侵检测大模型应用攻击溯源RAG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。