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.References
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.8081Downloads
Published
How to Cite
Issue
Section
License
I assign to Informatica, An International Journal of Computing and Informatics ("Journal") the copyright in the manuscript identified above and any additional material (figures, tables, illustrations, software or other information intended for publication) submitted as part of or as a supplement to the manuscript ("Paper") in all forms and media throughout the world, in all languages, for the full term of copyright, effective when and if the article is accepted for publication. This transfer includes the right to reproduce and/or to distribute the Paper to other journals or digital libraries in electronic and online forms and systems.
I understand that I retain the rights to use the pre-prints, off-prints, accepted manuscript and published journal Paper for personal use, scholarly purposes and internal institutional use.
In certain cases, I can ask for retaining the publishing rights of the Paper. The Journal can permit or deny the request for publishing rights, to which I fully agree.
I declare that the submitted Paper is original, has been written by the stated authors and has not been published elsewhere nor is currently being considered for publication by any other journal and will not be submitted for such review while under review by this Journal. The Paper contains no material that violates proprietary rights of any other person or entity. I have obtained written permission from copyright owners for any excerpts from copyrighted works that are included and have credited the sources in my article. I have informed the co-author(s) of the terms of this publishing agreement.
Copyright © Slovenian Society Informatika







