A Grey Wolf Optimization Improved Deep Belief Network for Employee Attrition Prediction

A Grey Wolf Optimization Improved Deep Belief Network for Employee Attrition Prediction

© 2021 by IJETT Journal
Volume-69 Issue-10
Year of Publication : 2021
Authors : Usha.P.M, Dr. N.V. Balaji
DOI :  10.14445/22315381/IJETT-V69I10P210

How to Cite?

Usha.P.M, Dr. N.V. Balaji, "A Grey Wolf Optimization Improved Deep Belief Network for Employee Attrition Prediction," International Journal of Engineering Trends and Technology, vol. 69, no. 10, pp. 72-81, 2021. Crossref, https://doi.org/10.14445/22315381/IJETT-V69I10P210

The proposed study enables machine learning techniques to predict the probability of employee attrition. The study improves the accuracy of employee attrition prediction by developing an enhanced model using a Deep belief network or DBN. A deep belief network is a form of deep neural network made up of several layers of variables that are latent or hidden. This work uses the Restricted Boltzmann machine, which creates a stack of the network to analyze the pattern of the attrition dataset. The parameters involved in Deep Belief Network are fine-tuned by adapting a novel behavioral inspiration algorithm instead of random assignment of the values. The algorithm used here is the metaheuristic Grey Wolf algorithm which is an optimization algorithm that imitates the hunting behavior of Grey Wolves. Thus, it will increase the performance of the proposed model.

Machine learning, Attrition prediction, Restricted Boltzmann machine, Grey Wolf Optimisation

