Skip to content

Repository files navigation

Malware Detection System

Overview

The Malware Detection System is a Machine Learning-based web application that analyzes executable files and predicts whether they are Malware or Benign. The project leverages a trained Random Forest Classifier and integrates threat intelligence services to provide real-time security insights through an interactive Streamlit dashboard.


Demo

This application provides:

  • Malware/Benign prediction using Machine Learning
  • VirusTotal threat intelligence integration
  • File hash generation (MD5 & SHA256)
  • Feature importance visualization
  • PDF security report generation
  • Scan history tracking and analytics

Features

  • Malware Detection using Machine Learning
  • File Hash Generation (MD5 & SHA256)
  • VirusTotal Threat Intelligence Integration
  • Feature Importance Visualization
  • PDF Security Report Generation
  • Scan History Dashboard
  • Interactive Data Visualizations

Machine Learning Model

Algorithm: Random Forest Classifier

Features Used

  • File Size
  • Entropy
  • Number of Sections
  • Suspicious API Calls
  • Network Connections
  • Registry Modifications
  • File Operations
  • Process Injections
  • DLL Loads
  • Suspicious Strings

Tech Stack

  • Python
  • Streamlit
  • Scikit-learn
  • Pandas
  • NumPy
  • Plotly
  • ReportLab
  • VirusTotal API

Project Structure

Malware-Detection-System/
├── streamlit_app.py
├── Random_Forest_model.pkl
├── scaler.pkl
├── requirements.txt
├── screenshots/
└── README.md

Installation

pip install -r requirements.txt
streamlit run streamlit_app.py

Screenshots

Home Page

Home

Analysis Dashboard

Analysis

Feature Importance

Feature Importance

Scan History Dashboard

Scan History


Future Enhancements

  • SHAP Explainable AI Integration
  • Malware Family Classification
  • Cloud Deployment
  • Real-time Threat Monitoring

Author

Omm Miriyala B.Tech CSE (Data Science)


About

Machine Learning based Malware Detection System with VirusTotal Integration, PDF Reporting, Feature Importance Analysis, and Scan History Dashboard.

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages