Skip to content

Latest commit

 

History

History
44 lines (36 loc) · 1.4 KB

README.md

File metadata and controls

44 lines (36 loc) · 1.4 KB
Raw

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

@article{hindy2020machine,
  title={Machine Learning Based IoT Intrusion Detection System: An MQTT Case Study},
  author={Hindy, Hanan and Bayne, Ethan and Bures, Miroslav and Atkinson, Robert and Tachtatzis, Christos and Bellekens, Xavier},
  journal={arXiv preprint arXiv:2006.15340},
  year={2020}
}

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.