arXiv:2604.14166cs.CL2026-04被引 1

用分层检索提升威胁情报标注准确率与效率

Hierarchical Retrieval Augmented Generation for Adversarial Technique Annotation in Cyber Threat Intelligence Text

论文配图:Hierarchical Retrieval Augmented Generation for Adversarial Technique Annotation in Cyber Threat Intelligence Text
图 1 · 摘自论文原文
  • 分两阶段检索:先定战术目标,再查具体技术
  • F1提升3.8%,推理延迟降低62.4%
  • 适合需要可解释性的网络安全分析场景

将网络威胁情报(CTI)文本映射到MITRE ATT&CK技术编号是理解攻击者行为和自动化防御的关键任务。现有检索增强生成(RAG)方法依赖扁平化检索,忽略ATT&CK框架中技术依战术分层的结构。本文提出H-TechniqueRAG,通过引入战术-技术层级作为强归纳偏置,实现高效精准标注。该方法采用两阶段分层检索:首先识别宏观战术(攻击者技术目标),再在对应战术内缩小技术搜索范围,候选空间减少77.5%。为弥合检索与生成差距,设计了战术感知重排序模块和层级约束上下文组织策略,缓解大模型上下文过载,提升推理精度。在三个不同CTI数据集上的实验表明,H-TechniqueRAG不仅比最先进方法TechniqueRAG的F1分数高出3.8%,还实现推理延迟降低62.4%、LLM API调用减少60%。进一步分析显示,其层级结构先验增强了跨领域泛化能力,并为安全分析师提供可解释的逐步决策路径。

原文摘要 · Abstract (English)

Mapping Cyber Threat Intelligence (CTI) text to MITRE ATT\&CK technique IDs is a critical task for understanding adversary behaviors and automating threat defense. While recent Retrieval-Augmented Generation (RAG) approaches have demonstrated promising capabilities in this domain, they fundamentally rely on a flat retrieval paradigm. By treating all techniques uniformly, these methods overlook the inherent taxonomy of the ATT\&CK framework, where techniques are structurally organized under high-level tactics. In this paper, we propose H-TechniqueRAG, a novel hierarchical RAG framework that injects this tactic-technique taxonomy as a strong inductive bias to achieve highly efficient and accurate annotation. Our approach introduces a two-stage hierarchical retrieval mechanism: it first identifies the macro-level tactics (the adversary's technical goals) and subsequently narrows the search to techniques within those tactics, effectively reducing the candidate search space by 77.5\%. To further bridge the gap between retrieval and generation, we design a tactic-aware reranking module and a hierarchy-constrained context organization strategy that mitigates LLM context overload and improves reasoning precision. Comprehensive experiments across three diverse CTI datasets demonstrate that H-TechniqueRAG not only outperforms the state-of-the-art TechniqueRAG by 3.8\% in F1 score, but also achieves a 62.4\% reduction in inference latency and a 60\% decrease in LLM API calls. Further analysis reveals that our hierarchical structural priors equip the model with superior cross-domain generalization and provide security analysts with highly interpretable, step-by-step decision paths.

威胁情报分层检索ATT&CKRAG

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