A Link Quality Prediction Method for Wireless Sensor Networks Based on XGBoost

• Main Authors: Yi Feng, Linlan Liu, Jian Shu
• Format: Article
• Language: English
• Published: IEEE 2019-01-01
• Series: IEEE Access
• Subjects: Wireless sensor networks; link quality prediction; XGBoost; improved fuzzy C-means algorithm
• Online Access: https://ieeexplore.ieee.org/document/8883148/

Link quality is an important factor for nodes selecting communication links in wireless sensor networks. Effective link quality prediction helps to select high quality links for communication, so as to improve stability of communication. We propose the improved fuzzy C-means clustering algorithm (SUBXBFCM) and use it to adaptively divide the link quality grades according to the packet reception rate. The Pearson correlation coefficient is employed to analyse the correlation between the hardware parameters and packet reception rate. The averages of the received signal strength indicator, link quality indicator and the signal to noise ratio are selected as the inputs of the link quality estimation model based on the XGBoost (XGB_LQE). The XGB_LQE is constructed to estimate the current link quality grade, which takes the classification advantages of XGBoost. Based on the estimated results of the XGB_LQE, the link quality prediction model (XGB_LQP) is constructed by using the XGBoost regression algorithm, which can predict the link quality grade at the next moment with historical link quality information. Experiment results in single-hop scenarios of square, laboratory, and grove show that the SUBXBFCM algorithm is effective at dividing the link quality grades compared with the normal division methods. Compared with link quality prediction methods based on the Support Vector Regression and 4C, XGB_LQP makes better predictions in single-hop wireless sensor networks.