Machine Learning-based Regression Analysis and Feature Ranking for Localization Error Prediction in Wireless Sensor Networks
Abstract
Wireless Sensor Networks (WSNs) localization is crucial for identifying the position of sensor nodes, as many applications, including environmental monitoring, target tracking, and disaster management, require accurate location information. The objective of this research is to conduct extensive data analytics using visualization techniques to explore key factors influencing localization error and to develop machine learning models for forecasting Average Localization Error (ALE) in WSNs. A dataset containing 107 records, sourced from Kaggle’s online repository, was analyzed using eXtreme Gradient Boosting (XGB) for feature ranking to determine the most influential factors affecting ALE. Multiple regression models, including Support Vector Regression (SVR), Decision Tree (DT), K-Nearest Neighbors (KNN), and AdaBoost Regressor, were applied to predict ALE. The models were evaluated using R-squared (R²), Root Mean Square Error (RMSE), and computational efficiency. The results indicate that SVR achieved the highest accuracy with R² = 0.99 and the lowest RMSE of 0.01, significantly outperforming the other models (KNN: R² = 0.55, RMSE = 0.14; DT: R² = 0.41, RMSE = 0.16; AdaBoost: R² = 0.72, RMSE = 0.16). This study demonstrates that SVR is a highly effective model for ALE prediction, reinforcing the importance of feature ranking and selection in improving localization accuracy. The findings contribute to advancing machine learning-driven localization error prediction in WSNs and provide a foundation for further exploration of hybrid and deep learning-based models.
Full Text:
PDFReferences
Singh, A., Kotiyal, V., Sharma, S., Nagar, J., & Lee, C. C. (2020). A machine learning approach to predict the average localization error with applications to wireless sensor networks. IEEE Access, 8, 208253-208263.
Watanabe, F. (2021). Wireless sensor network localization using AoA measurements with two-step error variance-weighted least squares. IEEE Access, 9, 10820-10828.
Zhu, Z., & Zhu, M. (2024). Utilizing Machine Learning Approach to Forecast Average Location Determination Errors in Wireless Sensor Networks. Journal of Artificial Intelligence and System Modelling, 1(02), 83-98.
Lalama, Z., Boulfekhar, S., & Semechedine, F. (2022). Localization optimization in WSNs using meta-heuristics optimization algorithms: a survey. Wireless Personal Communications, 122(2), 1197-1220.
He, J., Fakhreddine, A., Vanwynsberghe, C., Wymeersch, H., & Alexandropoulos, G. C. (2023). 3D localization with a single partially-connected receiving RIS: Positioning error analysis and algorithmic design. IEEE Transactions on Vehicular Technology, 72(10), 13190-13202.
Zou, Y., & Liu, H. (2021). RSS-based target localization with unknown model parameters and sensor position errors. IEEE Transactions on Vehicular Technology, 70(7), 6969-6982.
Jia, Z., & Guan, B. (2018). Received signal strength difference–based tracking estimation method for arbitrarily moving target in wireless sensor networks. International Journal of Distributed Sensor Networks, 14(3), 1550147718764875.
Badshah, A., Daud, A., Khan, H. U., Alghushairy, O., & Bukhari, A. (2024). Optimizing the Over and Underutilization of Network Resources During Peak and Off-Peak Hours. IEEE Access.
Han, G., Yang, X., Liu, L., Zhang, W., & Guizani, M. (2017). A disaster management-oriented path planning for mobile anchor node-based localization in wireless sensor networks. IEEE Transactions on Emerging Topics in Computing, 8(1), 115-125.
Sorbelli, F. B., Pinotti, C. M., Silvestri, S., & Das, S. K. (2020). Measurement errors in range-based localization algorithms for UAVs: Analysis and experimentation. IEEE Transactions on Mobile Computing, 21(4), 1291-1304.
Ali, T., Khan, H. U., Alarfaj, F. K., & AlReshoodi, M. (2024). Hybrid deep learning and evolutionary algorithms for accurate cloud workload prediction. Computing, 106(12), 3905-3944.
Sneha, V., & Nagarajan, M. (2020). Localization in wireless sensor networks: a review. Cybernetics and Information Technologies, 20(4), 3-26.
Tian, Y., Huang, B., Jia, B., & Zhao, L. (2020). Optimizing AP and Beacon Placement in WiFi and BLE hybrid localization. Journal of Network and Computer Applications, 164, 102673.
Abhale, A. B., & Manivannan, S. S. (2020). Supervised machine learning classification algorithmic approach for finding anomaly type of intrusion detection in wireless sensor network. Optical Memory and Neural Networks, 29(3), 244-256.
Shahraki, A., Taherkordi, A., Haugen, Ø., & Eliassen, F. (2020). Clustering objectives in wireless sensor networks: A survey and research direction analysis. Computer Networks, 180, 107376.
Yang, S., Yuan, Z., & Li, W. (2020). Error data analytics on RSS range-based localization. Big Data Mining and Analytics, 3(3), 155-170.
Janczak, D., Walendziuk, W., Sadowski, M., Zankiewicz, A., Konopko, K., & Idzkowski, A. (2022). Accuracy analysis of the indoor location system based on Bluetooth low-energy RSSI measurements. Energies, 15(23), 8832.
Zheng, Y., Liu, J., Sheng, M., Han, S., Shi, Y., & Valaee, S. (2020). Toward practical access point deployment for angle-of-arrival based localization. IEEE Transactions on Communications, 69(3), 2002-2014.
Lakshmi, Y. V., Singh, P., Mahajan, S., Nayyar, A., & Abouhawwash, M. (2024). Accurate range-free localization with hybrid DV-hop algorithms based on PSO for UWB wireless sensor networks. Arabian Journal for Science and Engineering, 49(3), 4157-4178.
Kumar, S., Kumar, S., & Batra, N. (2021). Optimized distance range free localization algorithm for WSN. Wireless Personal Communications, 117, 1879-1907.
Wu, Y., Zhang, C., Tong, L., & Shi, X. (2023). Location Optimization based on Improved 3D DV-HOP Algorithm in Wireless Sensor Networks. IEEE Access.
Arya, R. (2021). C-TOL: Convex triangulation for optimal node localization with weighted uncertainties. Physical Communication, 46, 101300.
Kuriakose, J., Joshi, S., & Bairwa, A. K. (2023). Embn-manet: A method to eliminating malicious beacon nodes in ultra-wideband (uwb) based mobile ad-hoc network. Ad Hoc Networks, 140, 103063.
Yang, H., Wang, Y., Seow, C. K., Sun, M., Si, M., & Huang, L. (2023). UWB sensor-based indoor LOS/NLOS localization with support vector machine learning. IEEE Sensors Journal, 23(3), 2988-3004.
Olejniczak, A., Blaszkiewicz, O., Cwalina, K. K., Rajchowski, P., & Sadowski, J. (2024). LOS and NLOS identification in real indoor environment using deep learning approach. Digital Communications and Networks, 10(5), 1305-1312.
Akram, J., Javed, A., Khan, S., Akram, A., Munawar, H. S., & Ahmad, W. (2021, March). Swarm intelligence based localization in wireless sensor networks. In Proceedings of the 36th annual ACM symposium on applied computing (pp. 1906-1914).
Li, Y., Zhuang, Y., Hu, X., Gao, Z., Hu, J., Chen, L., ... & El-Sheimy, N. (2020). Toward location-enabled IoT (LE-IoT): IoT positioning techniques, error sources, and error mitigation. IEEE Internet of Things Journal, 8(6), 4035-4062.
Yadav, P., & Sharma, S. C. (2023). A systematic review of localization in WSN: Machine learning and optimization‐based approaches. International journal of communication systems, 36(4), e5397.
Mohammed, S. K., Singh, S., Mizouni, R., & Otrok, H. (2023). A deep learning framework for target localization in error-prone environment. Internet of Things, 22, 100713.
Jondhale, S. R., Shubair, R., Labade, R. P., Lloret, J., & Gunjal, P. R. (2020). Application of supervised learning approach for target localization in wireless sensor network. Handbook of Wireless Sensor Networks: Issues and Challenges in Current Scenario's, 493-519.
Lemic, F., Handziski, V., & Famaey, J. (2019, April). Toward regression-based estimation of localization errors in fingerprinting-based localization. In 2019 IEEE 89th Vehicular Technology Conference (VTC2019-Spring) (pp. 1-5). IEEE.
Giannopoulos, A., Spantideas, S., Nomikos, N., Kalafatelis, A., & Trakadas, P. (2023, April). Learning to fulfill the user demands in 5G-enabled wireless networks through power allocation: A reinforcement learning approach. In 2023 19th International Conference on the Design of Reliable Communication Networks (DRCN) (pp. 1-7). IEEE.
Hajiakhondi-Meybodi, Z., Mohammadi, A., Hou, M., & Plataniotis, K. N. (2022). DQLEL: Deep Q-learning for energy-optimized LoS/NLoS UWB node selection. IEEE Transactions on Signal Processing, 70, 2532-2547.
Altay, O., Erel-Özçevik, M., Varol Altay, E., & Özçevik, Y. (2024). Average Localization Error Prediction for 5G Networks: An Investigation of Different Machine Learning Algorithms. Wireless Personal Communications, 1-31.
Poulose, A., & Han, D. S. (2020). UWB indoor localization using deep learning LSTM networks. Applied Sciences, 10(18), 6290.
Zhao, L., Huang, H., Li, X., Ding, S., Zhao, H., & Han, Z. (2019). An accurate and robust approach of device-free localization with convolutional autoencoder. IEEE Internet of Things Journal, 6(3), 5825-5840.
Park, J., Moon, J., Kim, T., Wu, P., Imbiriba, T., Closas, P., & Kim, S. (2022). Federated learning for indoor localization via model reliability with dropout. IEEE Communications Letters, 26(7), 1553-1557.
Ghous, M., Nguyen, T. L., Do, T. N., & Kaddoum, G. (2024). Deep Transfer Learning-based Performance Prediction of URLLC in independent and not necessarily identically distributed Interference Networks. IEEE Access.
Ishfaq, U., Shabbir, D., Khan, J., Khan, H. U., Naseer, S., Irshad, A., ... & Hamam, H. (2022). Empirical analysis of machine learning algorithms for multiclass prediction. Wireless Communications and Mobile Computing, 2022(1), 7451152.
Masood, A., Gulzar Ahmad, S., Ullah Khan, H., & Ullah Munir, E. (2020). Network reconfiguration algorithm (NRA) for scheduling communication-intensive graphs in heterogeneous computing environment. Cluster Computing, 23(2), 1419-1438.
Rahman, M. M., & Nisher, S. A. (2023, January). Predicting average localization error of underwater wireless sensors via decision tree regression and gradient boosted regression. In Proceedings of International Conference on Information and Communication Technology for Development: ICICTD 2022 (pp. 29-41). Singapore: Springer Nature Singapore.
Jondhale, S. R., Mohan, V., Sharma, B. B., Lloret, J., & Athawale, S. V. (2022). Support vector regression for mobile target localization in indoor environments. Sensors, 22(1), 358.
Nosrati, L., Fazel, M. S., & Ghavami, M. (2022). Improving indoor localization using mobile UWB sensor and deep neural networks. IEEE Access, 10, 20420-20431.
DOI: https://doi.org/10.31449/inf.v49i20.8081
This work is licensed under a Creative Commons Attribution 3.0 License.