What are you looking for today?

Ajayi, Bamidele, Barakat, Basel, McGarry, Kenneth and Abukeshek, Mays (2024) Exploring the Application of Transfer Learning in Malware Detection by Fine-tuning Pre-Trained Models on Binary Classification to New Datasets on Multi-class Classification. In: 2024 29th International Conference on Automation and Computing (ICAC). IEEE, pp. 1-6. ISBN 979-8-3503-6088-2

Document

  • [thumbnail of Author Accepted Manuscript]
    Author Accepted Manuscript
    Exploring the application of transfer learning in malware detection.pdf
    Available under License Creative Commons Attribution.

Abstract

Abstract

his research presents a method for classifying malicious and benign binary files using Convolutional Neural Networks (CNNs), transitioning from binary to multiclass classification. Three commonly used datasets were tested: EMBER, BODMAS, and MALIMG, with EMBER and BODMAS serving as training and testing sets for the base model. Data from these datasets is converted into image representations and analyzed by CNN models, achieving a high accuracy of 98%. A transfer learning model is then developed, incorporating knowledge from EMBER and BODMAS. This model reduces training time significantly and achieves 97% accuracy with just 5 epochs and a batch size of 25 across 25 malware family sets, averaging a perfect AUC of 1.00. This indicates perfect discrimination between positive and negative classes, with 100% correct predictions, underscoring the robustness of the method.

Item Type: Book Section
Additional Information: Proceedings of the 2024 29th International Conference on Automation and Computing (ICAC) DOI: 10.1109/ICAC61394.2024 28-30 Aug. 2024
Related URLs:
SWORD Depositor: Publication Router
Depositing User: Publication Router
URI: https://sure.sunderland.ac.uk/id/eprint/18417

Further Details

More Information

Additional Information
Proceedings of the 2024 29th International Conference on Automation and Computing (ICAC) DOI: 10.1109/ICAC61394.2024 28-30 Aug. 2024
SWORD Depositor
Depositing User

Identifiers

Item ID
18417
Identification Number
ISBN
979-8-3503-6088-2

Users with ORCIDS

ORCID for Basel Barakat
ORCID for Kenneth McGarry

Catalogue record

Date Deposited
04 Nov 2024 11:20
Last Modified
04 Jun 2025 14:57

Contributors

Author
Bamidele Ajayi
Author
Mays Abukeshek

University Divisions

Subjects

Actions (login required)

View Item (Repository Staff Only)

Downloads per month over past year