https://github.com/gaurav0502/malware-classification
Malware Byteplot Image Classification using Machine Learning and Deep Learning
https://github.com/gaurav0502/malware-classification
byteplot-image-classification computer-vision convolutional-neural-networks deep-learning image-classification malimg-dataset malware-analysis malware-classification python tensorflow
Last synced: 8 months ago
JSON representation
Malware Byteplot Image Classification using Machine Learning and Deep Learning
- Host: GitHub
- URL: https://github.com/gaurav0502/malware-classification
- Owner: Gaurav0502
- Created: 2022-12-13T08:50:03.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2023-10-09T09:20:24.000Z (over 2 years ago)
- Last Synced: 2024-12-19T19:17:28.127Z (over 1 year ago)
- Topics: byteplot-image-classification, computer-vision, convolutional-neural-networks, deep-learning, image-classification, malimg-dataset, malware-analysis, malware-classification, python, tensorflow
- Language: Jupyter Notebook
- Homepage: https://arxiv.org/abs/2310.02742
- Size: 875 MB
- Stars: 13
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Malware Byteplot Image Classification
## Aim
A comparison of variation in model convergence and performance with change in class imbalance. To assess the variation three datasets are created with varying imbalance in class distribution namely Malimg dataset, Malevis dataset, and Blended dataset. For comparison, the state-of-the-art CNNs from Keras library are explored namely: InceptionNet, ResNet50, DenseNet169, EfficientNetB4, InceptionResNetV2, and VGG16
## Datasets
- The Malimg dataset is available here.
- The Malevis dataset is available here and its website is here.
- The Blended Malware dataset is available here.
## Kaggle Notebooks
The tasks of Malware Classification and a minor fix to it were carried out in Kaggle using GPU P100. The codes can be found in the respective notebooks:
- Malware Classification.
- Malware Classification (Fixed).
## Results
The tabulated results for the comparison are shown in the figure below:
## Conference Paper Citation
```
@misc{m2023comparative,
title={Comparative Analysis of Imbalanced Malware Byteplot Image Classification using Transfer Learning},
author={Jayasudha M and Ayesha Shaik and Gaurav Pendharkar and Soham Kumar and Muhesh Kumar B and Sudharshanan Balaji},
year={2023},
eprint={2310.02742},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
```