Published online Aug 31, 2017.
https://doi.org/10.4306/jknpa.2017.56.3.103
A Case Study of a Machine-Learning Approach in Differential Diagnosis of Schizophrenia: The Predictive Capacity of WAIS-IV
Abstract
Objectives
Machine learning (ML) encompasses a body of statistical approaches that can detect complex interaction patterns from multi-dimensional data. ML is gradually being adopted in medical science, for example, in treatment response prediction and diagnostic classification. Cognitive impairment is a prominent feature of schizophrenia, but is not routinely used in differential diagnosis. In this study, we investigated the predictive capacity of the Wechsler Adult Intelligence Scale IV (WAIS-IV) in differentiating schizophrenia from non-psychotic illnesses using the ML methodology. The purpose of this study was to illustrate the possibility of using ML as an aid in differential diagnosis.
Methods
The WAIS-IV test data for 434 psychiatric patients were curated from archived medical records. Using the final diagnoses based on DSM-IV as the target and the WAIS-IV scores as predictor variables, predictive diagnostic models were built using 1) linear 2) non-linear/non-parametric ML algorithms. The accuracy obtained was compared to that of the baseline model built without the WAIS-IV information.
Results
The performances of the various ML models were compared. The accuracy of the baseline model was 71.5%, but the best non-linear model showed an accuracy of 84.6%, which was significantly higher than that of non-informative random guessing (p=0.002). Overall, the models using the non-linear algorithms showed better accuracy than the linear ones.
Conclusion
The high performance of the developed models demonstrated the predictive capacity of the WAIS-IV and justified the application of ML in psychiatric diagnosis. However, the practical application of ML models may need refinement and larger-scale data collection.
Fig. 1
Receiver operation curve depicting the model performances of the best nonlinear model (radial basis function kernel support vector machine) and the best linear model (penalized logistic regression). RBF-SVM : Radial basis function kernel support vector machine, Lasso : Penalized logistic regression, AUC : Area under the curve.
Table 1
The final diagnoses of included subjects based on DSM-IV
Table 2
Demographic characteristics of the included subjects
Table 3
The WAIS-IV subtests, index scores and their comparison between schizophrenia group and non-psychotic illness group
Table 4
The performance of the diagnostic models built by different machine learning algorithms
Conflicts of Interest:The authors have no financial conflicts of interest.
References
-
Hastie T, Tibshirani R, Friedman J. In: The elements of statistical learning: data mining, inference, and prediction. 2nd ed. New York: Springer Science & Business Media; 2009.
-
-
Clarke B, Fokoue E, Zhang HH. In: Principles and theory for data mining and machine learning. New York: Springer Science & Business Media; 2009.
-
-
Zhou ZH. In: Ensemble methods: foundations and algorithms. Boca Raton: CRC Press; 2012.
-
-
Tomasik J, Schwarz E, Guest PC, Bahn S. Blood test for schizophrenia. Eur Arch Psychiatry Clin Neurosci 2012;262 Suppl 2:S79–S83.
-
-
Trivedi MH, Greer TL. Cognitive dysfunction in unipolar depression: implications for treatment. J Affect Disord 2014:152–154. 19–27.
-
-
Wechsler D. The psychometric tradition: developing the wechsler adult intelligence scale. Contemp Educ Psychol 1981;6:82–85.
-
-
Hwang ST, Kim JH, Park KB, Choi JY, Hong SH, editors. Standardization of the K-WAIS-IV; Proceedings of Korean Psychological Association Annual Conference; 2012 Aug 24; Chuncheon, Korea. Seoul: Korean Psychological Association; 2012.
-
-
R Development Core Team. R: a language and environment for statistical computing. 3.2.4 ed. Vienna: R Foundation for Statistical Computing; 2016.
-
-
Kuhn M. In: Caret: classification and regression training. 6.0-68 ed. R package version; 2016.
-
-
Harrison-Read P. IQ tests as aids to diagnosis and management in early schizophrenia. Adv Psychiatr Treat 2008;14:235–240.
-
-
Zielesny A. In: From curve fitting to machine learning: an illustrative guide to scientific data analysis and computational intelligence. 2nd ed. Switzerland: Springer International Publishing; 2016.
-