Login
Section Computer Science

EEG Schizophrenia Classification with Comparison of Three Machine Learning Algorithms

Klasifikasi Skizofrenia EEG dengan Perbandingan Tiga Algoritma Machine Learning
Vol. 10 No. 2 (2025): December:

Elizabeth Juli Angelina Saragi (1), Dafid Riswanto Zebua (2), Sekarayu Larasati (3), Ravi Telaumbanua (4), Dhanny Rukmana Manday (5)

(1) Program Studi Teknik Informatika, Universitas Prima Indonesia, Indonesia
(2) Program Studi Teknik Informatika, Universitas Prima Indonesia, Indonesia
(3) Program Studi Teknik Informatika, Universitas Prima Indonesia, Indonesia
(4) Program Studi Teknik Informatika, Universitas Prima Indonesia, Indonesia
(5) Program Studi Teknik Informatika, Universitas Prima Indonesia, Indonesia
Fulltext View | Download

Abstract:

General Background: Schizophrenia is a chronic mental disorder affecting millions globally, requiring improved diagnostic methods. Specific Background: EEG signals have emerged as promising biomarkers for schizophrenia classification through machine learning. Knowledge Gap: Despite prior advances, no systematic comparison of key machine learning algorithms—Logistic Regression, Random Forest, and Decision Tree—using EEG data for schizophrenia classification has been conducted. Aims: This study aims to compare the performance of these three algorithms in classifying schizophrenia from EEG signals using a dataset of 1932 samples. Results: Random Forest achieved the highest classification accuracy (86%) and AUC (0.912), outperforming Logistic Regression (accuracy 82%, AUC 0.865) and Decision Tree (accuracy 81%, AUC 0.871). Novelty: Unlike previous studies, this research provides a comprehensive algorithmic comparison using EEG-derived features, integrating feature importance, calibration, learning curves, and statistical tests. Implications: The findings establish Random Forest as a robust classifier for EEG-based schizophrenia detection, offering a foundation for developing clinically relevant, cloud-based diagnostic support tools that can facilitate early detection and personalized treatment planning in mental health care.
Highlight :




  • Random Forest achieved the highest accuracy and AUC in schizophrenia classification.




  • EEG data were processed using STFT, Wavelet Transform, and Band Power features.




  • Comparison of three algorithms offers a systematic basis for clinical application.




Keywords : Machine Learning Classification, Random Forest, Logistic Regression, Decision Tree, Learning Curve


 

Downloads

Download data is not yet available.

References

