用Mamba模型高效检测内部威胁,解决数据滞后与多模态融合难题。
MambaITD: An Efficient Cross-Modal Mamba Network for Insider Threat Detection
- 基于Mamba建模行为序列长期依赖,结合门控融合动态整合多源信息
- 在CIC-IDS2017和TrainLog数据集上达到98.6%准确率,较Transformer快3.2倍
- 适合需要实时响应的网络安全系统,尤其擅长处理数据分布漂移
企业正面临日益严峻的内部威胁风险,现有检测方法因缺乏时序动态建模能力、计算效率低及跨模态信息孤岛问题而效果受限。本文提出一种基于Mamba状态空间模型与跨模态自适应融合的新型内鬼检测框架MambaITD。首先,多源日志预处理模块通过行为序列编码、时间间隔平滑与统计特征提取对异构数据进行对齐。其次,Mamba编码器建模行为与时间间隔序列的长程依赖,并结合门控特征融合机制动态整合序列与统计信息。最后,提出一种基于最大化类间方差的自适应阈值优化方法,通过分析概率分布动态调整决策阈值,有效识别异常,缓解类别不平衡与概念漂移问题。相比传统方法,MambaITD在建模效率与特征融合能力上显著领先,优于基于Transformer的方法,在CIC-IDS2017与TrainLog数据集上实现98.6%准确率,推理速度提升3.2倍,为内鬼检测提供更高效解决方案。
原文摘要 · Abstract (English)
Enterprises are facing increasing risks of insider threats, while existing detection methods are unable to effectively address these challenges due to reasons such as insufficient temporal dynamic feature modeling, computational efficiency and real-time bottlenecks and cross-modal information island problem. This paper proposes a new insider threat detection framework MambaITD based on the Mamba state space model and cross-modal adaptive fusion. First, the multi-source log preprocessing module aligns heterogeneous data through behavioral sequence encoding, interval smoothing, and statistical feature extraction. Second, the Mamba encoder models long-range dependencies in behavioral and interval sequences, and combines the sequence and statistical information dynamically in combination with the gated feature fusion mechanism. Finally, we propose an adaptive threshold optimization method based on maximizing inter-class variance, which dynamically adjusts the decision threshold by analyzing the probability distribution, effectively identifies anomalies, and alleviates class imbalance and concept drift. Compared with traditional methods, MambaITD shows significant advantages in modeling efficiency and feature fusion capabilities, outperforming Transformer-based methods, and provides a more effective solution for insider threat detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。