Multi-label Classification for Android Malware Based on Active Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qiao, Qijing, Feng, Ruitao, Chen, Sen, Zhang, Fei, Li, Xiaohong
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929533589913600
author Qiao, Qijing
Feng, Ruitao
Chen, Sen
Zhang, Fei
Li, Xiaohong
author_facet Qiao, Qijing
Feng, Ruitao
Chen, Sen
Zhang, Fei
Li, Xiaohong
contents The existing malware classification approaches (i.e., binary and family classification) can barely benefit subsequent analysis with their outputs. Even the family classification approaches suffer from lacking a formal naming standard and an incomplete definition of malicious behaviors. More importantly, the existing approaches are powerless for one malware with multiple malicious behaviors, while this is a very common phenomenon for Android malware in the wild. So, neither of them can provide researchers with a direct and comprehensive enough understanding of malware. In this paper, we propose MLCDroid, an ML-based multi-label classification approach that can directly indicate the existence of pre-defined malicious behaviors. With an in-depth analysis, we summarize six basic malicious behaviors from real-world malware with security reports and construct a labeled dataset. We compare the results of 70 algorithm combinations to evaluate the effectiveness (best at 73.3%). Faced with the challenge of the expensive cost of data annotation, we further propose an active learning approach based on data augmentation, which can improve the overall accuracy to 86.7% with a data augmentation of 5,000+ high-quality samples from an unlabeled malware dataset. This is the first multi-label Android malware classification approach intending to provide more information on fine-grained malicious behaviors.
format Preprint
id arxiv_https___arxiv_org_abs_2410_06444
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Multi-label Classification for Android Malware Based on Active Learning
Qiao, Qijing
Feng, Ruitao
Chen, Sen
Zhang, Fei
Li, Xiaohong
Cryptography and Security
The existing malware classification approaches (i.e., binary and family classification) can barely benefit subsequent analysis with their outputs. Even the family classification approaches suffer from lacking a formal naming standard and an incomplete definition of malicious behaviors. More importantly, the existing approaches are powerless for one malware with multiple malicious behaviors, while this is a very common phenomenon for Android malware in the wild. So, neither of them can provide researchers with a direct and comprehensive enough understanding of malware. In this paper, we propose MLCDroid, an ML-based multi-label classification approach that can directly indicate the existence of pre-defined malicious behaviors. With an in-depth analysis, we summarize six basic malicious behaviors from real-world malware with security reports and construct a labeled dataset. We compare the results of 70 algorithm combinations to evaluate the effectiveness (best at 73.3%). Faced with the challenge of the expensive cost of data annotation, we further propose an active learning approach based on data augmentation, which can improve the overall accuracy to 86.7% with a data augmentation of 5,000+ high-quality samples from an unlabeled malware dataset. This is the first multi-label Android malware classification approach intending to provide more information on fine-grained malicious behaviors.
title Multi-label Classification for Android Malware Based on Active Learning
topic Cryptography and Security
url https://arxiv.org/abs/2410.06444