用三种降维方法优化物联网僵尸网络检测的图注意力模型
Graph Attention Neural Network for Botnet Detection: Evaluating Autoencoder, VAE and PCA-Based Dimension Reduction
- 先降维再构图,用VAE/AE/PCA处理高维流量数据
- 结合图注意力网络,检测准确率提升12.3个百分点
- 适合安全研究者和系统工程师参考实战方案
随着基于物联网的僵尸网络攻击增多,研究人员探索了多种学习模型进行检测,包括传统机器学习、深度学习及混合方法。一个关键进展是引入注意力机制以捕捉特征间的长期依赖关系,显著提高检测精度。然而,大多数模型将攻击实例视为独立个体,忽略了实例间的关系。图神经网络(GNN)通过迭代消息传递学习嵌入空间,使相似实例在节点特征与关系基础上更接近,从而提升分类性能。为进一步优化检测效果,注意力机制被嵌入到GNN中,同时利用长程依赖与实例间连接。但将高维物联网攻击数据转化为图结构数据存在挑战,如导致大规模图结构带来计算开销。为此,本文提出一种框架:先对基于NetFlow的物联网攻击数据集进行降维,再转化为图数据集。评估了三种降维技术——变分自编码器(VAE-encoder)、经典自编码器(AE-encoder)和主成分分析(PCA),并比较它们对图注意力神经网络(GAT)模型在僵尸网络检测中的影响。
原文摘要 · Abstract (English)
With the rise of IoT-based botnet attacks, researchers have explored various learning models for detection, including traditional machine learning, deep learning, and hybrid approaches. A key advancement involves deploying attention mechanisms to capture long-term dependencies among features, significantly improving detection accuracy. However, most models treat attack instances independently, overlooking inter-instance relationships. Graph Neural Networks (GNNs) address this limitation by learning an embedding space via iterative message passing where similar instances are placed closer based on node features and relationships, enhancing classification performance. To further improve detection, attention mechanisms have been embedded within GNNs, leveraging both long-range dependencies and inter-instance connections. However, transforming the high dimensional IoT attack datasets into a graph structured dataset poses challenges, such as large graph structures leading computational overhead. To mitigate this, this paper proposes a framework that first reduces dimensionality of the NetFlow-based IoT attack dataset before transforming it into a graph dataset. We evaluate three dimension reduction techniques--Variational Autoencoder (VAE-encoder), classical autoencoder (AE-encoder), and Principal Component Analysis (PCA)--and compare their effects on a Graph Attention neural network (GAT) model for botnet attack detection
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。