A Performance Evaluation of Neural Networks for Botnet Detection in the Internet of Things

A Performance Evaluation of Neural Networks for Botnet Detection in the Internet of Things

연구 분야: Networking

학회: Journal of Network and Systems Management

IoT (Internet of Things) devices are fundamental to sectors such as smart homes and cities, industry 4.0, and smart grids. Despite the benefits brought by IoT, the existence of billions of devices with limited computing resources makes them ideal targets for botnets. Thus, multiple proposals have been made to detect this type of attack. However, comparing different proposals is difficult since they apply varied preprocessing methods, use different algorithms and hyperparameters, and consider distinct evaluation metrics. This paper implements and compares the performance of eight neural network architectures applied to the BoT-IoT and N-BaIoT datasets, which contain botnet attacks and labeled IoT network traffic. The models’ accuracy, precision, and recall are measured, as well as the loss during model training. Afterward, the models’ throughput on an edge environment is evaluated using a typical edge device, an NVIDIA Jetson Nano, while also implementing quantization and evaluating its impact on model accuracy. The results show that, after hyperparameter tuning, several BoT-IoT models exceed 99% accuracy while most N-BaIoT models surpass 80% accuracy. However, the throughput results show that the best-performing model might not scale in environments composed of a large number of IoT devices, even considering the influence of 8-bit quantization.