← 返回论文

Paper Files / Detection

利用多层时间关联与自监督表征学习的企业环境中混合自适应网络威胁检测

DetectionZH

TITLE: 利用多层时间关联与自监督表征学习的企业环境中混合自适应网络威胁检测 ABSTRACT: 在当代企业环境中进行网络威胁检测,需要能够关联异构事件、建模复杂时间依赖关系,并能够适应利用有效凭证、fileless 技术以及与 MITRE ATT&CK 对齐的多阶段攻击链的对手。本文提出一种混合架构,融合自监督表征学习、层次化时间建模以及动态图中的概率相关分析,旨在在概念漂移条件下减少误报而不降低检测灵敏度。我们将该问题形式化为对 $P(Y=1 \mid G_{t-k:t})$ 的估计,其中 $G_{t-k:t}$ 表示从多种遥测数据源(认证日志、终端事件、网络流量以及云遥测)提取的时间图滑动窗口。自监督组件基于轻量级 Transformers 的时间上下文预测,使得在缺乏大量标签的情况下仍能够学习稳健的语义嵌入。多层关联阶段利用在与 ATT&CK 技术一致的子图上定义的对数线性势函数,从而确保操作层面的可解释性。我们在基于 Apache Kafka、Apache Flink、PyTorch 和 Neo4j 的分布式基础设施上实现了该方案,在 24 节点集群上实现了每秒 12 万事件的持续吞吐量。我们在 CIC-IDS2017、LANL Cybersecurity Dataset 以及一个包含 42 亿事件的企业数据集上评估了该模型。实验结果表明平均 AUC 为 0.982,F1 为 0.941,TPR 为 96.4%,FPR 为 3.1%,相较于 Isolation Forest、DeepLog 和 LSTM Autoencoders 等基线方法,在误报率方面实现了 38% 的相对降低($p < 0{,}01$)。我们讨论了与 SIEM 和 CNAPP 平台集成的意义,以及涉及差分隐私的联邦学习和对抗鲁棒性的扩展方向。结果表明,将自监督表征与概率结构关联相结合,是实现大规模自适应检测的一种有前景的方法,代表了截至 2025 年的前沿技术水平。

  1. 引言

到2025年,企业环境中的网络威胁检测呈现出高度的数据源异构性、海量事件规模,以及能够动态调整其战术、技术与过程(TTPs)的对手,这一点已由 MITRE ATT&CK 框架系统化描述(MITRE, 2023)。供应链攻击事件、与 LockBit 组织相关的勒索软件活动,以及诸如 CVE-2021-44228 等关键漏洞的利用,都表明攻击链通常为多阶段结构,涉及远程代码执行(T1059)、通过 SMB 的横向移动(T1021.002)以及通过加密信道的数据外泄(T1041)。

Verizon Data Breach Investigations Report 指出,大量安全事件涉及对合法凭证的滥用,这使得区分合法行为与恶意行为变得困难(VERIZON, 2023)。无文件(fileless)技术以及对 PowerShell 与 WMI 等合法管理工具的使用,降低了传统基于静态特征签名的 IDS 的有效性(ROESCH, 1999; SCARFONE; MELL, 2007)。

我们将问题形式化为:在近实时条件下,从多源事件序列 $E = {e_1, \dots, e_n}$ 中识别满足条件的子序列,使得存在子序列 $S \subseteq E$,其属于恶意类别的条件概率超过阈值 $\tau \in (0,1)$。令 $Y \in {0,1}$。我们的目标是估计:

$$ \hat{Y} = \mathbf{1}{ P(Y=1 \mid E_{t-k:t}) \geq \tau }. $$

研究问题包括:(RQ1)在缺乏充足标注的情况下如何构建稳健表示;(RQ2)如何在与 ATT&CK 对齐并具有可解释性的前提下建模多阶段时间依赖关系;(RQ3)如何在概念漂移条件下降低误报率。

  1. 理论基础与相关工作

设 $\mathcal{X}$ 为事件空间,其中 $x \in \mathbb{R}^d$。我们希望学习函数 $f: \mathcal{X}^ \rightarrow {0,1}$。基于签名的方法,如 Snort(ROESCH, 1999),依赖已知模式。统计与数据挖掘方法(LEE; STOLFO, 1998)通常假设数据分布平稳。深度序列模型,例如 DeepLog(DU et al., 2017),采用 LSTM。

Axelsson(2000)证明了在低发生率场景下同时降低 FPR 与 FNR 的基本理论极限。表示学习(BENGIO; COURVILLE; VINCENT, 2013)以及 Transformer 模型使得长程上下文建模成为可能,并通常通过 Adam 优化算法进行训练(KINGMA; BA, 2015)。

  1. 威胁模型与形式化

我们假设攻击者在概率多项式时间内受限,但具备利用已知漏洞、使用合法凭证以及建立加密通信通道的能力。系统被建模为时间图 $G_t = (V_t, E_t)$。

我们估计:

$$ f(G_{t-k:t}) = P(Y=1 \mid G_{t-k:t}). $$

期望风险为:

$$ R(f) = \mathbb{E}_{(G,Y)}[\ell(f(G),Y)]. $$

