Botnet Attack Detection in IoT Using Machine Learning.

Journal: Computational intelligence and neuroscience
Published Date:

Abstract

There are an increasing number of Internet of Things (IoT) devices connected to the network these days, and due to the advancement in technology, the security threads and cyberattacks, such as botnets, are emerging and evolving rapidly with high-risk attacks. These attacks disrupt IoT transition by disrupting networks and services for IoT devices. Many recent studies have proposed ML and DL techniques for detecting and classifying botnet attacks in the IoT environment. This study proposes machine learning methods for classifying binary classes. This purpose is served by using the publicly available dataset UNSW-NB15. This dataset resolved a class imbalance problem using the SMOTE-OverSampling technique. A complete machine learning pipeline was proposed, including exploratory data analysis, which provides detailed insights into the data, followed by preprocessing. During this process, the data passes through six fundamental steps. A decision tree, an XgBoost model, and a logistic regression model are proposed, trained, tested, and evaluated on the dataset. In addition to model accuracy, F1-score, recall, and precision are also considered. Based on all experiments, it is concluded that the decision tree outperformed with 94% test accuracy.

Authors

  • Khalid Alissa
    Networks and Communications Department, College of Computer Science and Information Technology (CCSIT), Imam Abdulrahman Bin Faisal University (IAU), P.O. Box 1982, Dammam 31441, Saudi Arabia.
  • Tahir Alyas
    Department of Computer Science, Lahore Garrison University, Lahore 54000, Pakistan.
  • Kashif Zafar
    Department of Computer Science, National University of Computer and Emerging Sciences (NUCES), Lahore 54000, Pakistan.
  • Qaiser Abbas
    Faculty of Computer and Information Systems Islamic University Madinah, Madinah 42351, Saudi Arabia.
  • Nadia Tabassum
    Department of Computer Science, Virtual University of Pakistan, Lahore 54000, Pakistan.
  • Shadman Sakib
    Department of Finance and Banking, Jahangirnagar University, Bangladesh.