Comparison of Selected Machine Learning Algorithms in the Analysis of Mental Health Indicators

• Published in: Electronics (Basel) Vol. 12; no. 21; p. 4407
• Main Authors: Bieliński, Adrian, Rojek, Izabela, Mikołajewski, Dariusz
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.11.2023
• Subjects: Algorithms; Anxiety; Bipolar disorder; Comparative analysis; Data mining; Diagnosis; Employees; Machine learning; Medical diagnosis; Mental depression; Mental disorders; Mental health; Mental health care; Patients; Professional relationships; Questionnaires; Run time (computers); Stress; Poland
• ISSN: 2079-9292; 2079-9292
• DOI: 10.3390/electronics12214407

Machine learning is increasingly being used to solve clinical problems in diagnosis, therapy and care. Aim: the main aim of the study was to investigate how the selected machine learning algorithms deal with the problem of determining a virtual mental health index. Material and Methods: a number of machine learning models based on Stochastic Dual Coordinate Ascent, limited-memory Broyden–Fletcher–Goldfarb–Shanno, Online Gradient Descent, etc., were built based on a clinical dataset and compared based on criteria in the form of learning time, running time during use and regression accuracy. Results: the algorithm with the highest accuracy was Stochastic Dual Coordinate Ascent, but although its performance was high, it had significantly longer training and prediction times. The fastest algorithm looking at learning and prediction time, but slightly less accurate, was the limited-memory Broyden–Fletcher–Goldfarb–Shanno. The same data set was also analyzed automatically using ML.NET. Findings from the study can be used to build larger systems that automate early mental health diagnosis and help differentiate the use of individual algorithms depending on the purpose of the system.