AI-Driven Malware Detection with VGG Feature Extraction and Artificial Rabbits Optimized Random Forest Model
Source abstract
Abstract
Detecting cyber attacks in networks connected to the Internet of Things (IoT) is of utmost importance because of the growing vulnerabilities in the smart environment. Conventional models, such as Naive Bayes and support vector machine (SVM), as well as ensemble methods, such as Gradient Boosting and eXtreme gradient boosting (XGBoost), are often plagued by high computational costs, which makes it challenging for them to perform real-time detection. In this regard, we suggested an attack detection approach that integrates Visual Geometry Group 16 (VGG16), Artificial Rabbits Optimizer (ARO), and Random Forest Model to increase detection accuracy and operational efficiency in Internet of Things (IoT) networks. In the suggested model, the extraction of features from malware pictures was accomplished with the help of VGG16. The prediction process is carried out by the random forest model using the extracted features from the VGG16. Additionally, ARO is used to improve the hyper-parameters of the random forest model of the random forest. With an accuracy of 96.36%, the suggested model outperforms the standard models in terms of accuracy, F1-score, precision, and recall. The comparative research highlights our strategy’s success, which improves performance while maintaining a lower computational cost. This method is ideal for real-time applications, but it is effective.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Advanced Malware Detection Techniques · Network Security and Intrusion Detection · Anomaly Detection Techniques and Applications
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (30)
Show source-provided references
1. Classification of malware in https traffic using machine learning approach
2. Deep learning based hybrid analysis of malware detection and classification: a recent review
3. Forensic analysis of a ransomware10.35940/ijitee.C8385.019320 ↗
4. Particle swarm algorithm for smart contract vulnerability detection based on semantic web
5. Intelligent mirai malware detection for IoT nodes10.3390/electronics10111241 ↗
6. A review of cloud-based malware detection system: opportunities, advances and challenges10.24018/ejeng.2021.6.3.2372 ↗
7. Detection of malicious software and control using artificial intelligence
8. A network intrusion detection method for information systems using federated learning and improved transformer10.4018/IJSWIS.361772 ↗
9. A survey on malware analysis techniques: static, dynamic, hybrid and memory analysis10.18517/ijaseit.8.4-2.6827 ↗
10. A method for malware analysis by virtual machine introspection technique10.13053/rcs-147-12-1 ↗
11. Intelligent anomaly detection system through malware image augmentation in iiot environment based on digital twin10.3390/app131810196 ↗
12. IMCNN: intelligent malware classification using deep convolution neural networks as transfer learning and ensemble learning in honeypot enabled organizational network10.1016/j.comcom.2023.12.036 ↗
13. A secure framework to prevent three-tier cloud architecture from malicious malware injection attacks
14. Malware identification using visualization images and deep learning10.1016/j.cose.2018.04.005 ↗
15. A novel deep federated learning-based model to enhance privacy in critical infrastructure systems10.4018/IJSSCI.334711 ↗
16. A hybrid deep learning image-based analysis for effective malware detection
17. Earthworm optimization algorithm based cascade LSTM-GRU model for android malware detection10.1016/j.csa.2024.100083 ↗
18. Deep learning for image-based mobile malware detection10.1007/s11416-019-00346-7 ↗
19. CloudIntellMal: an advanced cloud based intelligent malware detection framework to analyze android applications10.1016/j.compeleceng.2024.109483 ↗
20. Malware detection in industrial internet of things based on hybrid image visualization and deep learning model10.1016/j.adhoc.2020.102154 ↗
21. From data and model levels: improve the performance of few-shot malware classification10.1109/TNSM.2022.3200866 ↗
22. Ghahramani M, Taheri R, Shojafar M, Javidan R, Wan S. Deep image: a precious image based deep learning method for online malware detection in IoT Environment. arXiv:2204.01690. 2022.
23. Orchestration of APT malware evasive manoeuvers employed for eluding anti-virus and sandbox defense10.1016/j.cose.2022.102627 ↗
24. SoK: visualization-based malware detection techniques
25. IMCLNet: a lightweight deep neural network for image-based malware classification
26. IVMCT: image visualization based multiclass malware classification using transfer learning
27. Malware analysis using visualized images and entropy graphs10.1007/s10207-014-0242-0 ↗
28. Matthew M. Malware as Images. 2020. [cited 2024 Jan 30]. Available from: https://www.kaggle.com/datasets/matthewfields/malware-as-images.
29. Artificial rabbits optimization: a new bio-inspired meta-heuristic algorithm for solving engineering optimization problems10.1016/j.engappai.2022.105082 ↗
30. Computational complexity and analysis of supervised machine learning algorithms