Cyber-threat detection system using a hybrid approach of transfer learning and multi-model image representation
Article
Ullah, F., Ullah, S., Naeem, M., Mostarda, L., Rho, S. and Cheng, X. 2022. Cyber-threat detection system using a hybrid approach of transfer learning and multi-model image representation. Sensors. 22 (15), pp. 1-26. https://doi.org/10.3390/s22155883
Type | Article |
---|---|
Title | Cyber-threat detection system using a hybrid approach of transfer learning and multi-model image representation |
Authors | Ullah, F., Ullah, S., Naeem, M., Mostarda, L., Rho, S. and Cheng, X. |
Abstract | Currently, Android apps are easily targeted by malicious network traffic because of their constant network access. These threats have the potential to steal vital information and disrupt the commerce, social system, and banking markets. In this paper, we present a malware detection system based on word2vec-based transfer learning and multi-model image representation. The proposed method combines the textual and texture features of network traffic to leverage the advantages of both types. Initially, the transfer learning method is used to extract trained vocab from network traffic. Then, the malware-to-image algorithm visualizes network bytes for visual analysis of data traffic. Next, the texture features are extracted from malware images using a combination of scale-invariant feature transforms (SIFTs) and oriented fast and rotated brief transforms (ORBs). Moreover, a convolutional neural network (CNN) is designed to extract deep features from a set of trained vocab and texture features. Finally, an ensemble model is designed to classify and detect malware based on the combination of textual and texture features. The proposed method is tested using two standard datasets, CIC-AAGM2017 and CICMalDroid 2020, which comprise a total of 10.2K malware and 3.2K benign samples. Furthermore, an explainable AI experiment is performed to interpret the proposed approach. |
Keywords | malware detection; malware visualization; transfer learning; network traffic; explainable AI; cyber security |
Publisher | MDPI AG |
Journal | Sensors |
ISSN | 1424-8220 |
Publication dates | |
Online | 06 Aug 2022 |
01 Aug 2022 | |
Publication process dates | |
Deposited | 09 Aug 2022 |
Submitted | 14 Jun 2022 |
Accepted | 04 Aug 2022 |
Output status | Published |
Publisher's version | License |
Copyright Statement | Copyright: © 2022 by the authors. Licensee MDPI, Basel, Switzerland. |
Additional information | This article belongs to the Special Issue Cybersecurity Issues in Smart Grids and Future Power Systems |
Digital Object Identifier (DOI) | https://doi.org/10.3390/s22155883 |
Web of Science identifier | WOS:000839698900001 |
Language | English |
https://repository.mdx.ac.uk/item/89y16
Download files
70
total views28
total downloads0
views this month0
downloads this month