Network Intrusion Detection via Flow-to-Image Conversion and Vision Transformer Classification

Bibliographic Details
Title: Network Intrusion Detection via Flow-to-Image Conversion and Vision Transformer Classification
Authors: Chi Mai Kim Ho, Kin-Choong Yow, Zhongwen Zhu, Sarang Aravamuthan
Source: IEEE Access, Vol 10, Pp 97780-97793 (2022)
Publisher Information: IEEE, 2022.
Publication Year: 2022
Collection: LCC:Electrical engineering. Electronics. Nuclear engineering
Subject Terms: Network intrusion detection, flow-to-image conversion, convolutional neural networks, vision transformers, image classification, Electrical engineering. Electronics. Nuclear engineering, TK1-9971
More Details: In recent years, computer networks have become an indispensable part of our life, and these networks are vulnerable to various type of network attacks, compromising the security of our data and the freedom of our communications. In this paper, we propose a new intrusion detection method that uses image conversion from network data flow to produce an RGB image that can be classified using advanced deep learning models. In this method, we proposed to use the decision tree algorithm to identify the important features, and a windowing and overlapping mechanism to convert the varying input size to a standard size image for the classifier. We then use a Vision Transfomer (ViT) classifier to classify the resulting image. Our experimental results show that we can achieve 98.5% accuracy in binary classification on the CIC IDS2017 dataset, and 96.3% on the UNSW-NB15 dataset, which is 8.09% higher than the next best algorithm, the Deep Belief Network with Improved Kernel-Based Extreme Learning (DBN-KELM) method. For multi-class classification, our proposed method can achieve a testing accuracy of 96.4%, which is 5.6% higher than the next best method, the DBN-KELM.
Document Type: article
File Description: electronic resource
Language: English
ISSN: 2169-3536
Relation: https://ieeexplore.ieee.org/document/9862964/; https://doaj.org/toc/2169-3536
DOI: 10.1109/ACCESS.2022.3200034
Access URL: https://doaj.org/article/195bcb3ea92b4a909d37cb7be3575d72
Accession Number: edsdoj.195bcb3ea92b4a909d37cb7be3575d72
Database: Directory of Open Access Journals
More Details
ISSN:21693536
DOI:10.1109/ACCESS.2022.3200034
Published in:IEEE Access
Language:English