Classification of Botnet Attacks in IoT Using a Convolutional Neural Network
Andressa A. Cunha, João Batista Borges, Antônio A. F. Loureiro · 2022
Detecting malicious attacks on Internet of Things (IoT) devices is a current research trend due to the rise of Botnet attacks across different IoT environments and the lack of standardization in IoT's security field. To tackle these issues, deep learning techniques are a promising strategy to detect and prevent attacks on IoT ecosystems. Most proposals are only concerned with detecting the occurrence of the attack, but classifying its type could be an important additional step. This work proposes a Convolutional Neural Network (CNN) model that can be employed to classify the type of the attack after a proper botnet detection evaluated with N-BaIoT and Bot-IoT datasets. The model achieved a 98% of F1-score and accuracy on N-BaIoT and 100% of F1-score and accuracy on Bot-IoT. Moreover, our model was also compared with recent literature results, including Naive Bayes (NB) and KNN (K-Nearest Neighbours). The work also evaluates the time interval spent on classification tasks, which is an important feature to consider when implementing solutions in edge computing environments.