EEG Signals Classification Using Machine Learning for The Identification and Diagnosis of Schizophrenia.

Journal: Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
Published Date:

Abstract

This paper presents the design of a machine learning-based classifier for the differentiation between Schizophrenia patients and healthy controls using features extracted from electroencephalograph(EEG) signals based on event related potential(ERP). A number of features are extracted from an online EEG dataset with 81 subjects, including 32 healthy controls and 49 Schizophrenia patients. The EEG signals are preprocessed and since the dataset is relatively small, the random forest machine learning algorithm is chosen to be applied on different combinations of feature sets for classification. It is found that the classification accuracy can be improved by adding certain features extracted from EEG signals.

Authors

  • Lei Zhang
    Division of Gastroenterology, Union Hospital, Tongji Medical College Medical College, Huazhong University of Science and Technology, Wuhan, China.