arrow
Return

Trace2Vec: Detecting complex multi-step attacks with explainable graph neural network

delete2025-06-01
delete0
PRE
AI
刘玮 cover
刘玮 (Wei Liu)
Z
Zhang, Haotian
L
Li, Ke
Y
Yang, Weiyong
W
Wei, Xingshen
S
Shu, Jiwu
DOI:10.1016/j.patcog.2025.111363delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Complex multi-step attacks have caused significant damage to numerous critical infrastructures. Recognizing the pattern of such attacks is a challenging task, and graph neural network based methods have shown promising results via modeling the system's events as a graph. However, existing methods still face several limitations when deployed in practice. First, there is a lack of sufficient real attack data especially considering the large volume of normal data. Second, attack patterns are complex and evolve over time due to the dynamic and heterogeneous nature of events. Third, the lack of explanation in learned models undermines the trustworthiness of such methods in production environments. To address the above challenges, in this paper, we propose an attack detection method, TRACE2VEC. The approach first designs an corruption function to augment rare attack samples, and integrates them into the event graphs. Next, it models the event graphs via a continuous-time dynamic heterogeneous graph neural network. Finally, it employs the Monte Carlo tree search algorithm to identify events with greater contributions to the attack, thus enhancing the explainability of the detection result. We have implemented a prototype for TRACE2VEC, and the experimental evaluations demonstrate its superior detection and explanation performance compared to existing methods.
Keywords:
Complex multi-step attack
Anomaly detection
Dynamic heterogeneous graph
Attack sample construction
Explainability

Journal

Pattern Recognition cover
Pattern Recognition
IF:
7.6
Papers:
1.3W
Citations:
4.5W

Organization

No organization information available
Cited Papers

Cited Papers

Quantum-based subgraph convolutional neural networks
err2019-04-01
err45
errOAAI
errZhang, Zhihong; Chen, Dongdong; Wang, Jianjia; Bai, Lu; Hancock, Edwin R.
errShare
errSave
errShare
errSave
MF-Net: Multi-frequency intrusion detection network for Internet traffic data
err2024-02-01
err7
PREAI
errDing, Zhaoxu; Zhong, Guoqiang; Qin, Xianping; Li, Qingyang; Fan, Zhenlin; Deng, Zhaoyang; Ling, Xiao; Xiang, Wei
errShare
errSave
Depth-based subgraph convolutional auto-encoder for network representation learning
err2019-06-01
err25
errOAAI
errZhang, Zhihong; Chen, Dongdong; Wang, Zeli; Li, Heng; Bai, Lu; Hancock, Edwin R.
errShare
errSave
Graph Motif Entropy for Understanding Time-Evolving Networks
err2023-04-01
err11
errOAAI
errZhang, Zhihong; Chen, Dongdong; Bai, Lu; Wang, Jianjia; Hancock, Edwin R.
errShare
errSave
Log2vec
err2019-11-06
err0
PREAI
errFucheng Liu; Yu Wen; Dongxue Zhang; Xihe Jiang; Xinyu Xing; Dan Meng
errShare
errSave
POIROT
err2019-11-06
err0
errOAAI
errSadegh M. Milajerdi; Birhanu Eshete; Rigel Gjomemo; V.N. Venkatakrishnan
errShare
errSave
Detecting Unknown Threat Based on Continuous-Time Dynamic Heterogeneous Graph Network
err2022-07-25
err0
errOAAI
errPeng Gao; Weiyong Yang; Haotian Zhang; Xingshen Wei; Hao Huang; Wang Luo; Zhimin Guo; Yunhe Hao
errShare
errSave
errShare
errSave
APTSHIELD: A Stable, Efficient and Real-Time APT Detection System for Linux Hosts
err2023-11-01
err14
errOAAI
errZhu, Tiantian; Yu, Jinkai; Xiong, Chunlin; Cheng, Wenrui; Yuan, Qixuan; Ying, Jie; Chen, Tieming; Zhang, Jiabo; Lv, Mingqi; Chen, Yan; Wang, Ting; Fan, Yuan
errShare
errSave
researcher View more