Hybridization of Machine Learning Techniques in Predicting Mental Disorder
Abstract
This study applied a hybrid Random Forest and Artificial Neural Network (RF-ANN) model in predicting the chances of IT employees developing mental disorder. To measure the performance of the model, Random Forest and Artificial Neural Networks algorithms were separately developed, their results were recorded and compared with the results of the hybridized model. The result obtained from the hybridized model showed a significant improvement in its performance over the individual performances of the Random Forest model and Artificial Neural Networks models. Hybridizing Random Forest and Artificial Neural Networks using “Bagging Ensemble” produced a model that was able to correctly predict the chances of IT employees developing Mental Disorder with 94% recall and 80% F1-score compared to 65% and 60% respectively in the Random Forest Model. With these results, applying the RF-ANN model on improved dataset could be investigated and compared with the results found in this study.
References
P. Sandhya, and K. Mahek, “Prediction of Mental Disorder for employees in IT Industry.” IJITEE (International Journal of Innovative Technology and Exploring Engineering, 2019), 8(6s): 2278-3075.
S. S. Lim, T. Vos, A. D. Flaxman, G. Danaei, K. Shibuya, H. Adair-Rohani, M. A. AlMazroa, M. Amann, H. R. Anderson, and K. G. Andrews. “A comparative risk assessment of burden of disease and injury attributable to 67 risk factors and risk factor clusters in 21 regions, 1990 – 2010: a systematic analysis for the Global Burden of Disease Study 2010.” The Lancet, 2012, 380(9859):2224-2260.
J. F. Dipnall, J. A. Pasco, M. Berk, L. J. Williams, S. Dodd, F. N. Jacka, and D. Meyer. “Fusing data mining, machine learning and traditional statistics to detect biomarkers associated with depression.” PloS one, (2016) 11(2):148-195.
J. Han, J. Pei, and M. Kamber. “Data Mining: Concepts and Techniques.” Elsevier, New York, United States, (2011).
T. M. Awoyelu A. R. Iyanda and S. K. Mosaku. “Formulation of a Predictive Model for the Determination of Depression among University Students.” Proceeding of the 14th International Conference on Smart Nations, Digital Economies and Meaningful Lives Nigerian Computer Society, (NCS, 2019), 098- 109, Nigeria.
S. Wang, C. Aggarwal, and H. Liu. Random-Forest-Inspired Neural Networks. ACM Trans. Intell. Syst. Technol., 2018. 9, 6(69), 25 pages. https://doi.org/10.1145/3232230.
P. Gilbert. Depression: The Evolution of Powerlessness. Routledge, London, United Kingdom, 2016.
L. A. Cooper, J. J. Gonzales, J. J Gallo, K. M Rost, L. S. Meredith, L. V. Rubenstein, N. Y. Wang, and D. E. Ford. The Acceptability of Treatment for Depression among African-American, Hispanic, and White Primary Care Patients. Medical Care, (2003). 41(4):479-489.
M. Kantardzic. Data Mining: Concepts, Models, Methods, and Algorithms. John Wiley and Sons, New Jersey, United States, 2011.
P. Groves, B. Kayyali, D. Knott, and S. Van Kuiken, S. The Big Data Revolution in Healthcare. McKinsey Quarterly, 2:3-5, 2013.
M. R. Sumathi, and B. Poorna. “Prediction of Mental Health Problems Among Children Using Machine Learning Techniques.” IJACSA (International Journal of Advanced Computer Science and Applications, 2016). 7(1).
G. Azar, C. Gloster, N. El-Bathy, S. Yu, R. H. Neela, and I. Alothman. “Intelligent Data Mining and Machine Learning for Mental Health Diagnosis using Genetic Algorithm.” (2015)
S. Chauhan, and A. Garg. Predictive Research for Mental Health Disease. Blue Eyes Intelligence Engineering & Sciences Publication, 2019. https://doi.org/10.35940/ijitee.I1038.0789S219.
G. Orrù, W. Pettersson-Yeo, A. F. Marquand, G. Sartori, and A. Mechelli. “Using Support Vector Machine to Identify Imaging Biomarkers of Neurological and Psychiatric Disease: A Critical Review.” Neuroscience and Biobehavioral Reviews 36 (2012) 1140–1152.
L. Jing. “Mental Health Evaluation Model based on Fuzzy Neural Network.” 2016 International Conference on Smart Grid and Electrical Automation (2016).
F. Sadeque, D. Xu, and S. Bethard. “UArizona at the CLEF eRisk 2017 Pilot Task: Linear and Recurrent Models for Early Depression Detection.” In CEUR workshop proceedings, volume 1866. NIH Public Access (2017).
M. Srividya, S. Mohanavalli, and N. Bhalaji. “Behavioral Modelling for Mental Health using Machine Learning Algorithms.” Journal of Medical Systems (2018). https://doi.org/10.1007/s10916-018-0934-5.
A. Khan, M. H. Husain, and A. Khan. “Analysis of Mental State of Users using Social Media to Predict Depression! A Survey.” Conference Paper at the International Journal of Advanced Research in Computer Science (2018). 9(2):0976-5697.