This repository contains the implementation and results of the DistillGuard framework (Knowledge Distillation for IoT Security) applied to three major datasets:
- ToN-IoT (Heterogeneous)
- RT-IoT (Real-Time)
- N-BaIoT (Botnet Traffic)
DistillGuard-ToN-IoT/: Implementation for ToN-IoT.DistillGuard-RT-IoT/: Implementation for RT-IoT.DistillGuard-N-BaIoT/: Implementation for N-BaIoT.
| Dataset | Teacher Acc | Student Acc | Compression | Robustness |
|---|---|---|---|---|
| ToN-IoT | 96% | 95.9% | 4x | Low |
| RT-IoT | 99% | 99.5% | 4x | Low |
| N-BaIoT | 17% (Fail) | 82.6% (Success) | 4x | Low |
See DistillGuard-N-BaIoT/comparison_study.pdf for the full report.
Each directory contains its own README.md and scripts:
data_processor.pytrain_distillguard.pyevaluate.py
- PyTorch
- Pandas
- Scikit-Learn
- Matplotlib