Classification of First-Episode Psychosis with EEG Signals: ciSSA and Machine Learning Approach

Abstract

First-episode psychosis (FEP) typically marks the onset of severe psychiatric disorders and represents a critical period in the field of mental health. The early diagnosis of this condition is essential for timely intervention and improved clinical outcomes. In this study, the classification of FEP was investigated using the analysis of electroencephalography (EEG) signals and circulant spectrum analysis (ciSSA) sub-band signals. FEP poses a significant diagnostic challenge in the realm of mental health, and it is aimed at introducing a novel and effective approach for early diagnosis. To achieve this, the LASSO method was utilized to select the most significant features derived from entropy, frequency, and statistical-based characteristics obtained from ciSSA sub-band signals, as well as their hybrid combinations. Subsequently, a high-performance classification model has been developed using machine learning techniques, including ensemble, support vector machine (SVM), and artificial neural network (ANN) methods. The results of this study demonstrated that the hybrid features extracted from EEG signals’ ciSSA sub-bands, in combination with the SVM method, achieved a high level of performance, with an area under curve (AUC) of 0.9893, an accuracy of 96.23%, a sensitivity of 0.966, a specificity of 0.956, a precision of 0.9667, and an F1 score of 0.9666. This has revealed the effectiveness of the ciSSA-based method for classifying FEP from EEG signals.