A Machine Learning Approach to Mental Disorder Prediction: Handling the Missing Data Challenge

Tsholofelo Mokheleli, Tebogo Bokaba, Tinofirei Museba, Nompumelelo Ntshingila

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In recent years, the application of Machine Learning (ML) to predict mental disorders has gained significant attention due to its potential for early prediction. This study highlights the challenges of ML in mental disorders prediction, such as missing data in mental health datasets, by comparing four data imputation methods: Mode, Multivariate Imputation by Chained Equations, Hot Deck, and K-Nearest Neighbor (K-NN) to enhance predictive accuracy; and utilizing four ML classifiers and three ensemble methods: Bagging, Boosting, and Stacking, with Mode and K-NN imputation datasets to show consistent performance. The study ultimately contributes to early mental disorder diagnosis and intervention in alignment with the United Nations Sustainable Development Goal 3 (SDG 3) for global health and well-being, by highlighting ML and data imputation’s potential in mental health analysis and paving the way for further advancements in the field.

Original languageEnglish
Title of host publicationEmerging Technologies for Developing Countries - 6th EAI International Conference, AFRICATEK 2023, Proceedings
EditorsMuthoni Masinde, Sabine Möbs, Antoine Bagula
PublisherSpringer Science and Business Media Deutschland GmbH
Pages93-106
Number of pages14
ISBN (Print)9783031639982
DOIs
Publication statusPublished - 2024
Event6th EAI International Conference on Emerging Technologies for Developing Countries, AFRICATEK 2023 - Arusha, Tanzania, United Republic of
Duration: 11 Dec 202313 Dec 2023

Publication series

NameLecture Notes of the Institute for Computer Sciences, Social-Informatics and Telecommunications Engineering, LNICST
Volume520 LNICST
ISSN (Print)1867-8211
ISSN (Electronic)1867-822X

Conference

Conference6th EAI International Conference on Emerging Technologies for Developing Countries, AFRICATEK 2023
Country/TerritoryTanzania, United Republic of
CityArusha
Period11/12/2313/12/23

Keywords

  • Data Imputation
  • Machine Learning
  • Mental Disorders Prediction
  • Missing Values

ASJC Scopus subject areas

  • Computer Networks and Communications

Fingerprint

Dive into the research topics of 'A Machine Learning Approach to Mental Disorder Prediction: Handling the Missing Data Challenge'. Together they form a unique fingerprint.

Cite this