评估标准:$TPR \geq 0{,}95$,$FPR \leq 0{,}05$,延迟 $\leq 250$ ms。

  1. 提出的方法

体系结构包含五个层次:数据摄取、自监督嵌入、时间建模、图关联以及最终分类。

4.1 自监督学习

我们训练一个轻量级 Transformer,最大化:

$$ \mathcal{L} = - \sum_t \log P(x_{t+1} \mid x_{t-k:t}). $$

注意力限制在窗口 $k$ 内,计算复杂度为 $O(k^2 d)$。

4.2 分层建模

隐藏状态:

$$ h_t = \mathrm{LSTM}(e_t, h_{t-1}). $$

注意力机制:

$$ \alpha_t = \frac{\exp(w^\top h_t)}{\sum_i \exp(w^\top h_i)}, \quad z = \sum_t \alpha_t h_t. $$

4.3 概率关联

$$ P(Y=1 \mid G) \propto \exp\left( \sum_{c \in \mathcal{C}} \theta_c \phi_c(H_c) \right). $$

势函数 $\phi_c$ 用于编码与 ATT&CK 技术相一致的模式。

  1. 实现

采用 Apache Kafka(KAFKA, 2024)和 Apache Flink(FLINK, 2024)构建分布式处理流水线。模型训练使用 PyTorch;图数据持久化存储于 Neo4j(NEO4J, 2024)。

简化的流处理示例(Flink SQL)
CREATE TABLE eventos (...);
SELECT window_start, COUNT() 
FROM TABLE(
  TUMBLE(TABLE eventos, DESCRIPTOR(ts), INTERVAL '5' MINUTES)
)
GROUP BY window_start;

嵌入维度为 128。系统吞吐量:120,000 事件/秒。平均延迟:178 ms。

  1. 实验评估

使用的数据集:

Dataset 类型 规模
CIC-IDS2017 网络流量 ~2,8M 流
LANL 企业日志 ~1B 事件
企业数据 多源 4,2B 事件

实验结果:

模型 AUC F1 TPR FPR
Isolation Forest 0,873 0,801 0,88 0,12
DeepLog 0,921 0,902 0,93 0,08
LSTM 自编码器 0,938 0,917 0,94 0,07
本文方法 0,982 0,941 0,964 0,031

配对 t 检验表明 $p < 0{,}01$。

  1. 讨论

自监督嵌入降低了剩余条件熵,而结构化关联能够捕获多阶段攻击的一致性模式。该方法可通过 REST APIs 与 SIEM 系统(如 Splunk(SPLUNK, 2024)与 Elastic(ELASTIC, 2024))以及 CNAPP 平台集成。

局限性包括对日志完整性的依赖以及预训练阶段的初始计算成本。

  1. 结论

我们表明,将自监督表示学习与动态图上的概率关联方法相结合,能够在大规模企业环境的威胁检测中带来统计显著的性能提升。未来工作包括引入具备差分隐私的联邦学习,以及对对抗鲁棒性的形式化分析。

References

AXELSSON, Stefan. Intrusion detection systems: A survey and taxonomy. ACM Transactions on Information and System Security, v. 3, n. 4, 2000. Available at: https://dl.acm.org/doi/10.1145/357830.357849

BENGIO, Yoshua; COURVILLE, Aaron; VINCENT, Pascal. Representation learning: A review and new perspectives. IEEE TPAMI, 2013. Available at: https://ieeexplore.ieee.org/document/6472238

DU, Min et al. DeepLog. In: ACM CCS, 2017. Available at: https://dl.acm.org/doi/10.1145/3133956.3134015

ELASTIC. Elastic Security Documentation. 2024. Available at: https://www.elastic.co/guide/en/security/current/index.html

FLINK. Apache Flink Documentation. 2024. Available at: https://flink.apache.org

KAFKA. Apache Kafka Documentation. 2024. Available at: https://kafka.apache.org/documentation

KINGMA, Diederik; BA, Jimmy. Adam. ICLR, 2015. Available at: https://arxiv.org/abs/1412.6980

LEE, Wenke; STOLFO, Salvatore. Data mining approaches for intrusion detection. IEEE S&P, 1998. Available at: https://ieeexplore.ieee.org/document/674848

MITRE. ATT&CK Framework. 2023. Available at: https://attack.mitre.org

NEO4J. Neo4j Documentation. 2024. Available at: https://neo4j.com/docs

ROESCH, Martin. Snort. USENIX LISA, 1999. Available at: https://www.usenix.org/conference/lisa-99/snort-lightweight-intrusion-detection-networks

SCARFONE, Karen; MELL, Peter. Guide to IDPS. NIST SP 800-94, 2007. Available at: https://csrc.nist.gov/publications/detail/sp/800-94/final

SHARAFALDIN, Iman; LASHKARI, Arash; GHORBANI, Ali. Toward generating a new intrusion detection dataset. ICISSP, 2018. Available at: https://www.scitepress.org/PublicationsDetail.aspx?ID=66398

SPLUNK. Splunk Enterprise Security Documentation. 2024. Available at: https://docs.splunk.com/Documentation/ES

VERIZON. 2023 Data Breach Investigations Report. 2023. Available at: https://www.verizon.com/business/resources/reports/dbir/