World Health Organization, “Schizophrenia,” World Health Organization, [Online]. Available: [https://www.who.int/news-room/fact-sheets/detail/schizophrenia] [Accessed: Jul. 11, 2025].

J. Rahul, D. Sharma, L. D. Sharma, U. Nanda, and A. K. Sarkar, “A Systematic Review of EEG Based Automated Schizophrenia Classification Through Machine Learning and Deep Learning,” Frontiers in Human Neuroscience, vol. 18, 2024. doi: [https://doi.org/10.3389/fnhum.2024.1347082]

C. Wang et al., “Multi-Feature Fusion RFE Random Forest for Schizophrenia Classification and Treatment Response Prediction,” Scientific Reports, vol. 15, 2025. doi: [https://doi.org/10.1038/s41598-025-89359-5]

S. G. Alonso, G. Marques, D. Agarwal, I. D. L. T. Díez, and M. F. Martín, “Comparison of Machine Learning Algorithms in the Prediction of Hospitalized Patients with Schizophrenia,” Sensors, vol. 22, no. 6, p. 2517, 2022. doi: [https://doi.org/10.3390/s22072517]

K. Kim, N. T. Duc, M. Coi, and B. Lee, “EEG Microstate Features for Schizophrenia Classification,” PLOS ONE, vol. 16, no. 5, 2021. doi: [https://doi.org/10.1371/journal.pone.0251842]

X. Li et al., “TMS-EEG Research to Elucidate the Pathophysiological Neural Bases in Patients with Schizophrenia: A Systematic Review,” Journal of Personalized Medicine, vol. 11, no. 5, 2021. doi: [https://doi.org/10.3390/jpm11050388]

M. A. Vázquez, A. Maghsoudi, and I. P. Mariño, “An Interpretable Machine Learning Method for the Detection of Schizophrenia Using EEG Signals,” Frontiers in Systems Neuroscience, vol. 15, 2021. doi: [https://doi.org/10.3389/fnsys.2021.652662]

Y. Wu et al., “Detection of Functional and Structural Brain Alterations in Female Schizophrenia Using Elastic Net Logistic Regression,” Brain Imaging and Behavior, vol. 16, no. 1, 2022. doi: [https://doi.org/10.1007/s11682-021-00501-z]

Y. Kang et al., “High-Order Brain Network Feature Extraction and Classification Method of First-Episode Schizophrenia: An EEG Study,” Frontiers in Human Neuroscience, vol. 18, no. 1, 2024. doi: [https://doi.org/10.3389/fnhum.2024.1452197]

S. R. Marder and D. Umbricht, “Negative Symptoms in Schizophrenia: Newly Emerging Measurements, Pathways, and Treatments,” Schizophrenia Research, vol. 258, pp. 71–77, 2023. doi: [https://doi.org/10.1016/j.schres.2023.07.010]

M. Marchi et al., “Machine-Learning for Prescription Patterns: Random Forest in the Prediction of Dose and Number of Antipsychotics Prescribed to People with Schizophrenia,” Clinical Psychopharmacology and Neuroscience, vol. 20, no. 3, pp. 450–461, 2022. doi: [https://doi.org/10.9758/cpn.2022.20.3.450].

A. Hudon et al., “Enhancing Predictive Power: Integrating a Linear Support Vector Classifier with Logistic Regression for Patient Outcome Prognosis in Virtual Reality Therapy for Treatment-Resistant Schizophrenia,” Journal of Personalized Medicine, vol. 13, no. 12, 2023. doi: [https://doi.org/10.3390/jpm13121660]

M. Léger et al., “The Mood Disorder Spectrum vs. Schizophrenia Decision Tree: EDIPHAS Research into the Childhood and Adolescence of 205 Patients,” BMC Psychiatry, vol. 22, no. 1, 2022. doi: [https://doi.org/10.1186/s12888-022-03835-0]

A. Parola et al., “Multimodal Assessment of Communicative-Pragmatic Features in Schizophrenia: A Machine Learning Approach,” Schizophrenia, vol. 7, no. 1, 2021. doi: [https://doi.org/10.1038/s41537-021-00153-4]

D. Gordillo et al., “The EEG Multiverse of Schizophrenia,” Cerebral Cortex, vol. 33, no. 7, pp. 3816–3826, 2023. doi: [https://doi.org/10.1093/cercor/bhac309]

A. Zaidi and A. S. M. A. Luhayb, “Two Statistical Approaches to Justify the Use of the Logistic Function in Binary Logistic Regression,” Mathematical Problems in Engineering, vol. 2023, no. 1, 2023. doi: [https://doi.org/10.1155/2023/5525675]

M. B. Antor et al., “A Comparative Analysis of Machine Learning Algorithms to Predict Alzheimer's Disease,” Journal of Healthcare Engineering, vol. 2021, 2021. doi: [https://doi.org/10.1155/2021/9917919].

Y. Salunke et al., “Fraud Detection: A Hybrid Approach with Logistic Regression, Decision Tree, and Random Forest,” Cureus Journal of Computer Science, 2025. doi: [https://doi.org/10.7759/s44389-024-02350-5]

H. Blockeel et al., “Decision Trees: From Efficient Prediction to Responsible AI,” Frontiers in Artificial Intelligence, vol. 6, 2023. doi: [https://doi.org/10.3389/frai.2023.1124553]

Z. M. Pirouz et al., “Development of Decision Tree Classification Algorithms in Predicting Mortality of COVID-19 Patients,” International Journal of Emergency Medicine, vol. 17, no. 1, 2024. doi: [https://doi.org/10.1186/s12245-024-00681-7]

S.-J. Lin et al., “Prediction Models Using Decision Tree and Logistic Regression Method for Predicting Hospital Revisits in Peritoneal Dialysis Patients,” Diagnostics, vol. 14, no. 6, 2024. doi: [https://doi.org/10.3390/diagnostics14060620]

A. N. S. Kinasih et al., “Comparative Analysis of Decision Tree and Random Forest Classifiers for Structured Data Classification in Machine Learning,” Science in Information Technology Letters, vol. 5, no. 2, pp. 13–24, 2024. doi: [http://dx.doi.org/10.31763/sitech.v5i2.1746]

S. A. A. Balabied and H. F. Eid, “Utilizing Random Forest Algorithm for Early Detection of Academic Underperformance in Open Learning Environments,” PeerJ Computer Science, 2023. doi: [https://doi.org/10.7717/peerj-cs.1708]

E. Cardon et al., “Random Forest Classification to Predict Response to High-Definition Transcranial Direct Current Stimulation for Tinnitus Relief: A Preliminary Feasibility Study,” Audiology & Neurotology, vol. 43, no. 6, pp. 1816–1823, 2022. doi: [https://doi.org/10.1097/aud.0000000000001246]

F. Chien, V. L. Kouznetsova, S. Kesari, and I. F. Tsigelny, “Electroencephalography-Based Diagnosis of Schizophrenia Using Machine Learning,” Cerebral Cortex, vol. 35, no. 7, 2025. doi: [https://doi.org/10.1093/cercor/bhaf184]

M. Abrar et al., “Enhanced Schizophrenia Detection Using Multichannel EEG and CAOA-RST-Based Feature Selection,” Scientific Reports, vol. 15, 2025. doi: [https://doi.org/10.1038/s41598-025-05028-7]

H.-H. Hwang et al., “Machine Learning-Based Differentiation of Schizophrenia and Bipolar Disorder Using Multiscale Fuzzy Entropy and Relative Power from Resting-State EEG,” Translational Psychiatry, vol. 15, 2025. doi: [https://doi.org/10.1038/s41398-025-03354-y]