arrow
Return

Network attack knowledge inference with graph convolutional networks and convolutional 2D KG embeddings

delete2025-10-03
delete0
delete
OA
AI
W
Weiwu Ren
H
Hewen Zhang
Y
Ying Lei *
DOI:10.1038/s41598-025-17941-ydelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
To address the challenge of analyzing large-scale penetration attacks under complex multi-relational and multi-hop paths, this paper proposes a graph convolutional neural network-based attack knowledge inference method, KGConvE, aimed at intelligent reasoning and effective association mining of implicit network attack knowledge. The core idea of this method is to obtain knowledge embeddings related to CVE, CWE, and CAPEC, which are then used to construct attack context feature data and a relation matrix. Subsequently, we employ a graph convolutional neural network model to classify the attacks, and use the KGConvE model to perform attack inference within the same attack category. Through improvements to the graph convolutional neural network model, we significantly enhance the accuracy and generalization capability of the attack classification task. Furthermore, we are the first to apply the KGConvE model to perform attack inference tasks. Experimental results show that this method can infer implicit relationships between CVE-CVE, CVE-CWE, and CVE-CAPEC, achieving a significant performance improvement in network attack knowledge inference tasks, with a mean reciprocal rank (MRR) of 0.68 and Hits@10 of 0.58, outperforming baseline methods.
Keywords:
Graph convolutional neural networks
Attack inference
Knowledge graph completion
Security knowledge graph

Journal

Scientific Reports cover
Scientific Reports
IF:
3.9
Papers:
27.8W
Citations:
83.5W

Organization

S
School of Computer Science and Technology
Scholars:
1.5K
Papers: 566
Citations: 0