Using Machine Learning in WSNs for Performance Prediction MAC Layer

Abstract

To monitor environments, Wireless Sensor Networks (WSNs) are used for collecting data in divers domains such as smart factories, smart buildings, etc. In such environments, different medium access control (MAC) protocols are available to sensor nodes for wireless communications and are of a paramount importance to enhance the network performance. Proposed MAC layer protocols for WSNs are generally designed to achieve a good performance in packet reception rate. Once chosen, the MAC protocol is used and remains the same throughout the network lifetime even if its performance decreases over time. In this paper, we adopt supervised machine learning techniques to predict the performance of CSMA/CA MAC protocol based on the packet reception rate. Our approach consists of three steps: experiments for data collection, offline modeling and performance evaluation. Our analysis shows that XGBoost prediction model is the better supervised machine learning technique to enhance network performance at the MAC layer level. In addition, we use SHAP method to explain predictions.