.png){kind=icon}
Return
Self-Supervised Learning Malware Traffic Classification Based on Masked Autoencoder
.png){kind=icon}
PRE
AI
DOI：10.1109/JIOT.2024.3357072.png)
.png){kind=icon}
.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Malware traffic classification (MTC) is one of the important techniques to ensure the security of cyberspace, which aims to detect anomalies and classify different types of network traffic. Recently, MTC methods based on deep learning (DL) have shown their excellent performance. However, these DL-based methods rely on data sets with manually labeled samples for training, which are costly and hard to obtain. To address this problem, this article proposes a novel self-supervised MTC method based on the framework of masked auto-encoder (MAE). Specifically, MAE first constructs a reasonable unsupervised pretext task with a random masking strategy, which reduces the redundant information in samples and speeds up the pretraining process. The transformer-based backbone network then efficiently extracts features from the nonredundant traffic data efficiently. The proposed MTC-MAE method employs self-supervised learning on a large-scale unlabeled data set to acquire unbiased features, and fine-tunes on specific data sets to adapt to diverse traffic classification scenarios. Simulation experiments show that our proposed MTC-MAE method is able to learn universal features with high quality and has excellent classification performance on various downstream data sets. The data sets we used, code implementation, and pretrained models are available on GitHub. Code available at https://github.com/TsuiHark/Self-supervised_MTC.
Keywords:
Task analysis
Feature extraction
Malware
Internet of Things
Training
Self-supervised learning
Cryptography
Malware traffic classification (MTC)
masked autoencoder
self-supervised learning (SSL)
transformer
Journal
IF:
8.9
Papers:
1.4W
Citations:
7.8W
Organization
Cited Papers
ShareSave
ShareSaveShareSaveShareSaveAn Ensemble Deep Learning-Based Cyber-Attack Detection in Industrial Control System.png)
IEEE ACCESS
OAAI
IEEE ACCESSIF3.6
OAAIShareSave
View more