arXiv:2412.04367cs.LGcs.AI2024-12被引 2

用思维理论模型预测黑客攻击目标和路径,让防御系统看得懂对手意图。

Machine Theory of Mind for Autonomous Cyber-Defence

  • 基于图神经网络构建可解释的攻击者行为预测模型
  • 在多种网络拓扑下准确预测未知攻击者的攻击目标与路径
  • 提出新度量标准NTD,支持不同规模网络的统一比较

智能自主代理在网络安全领域潜力巨大,但现有方法多依赖不可解释的黑箱模型,难以向利益相关方提供清晰可操作的洞察。为此,本文评估了思维理论(ToM)在自主网络防御中的应用。通过学习稳健先验,ToM模型仅需少量历史行为观测即可预测攻击者的目标、行为及上下文信念。本文提出一种新型图神经网络架构GIGO-ToM,专用于预测任意网络拓扑下的攻击目标与攻击路径。为评估该能力,我们引入一种改进的Wasserstein距离——网络传输距离(NTD),其通过图论归一化因子实现不同规模网络间的标准化比较,并设计加权函数以突出特定节点特征,使防御者可依据战略需求灵活调整关注重点。在抽象网络安全环境中,对比基准模型GIDO-ToM,实验表明GIGO-ToM能准确预测多种未见攻击者的目标与行为,并学习到有效表征其策略的嵌入表示。

原文摘要 · Abstract (English)

Intelligent autonomous agents hold much potential for the domain of cyber-security. However, due to many state-of-the-art approaches relying on uninterpretable black-box models, there is growing demand for methods that offer stakeholders clear and actionable insights into their latent beliefs and motivations. To address this, we evaluate Theory of Mind (ToM) approaches for Autonomous Cyber Operations. Upon learning a robust prior, ToM models can predict an agent's goals, behaviours, and contextual beliefs given only a handful of past behaviour observations. In this paper, we introduce a novel Graph Neural Network (GNN)-based ToM architecture tailored for cyber-defence, Graph-In, Graph-Out (GIGO)-ToM, which can accurately predict both the targets and attack trajectories of adversarial cyber agents over arbitrary computer network topologies. To evaluate the latter, we propose a novel extension of the Wasserstein distance for measuring the similarity of graph-based probability distributions. Whereas the standard Wasserstein distance lacks a fixed reference scale, we introduce a graph-theoretic normalization factor that enables a standardized comparison between networks of different sizes. We furnish this metric, which we term the Network Transport Distance (NTD), with a weighting function that emphasizes predictions according to custom node features, allowing network operators to explore arbitrary strategic considerations. Benchmarked against a Graph-In, Dense-Out (GIDO)-ToM architecture in an abstract cyber-defence environment, our empirical evaluations show that GIGO-ToM can accurately predict the goals and behaviours of various unseen cyber-attacking agents across a range of network topologies, as well as learn embeddings that can effectively characterize their policies.

思维理论网络安全图神经网络可解释性

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