Skip to Main Content
 

Global Search Box

 
 
 

ETD Abstract Container

Abstract Header

Few-Shot Malware Detection Using A Novel Adversarial Reprogramming Model

Kumar, Ekula Praveen

Abstract Details

2022, Master of Science in Cyber Security (M.S.C.S.), Wright State University, Computer Science.
The increasing sophistication of malware has made detecting and defending against new strains a major challenge for cybersecurity. One promising approach to this problem is using machine learning techniques that extract representative features and train classification models to detect malware in an early stage. However, training such machine learning-based malware detection models represents a significant challenge that requires a large number of high-quality labeled data samples while it is very costly to obtain them in real-world scenarios. In other words, training machine learning models for malware detection requires the capability to learn from only a few labeled examples. To address this challenge, in this thesis, we propose a novel adversarial reprogramming model for few-shot malware detection. Our model is based on the idea to re-purpose high-performance ImageNet classification model to perform malware detection using the features of malicious and benign files. We first embed the features of software files and a small perturbation to a host image chosen randomly from ImageNet, and then create an image dataset to train and test the model; after that, the model transforms the output into malware and benign classes. We evaluate the effectiveness of our model on a dataset of real-world malware and show that it significantly outperforms baseline few-shot learning methods. Additionally, we evaluate the impact of different pre-trained models, different data sizes, and different parameter values. Overall, our results suggest that the proposed adversarial reprogramming model is a promising direction for improving few-shot malware detection.
Lingwei Chen, Ph.D. (Advisor)
Tanvi Banerjee, Ph.D. (Committee Member)
Junjie Zhang, Ph.D. (Committee Member)
57 p.

Recommended Citations

Citations

  • Kumar, E. P. (2022). Few-Shot Malware Detection Using A Novel Adversarial Reprogramming Model [Master's thesis, Wright State University]. OhioLINK Electronic Theses and Dissertations Center. http://rave.ohiolink.edu/etdc/view?acc_num=wright1673288119841493

    APA Style (7th edition)

  • Kumar, Ekula Praveen. Few-Shot Malware Detection Using A Novel Adversarial Reprogramming Model. 2022. Wright State University, Master's thesis. OhioLINK Electronic Theses and Dissertations Center, http://rave.ohiolink.edu/etdc/view?acc_num=wright1673288119841493.

    MLA Style (8th edition)

  • Kumar, Ekula Praveen. "Few-Shot Malware Detection Using A Novel Adversarial Reprogramming Model." Master's thesis, Wright State University, 2022. http://rave.ohiolink.edu/etdc/view?acc_num=wright1673288119841493

    Chicago Manual of Style (17th edition)