Machine Learning Based IoT IntrusionDetection System: An MQTT Case Study

This work uses six different machine learning techniques to classify attacks in an MQTT network.

Dataset Used

The used dataset is published in IEEE DataPort

@data{bhxy-ep04-20,
doi = {10.21227/bhxy-ep04},
url = {http://dx.doi.org/10.21227/bhxy-ep04},
author = {Hanan Hindy; Christos Tachtatzis; Robert Atkinson; Ethan Bayne; Xavier Bellekens },
publisher = {IEEE Dataport},
title = {MQTT Internet of Things Intrusion Detection Dataset},
year = {2020} } 

Citation

TBU

Algorithms Used

• Logistic Regression
• k-Nearest Neighbours
• Gaussian Naive Bayes
• Decision Trees
• Random Forests
• Support Vector Machine (linear and RBF kernel)

How to Run it:

Clone this repository
Download dataset files and extract them in the same directory
run classification.py --mode [0: packet, 1: unidirectional, 2: bidirectional] --output [output_folder] --verbose [True/False]

• The classification outputs are added to the output folder.