LLM2D
物联网僵尸网络检测:视觉变换器在网络流 traffi 分类中的应用
IoT Botnet Detection: Application of Vision Transformer to Classification of Network Flow Traffic
作者: Hassan Wasswa, Timothy Lynar, Aziida Nanyonga, Hussein Abbass
发布日期: 4/29/2025
arXiv ID: oai:arXiv.org:2504.18781v1

摘要

arXiv:2504.18781v1 计算机视觉:交叉 摘要:尽管 transformer 模型在自然语言处理、图像分类和视频分类方面已被证明是有效的,但是现有的用于从捕获的物联网网络流数据包中提取特征的工具未能捕捉到序贯模式,从而限制了 transformer 模型的应用。本工作提出了一种新的预处理方法,以适应 transformer 模型,尤其是用于使用网络流数据包检测物联网恶意软件攻击的视觉 transformer (ViT)。该方法涉及从 .pcap 文件中提取特征,并将每个实例转换为单通道二维图像形状,从而使得基于 ViT 的分类成为可能。此外,ViT 模型得以增强,以便除了初始 ViT 论文中的多层感知机 (MLP) 之外,还可以使用任何分类器。包括传统前馈深度神经网络 (DNN)、长短期记忆 (LSTM) 和双向长期短期记忆 (BLSTM) 的模型,在两个物联网攻击数据集上进行评估时,在多分类攻击检测方面展示了竞争力的性能,具体表现在精确率、召回率和 F1 分数方面。