Federated learning for network attack detection using attention-based graph neural networks.

Sci Rep

Key Laboratory of Key Technologies for Open Data Fusion in Zhejiang Province, Hangzhou, 310007, China.

Published: August 2024

Federated Learning is an effective solution to address the issues of data isolation and privacy leakage in machine learning. However, ensuring the security of network devices and architectures deploying federated learning remains a challenge due to network attacks. This paper proposes an attention-based Graph Neural Network for detecting cross-level and cross-department network attacks. This method enables collaborative model training while protecting data privacy on distributed devices. By organizing network traffic information in chronological order and constructing a graph structure based on log density, enhances the accuracy of network attack detection. The introduction of an attention mechanism and the construction of a Federated Graph Attention Network (FedGAT) model are used to evaluate the interactivity between nodes in the graph, thereby improving the precision of internal network interactions. Experimental results demonstrate that our method achieves comparable accuracy and robustness to traditional detection methods while prioritizing privacy protection and data security.

Download full-text PDF

Source
http://www.ncbi.nlm.nih.gov/pmc/articles/PMC11330492PMC
http://dx.doi.org/10.1038/s41598-024-70032-2DOI Listing

Publication Analysis

Top Keywords

federated learning
12
network
9
network attack
8
attack detection
8
attention-based graph
8
graph neural
8
network attacks
8
graph
5
federated
4
learning network
4

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!