arXiv:2601.14556cs.LGcs.CR2026-01被引 2

用传统机器学习构建多层次分类模型,自动标注网络安全威胁文本。

Constructing Multi-label Hierarchical Classification Models for MITRE ATT&CK Text Tagging

  • 基于分层任务空间设计多标签层级分类模型,逐步优化标注流程。
  • 战术层级准确率达94%,技术层级达82%,超越GPT-4o的60%表现。
  • 仅依赖经典机器学习,适合需可解释性与低资源部署的安全团队。

MITRE ATT&CK 是一个网络安全知识库,将攻击者行为按战术和技法组织,用于描述攻击动机与方法。当前其文本标注仍主要依赖人工。本文提出一种分层的‘任务空间’框架,系统化归纳自动化研究路径,并基于通用威胁情报文本实验构建多标签层级分类模型。通过公开共享工具与模型(https://github.com/jpmorganchase/MITRE_models),实现战术层级约94%准确率、技术层级约82%准确率。模型性能优于或媲美现有最优结果,且不依赖大语言模型(LLMs)、RAG或复杂层级结构。实验还验证了在金融领域专家编写的威胁场景数据集上的有效性。

原文摘要 · Abstract (English)

MITRE ATT&CK is a cybersecurity knowledge base that organizes threat actor and cyber-attack information into a set of tactics describing the reasons and goals threat actors have for carrying out attacks, with each tactic having a set of techniques that describe the potential methods used in these attacks. One major application of ATT&CK is the use of its tactic and technique hierarchy by security specialists as a framework for annotating cyber-threat intelligence reports, vulnerability descriptions, threat scenarios, inter alia, to facilitate downstream analyses. To date, the tagging process is still largely done manually. In this technical note, we provide a stratified "task space" characterization of the MITRE ATT&CK text tagging task for organizing previous efforts toward automation using AIML methods, while also clarifying pathways for constructing new methods. To illustrate one of the pathways, we use the task space strata to stage-wise construct our own multi-label hierarchical classification models for the text tagging task via experimentation over general cyber-threat intelligence text -- using shareable computational tools and publicly releasing the models to the security community (via https://github.com/jpmorganchase/MITRE_models). Our multi-label hierarchical approach yields accuracy scores of roughly 94% at the tactic level, as well as accuracy scores of roughly 82% at the technique level. The models also meet or surpass state-of-the-art performance while relying only on classical machine learning methods -- removing any dependence on LLMs, RAG, agents, or more complex hierarchical approaches. Moreover, we show that GPT-4o model performance at the tactic level is significantly lower (roughly 60% accuracy) than our own approach. We also extend our baseline model to a corpus of threat scenarios for financial applications produced by subject matter experts.

威胁检测多标签分类安全建模

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