We compare SVM, random forest, light GBM, logistic regression, KNN, and neural networks for classifying whether a network connection is a cyberattack or normal IoT activity based on packet capture data. We analyze model accuracy, model memory usage, and inference times. Our classical models are trained using Sklearn while our neural nets are implemented in PyTorch. We apply post training quantization and quantization aware training (8-bit) in PyTorch and TensorRT to significantly reduce the inference times of our neural nets.
-
Notifications
You must be signed in to change notification settings - Fork 0
alexngUNC/IoT-Intrusion-Detection
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published