Event Details
Harnessing Image-Based Deep Learning for Advanced Malware Classification
Presenter: Ahmed A. Abouelkhaire
Supervisor:
Date: Fri, August 23, 2024
Time: 10:00:00 - 00:00:00
Place: Zoom, link below.
ABSTRACT
Join Zoom Meeting:
Meeting ID: 818 9739 4054
Password: 078447
One tap mobile
+16475580588,,81897394054# Canada
+17789072071,,81897394054# Canada
Dial by your location
+1 647 558 0588 Canada
+1 778 907 2071 Canada
Meeting ID: 818 9739 4054
Abstract:
This seminar explores the application of image-based deep learning models for malware classification, leveraging a subset of the extensive MalNet-Image dataset, which includes around 87,000 binary images from a base of 1.2 million binary images based on Android APK files. The core contribution of this work lies in the innovative use of multiple components that, as far as we know, have not been used before to tackle the malware classification problem. Harnessing the power of deep neural networks (DNNs), which have demonstrated exceptional capabilities in various classification tasks, we aim to enhance the accuracy and efficiency of malware detection. These include Feature Pyramid Networks (FPN) to handle the file size scale issue when converting to images and the application of data augmentation techniques like Mix-up and Trivial Augment. We employ transfer learning with pre-trained models on ImageNet and optimize them using the AdamW Schedule-Free optimizer. Our experimental results show that the integration of these techniques achieves remarkable improvement in classification accuracy, with our best model achieving an F1 score of 0.6927 compared to 0.65 reported on the provided split for MalNet-Tiny. This could be considered a step forward in the field of malware classification using image-based deep learning models.