[1] Mohammed, D. A. HR Analytics: A Modern Tool In Hr For Predicitve Decision Making. Journal of Management (JOM) (2019) 51-63.
[2] Randall Sexton, S. M. ,Employee turnover: A neural network solution. Computers & Operations Research, (2005).
[3] Valenta, D.; Langer, M.,2021,On Numerical 2D P Colonies Modelling the Grey Wolf Optimization Algorithm. Processes 2021, 9, 330. https://doi.org/10.3390/pr9020330
[4] Emary E., Zawbaa H.M., Grosan C., Hassenian A.E. Feature Subset Selection Approach by Gray-Wolf Optimization. In: Abraham A., Krömer P., Snasel V. (eds) Afro-European Conference for Industrial Advancement. Advances in Intelligent Systems and Computing, (2015) 334. Springer, Cham. https://doi.org/10.1007/978-3-319-13572-4_1
[5] Sun J., Steinecker A., Glocker P. Application of Deep Belief Networks for Precision Mechanism Quality Inspection. In: Ratchev S. (eds) Precision Assembly Technologies and Systems. IPAS 2014. IFIP Advances in Information and Communication Technology, vol 435. Springer, Berlin, Heidelberg, (2014), https://doi.org/10.1007/978-3-662-45586-9_12
[6] Medium website , https://medium.com/swlh/what-are-rbms-deepbelief- networks-and-why-are-they-important-to-deep-learning- 491c7de8937a,2020
[7] Yeh, S.-H., Wang, C.-J., & Tsai, M.-F. Deep belief networks for predicting corporate defaults. (2015) 24th Wireless and Optical Communication Conference (WOCC). doi:10.1109/wocc.2015.7346197
[8] 2020, https://www.explainthatstuff.com/introduction-to-neuralnetworks. html#:~:text=Information%20flows%20through%20a%20 neural,arrive%20at%20the%20output%20units.
[9] Arnold, Ludovic& Rebecchi, Sébastien &Chevallier, Sylvain &Paugam-Moisy, Hélène. An Introduction to Deep Learning. Proceedings of the European Symposium of ,Artificial Neural Network, ESANN2011. 1. 477-488.
[10] Sun, Jianwen, Wyss, Reto, Steinecker, Alexander and Glocker, Philipp. Automated fault detection using deep belief networks for the quality inspection of electromotors: tm - Technisches Messen, 81(5) (2014) 255-263. https://doi.org/10.1515/teme-2014-1006
[11] Syed Atif Ali Shah, Irfan Uddin, Furqan Aziz, Shafiq Ahmad, Mahmoud Ahmad Al-Khasawneh, Mohamed Sharaf, An Enhanced Deep Neural Network for Predicting Workplace Absenteeism, Complexity, 2020(2020), Article ID 5843932,. https://doi.org/10.1155/2020/5843932
[12] Dumitru Erhan and YoshuaBengio and Aaron Courville and Pierre- Antoine Manzagol and Pascal Vincent and SamyBengio, Why Does Unsupervised Pre-training Help DeepLearning?,Journal of Machine Learning Research,2010, volume11,625-660}, http://jmlr.org/papers/v11/erhan10a.html [13] Rashid, Tarik & Abbas, Dosti&Turel, Yalin. A Multi Hidden Recurrent Neural Network with a Modified Grey Wolf Optimizer. ,2019, PLoS ONE. 14. 1-23. 10.1371/journal.pone.0213237.
[14] S. Kamada and T. Ichimura, An adaptive learning method of Deep Belief Network by layer generation algorithm, IEEE Region 10 Conference (TENCON), (2016) 2967-2970, doi: 10.1109/TENCON.2016.7848589.
[15] Y. Bengio, P. Lamblin, D. Popovici, and H. Larochelle. Greedy layer-wise training of deep networks. In Advances in Neural Information Processing Systems (NIPS 19), (2007).
[16] H.Z. Wang, G.B. Wang, G.Q. Li, J.C. Peng, Y.T. Liu, Deep belief network based deterministic and probabilistic wind speed forecasting approach, Applied Energy, Volume 182, 2016, Pages 80-93, ISSN 0306-2619, https://doi.org/10.1016/j.apenergy.2016.08.108.
[17] Al-Rammahi, Atheer&Radif, Mustafa. Neural networks in business applications. Journal of Physics: Conference Series. , 2019, 1294. 042007. 10.1088/1742-6596/1294/4/042007.
[18] Yara Rizk, Nadine Hajj, Nicholas Mitri, Mariette Awad, Deep belief networks and cortical algorithms: A comparative study for supervised classification, Applied Computing and Informatics, 15(2) (2019) 81-93, ISSN 2210-8327, https://doi.org/10.1016/j.aci.2018.01.004.
[19] Shaoyong Jia1 ,Shibao Sun2 , Jing Wang3 and Fengkui Xu4, Research Based on The Improved Grey Wolf Optimization Algorithm, Journal of Physics: Conference Series, IOP Publishing, (2021).
[20] Qinghua Gu, Xuexian Li, Song Jiang, Hybrid Genetic Grey Wolf Algorithm for Large-Scale Global Optimization, Complexity, 2019(2019), 2653512, 18 pages, 2019. https://doi.org/10.1155/2019/2653512
[21] Zhang, Yue & Liu, Fangai..An Improved Deep Belief Network Prediction Model Based on Knowledge Transfer. Future Internet. 12. 188. 10.3390/fi12110188, 2020.
[22] Yuming Hua, Junhai Guo and Hua Zhao, Deep Belief Networks and deep learning, Proceedings of 2015 International Conference on Intelligent Computing and Internet of Things, (2015) 1-4, doi: 10.1109/ICAIOT.2015.7111524.
[23] X. Yuan, Y. Gu and Y. Wang, Supervised Deep Belief Network for Quality Prediction in Industrial Processes, in IEEE Transactions on Instrumentation and Measurement, 70(2021) 1-11, Art no. 2503711, doi: 10.1109/TIM.2020.3035464.
[24] X. Li and K. M. Luk, The Grey Wolf Optimizer and Its Applications in Electromagnetics, in IEEE Transactions on Antennas and Propagation, 68(3)(2020) 2186-2197, doi: 10.1109/TAP.2019.2938703.
[25] B. Li, C. Luo and Z. Wang, Application of GWO-SVM Algorithm in Arc Detection of Pantograph, in IEEE Access, vol. 8, pp. 173865- 173873, 2020, doi: 10.1109/ACCESS.2020.3025714.
[26] Chahal, Ayushi &Gulia, Preeti. Deep Learning: A Predictive Iot Data Analytics Method. International Journal of Engineering Trends and Technology, 2020. 68. 25-33. 10.14445/22315381/IJETTV68I7P205S.