A multi-label learning model for psychotic diseases in Nigeria

S.O. Folorunso, S.G. Fashoto, J. Olaomi, O.Y. Fashoto

Abstract:

The goal of Multi-Label Classification (MLC) is to allot an instance to a set of different labels. This task is usually
addressed by either transforming the problem into several binary problems, adapting machine learning models to
fit multi-label data or create an ensemble of models that can classify multi-label datasets. The communal relationship
between Bipolar, Insomnia, Schizophrenia, Vascular Dementia (VD) and Attention-Deficit/Hyperactivity
Disorder (ADHD) in the Psychotic Disorder Diseases (PDD) motivate the research for a diagnostic method that
classifies and evaluates each psychotic disorder simultaneously.
This study experimentally evaluates 15 MLC methods using 10 evaluation measures over a new PDD dataset.
The performance of these methods is measured with four ranking - based, three example-based and three labelbased
measures. Also, the efficiency of these methods is measured by their 90-10 Train-Test split with the 10
evaluation measures.
The results show that the Label Powerset (LC) and Pruned Sets (PS), MLC methods with Naïve Bayes (NB) and
Naïve Bayes Tree (NBTree) consistently performed best in terms of the evaluation measures on the PDD dataset.
Schizophrenia has the highest classification accuracy with Bipolar the lowest in the data split of 90-10. Logistic
model tree (LMT) is the best algorithm for Insomnia and Bipolar while Naïve Bayes (NB) is the best for
Schizophrenia, VD and MBD. Support vector machines (SVM) with ensemble learning and classification (ELC)
and Ensemble of Pruned Set (EPS) are the best classifiers for Bipolar while SVM with regression and threshold
(RT) is the least. The classifiers are statistically significantly different for Insomnia, VD and ADHD only.