Machine learning attack detection based-on stochastic classifier methods for enhancing of routing security in wireless sensor networks

Abstract

Wireless Sensor Networks (WSNs) are vulnerable to attacks during data transmission, and many techniques have been proposed to detect and secure routing data. In this paper, we introduce a novel stochastic predictive machine learning approach designed to discern untrustworthy events and unreliable routing attributes, aiming to establish an artificial intelligence-based attack detection system for WSNs. Our methodology leverages real-time analysis of the features of simulated WSN routing data. By integrating Hidden Markov Models (HMM) with Gaussian Mixture Models (GMM), we develop a robust classification framework. This framework effectively identifies outliers, pinpoints malicious network behaviors from their origins, and categorizes them as either trusted or untrusted network activities. In addition, dimensionality reduction techniques are used to improve interpretability, reduce computation and processing time, extract uncorrelated features from network data, and optimize performances. The main advantage of our approach is to establish an efficient stochastic machine learning method capable of analyzing and filtering WSN traffic to prevent suspicious and unsafe data, reduce the large dissimilarity in the collected routing features, and rapidly detect attacks before they occur. In this work, we exploit a well-tuned data set that provides a lot of routing information without losing any data. The experimental results show that the proposed stochastic attack detection system can effectively identify and categorize anomalies in wireless sensor networks with high accuracy. The classification rates of the system were found to be around 83.65%, 84.94% and 94.55%, which is significantly better than the existing classification approaches. Furthermore, the proposed system showed a positive prediction value of 11.84% higher than the existing approaches.