arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.14746cs.CRcs.AIcs.CLcs.NEcs.NI
收藏

FlowTransformer:一个用于基于流的网络入侵检测系统的Transformer框架

FlowTransformer: A Transformer Framework for Flow-based Network Intrusion Detection Systems

University of Queensland(昆士兰大学)

Liam Daly Manocchio, Siamak Layeghy, Wai Weng Lo, Gayan K. Kulatilleke, Mohanad Sarhan, Marius Portmann

AI总结:

本文提出 FlowTransformer 框架,支持灵活替换输入编码、transformer 和分类头等组件,用于构建基于流的网络入侵检测系统,并在三个基准数据集上评估了 GPT 2.0 和 BERT 等架构,发现分类头选择对性能影响最大,优化组件可减小模型并提升速度。

摘要:

本文提出了 FlowTransformer 框架,这是一种用于实现基于 transformer 的网络入侵检测系统(NIDSs)的新方法。FlowTransformer 利用 transformer 模型在识别网络长期行为和特征方面的优势,而这些行为和特征通常被大多数现有的 NIDSs 所忽视。通过捕获网络流量中的这些复杂模式,FlowTransformer 为网络安全社区中寻求使用基于 transformer 的模型来实现 NIDSs 的研究人员和从业者提供了一个灵活且高效的工具。FlowTransformer 允许直接替换各种 transformer 组件,包括输入编码、transformer、分类头,以及在任何基于流的网络数据集上对这些组件进行评估。为了证明 FlowTransformer 框架的有效性和高效性,我们利用它在三个常用的公共 NIDS 基准数据集上对各种常见的 transformer 架构(如 GPT 2.0 和 BERT)进行了广泛的评估。我们提供了准确率、模型大小和速度的结果。我们评估的一个关键发现是,分类头的选择对模型性能的影响最为显著。令人惊讶的是,在文本分类中常用的全局平均池化(Global Average Pooling)在 NIDS 环境中表现非常差。此外,我们表明,通过选择特定的输入编码和分类头,而不是其他常用的替代方案,可以将模型大小减小 50% 以上,并改善推理和训练时间,且不会损失准确率。

英文摘要:

This paper presents the FlowTransformer framework, a novel approach for implementing transformer-based Network Intrusion Detection Systems (NIDSs). FlowTransformer leverages the strengths of transformer models in identifying the long-term behaviour and characteristics of networks, which are often overlooked by most existing NIDSs. By capturing these complex patterns in network traffic, FlowTransformer offers a flexible and efficient tool for researchers and practitioners in the cybersecurity community who are seeking to implement NIDSs using transformer-based models. FlowTransformer allows the direct substitution of various transformer components, including the input encoding, transformer, classification head, and the evaluation of these across any flow-based network dataset. To demonstrate the effectiveness and efficiency of the FlowTransformer framework, we utilise it to provide an extensive evaluation of various common transformer architectures, such as GPT 2.0 and BERT, on three commonly used public NIDS benchmark datasets. We provide results for accuracy, model size and speed. A key finding of our evaluation is that the choice of classification head has the most significant impact on the model performance. Surprisingly, Global Average Pooling, which is commonly used in text classification, performs very poorly in the context of NIDS. In addition, we show that model size can be reduced by over 50\%, and inference and training times improved, with no loss of accuracy, by making specific choices of input encoding and classification head instead of other commonly used alternatives.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