Suppr超能文献

基于注意力机制的图神经网络在网络攻击检测中的联邦学习

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

作者信息

Jianping Wu, Guangqiu Qiu, Chunming Wu, Weiwei Jiang, Jiahe Jin

机构信息

College of Computer Science and Technology, Zhejiang University, Hangzhou, 310027, China.

Smart Government R &D Center (Laboratory) of Hangzhou Dianzi University, Hangzhou, 310018, China.

出版信息

Sci Rep. 2024 Aug 17;14(1):19088. doi: 10.1038/s41598-024-70032-2.

Abstract

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.

摘要

联邦学习是解决机器学习中数据隔离和隐私泄露问题的有效解决方案。然而,由于网络攻击,确保部署联邦学习的网络设备和架构的安全性仍然是一个挑战。本文提出了一种基于注意力的图神经网络,用于检测跨级别和跨部门的网络攻击。该方法在保护分布式设备上的数据隐私的同时,实现了协作模型训练。通过按时间顺序组织网络流量信息,并基于日志密度构建图结构,提高了网络攻击检测的准确性。引入注意力机制并构建联邦图注意力网络(FedGAT)模型,用于评估图中节点之间的交互性,从而提高内部网络交互的精度。实验结果表明,我们的方法在优先考虑隐私保护和数据安全的同时,与传统检测方法具有相当的准确性和鲁棒性。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/b9f5/11330492/4a9a14f90b73/41598_2024_70032_Fig1_HTML.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验