.png){kind=icon}
返回
MalSSL-Self-Supervised Learning for Accurate and Label-Efficient Malware Classification
.png){kind=icon}
OA
AI
DOI：10.1109/ACCESS.2024.3392251.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Malware classification with supervised learning requires a large dataset, which needs an expensive and time-consuming labeling process. In this paper, we explore the efficacy of self-supervised learning techniques for malware classification. We propose MalSSL, a self-supervised learning-based method utilizing image representation to classify malware. MalSSL classifies unlabeled malware images using contrastive learning and data augmentation. The model is initially trained on an unlabeled Imagenette dataset as a pretext task and subsequently retrained on an unlabeled malware dataset in downstream tasks. Two downstream tasks were employed to evaluate the system: 1) malware family classification and 2) malware benign classification. The obtained results include an accuracy of 98.4% for the malware family classification experiment on the Malimg dataset and an accuracy of 96.2% for the malware and benign dataset (Maldeb dataset). Our findings suggest that the proposed system accurately classifies malware without the need for labeled data, displaying higher accuracy compared to other self-supervised methods. This research not only contributes to advancing the state-of-the-art in malware classification but also underscores the potential of self-supervised learning methods as a viable solution for addressing the dynamic landscape of malware threats.
Keyword:
Image representation
Training
Support vector machines
Labeling
malware
malware classification
self-supervised learning
期刊
IF:
3.6
论文数:
9.8W
被引数:
29.4W
机构
引用论文
Using 3D-VGG-16 and 3D-Resnet-18 deep learning models and FABEMD techniques in the detection of malware使用3D-VGG-16和3D-Resnet-18深度学习模型和FABEMD技术检测恶意软件
OAAI
OAAI分享收藏分享收藏Applications of Self-Supervised Learning to Biomedical Signals: A Survey自监督学习在生物医学信号中的应用: 一项调查.png)
IEEE ACCESSOAAI
IEEE ACCESSIF3.6
OAAI分享收藏分享收藏
查看更多内容