Convolutional neural network based two-stage epileptic EEG classification system

Abstract

Epilepsy is a central nervous system disorder and the epileptic patients will exhibit ‘spikes’ characteristics in their brain wave. The objective of this project is to discover a better setup parameters of the Convolutional Neural Network (CNN) and techniques in improving the classification result of the automated epileptic patient classification system. Different architecture has been experimented on the system and the system was evaluated based on several metrics. Besides that, various of techniques such as background data swapping, early stopping, training with different background to spikes ratio and so on were also attempted in this project and the effect on the improvement of the system has been investigated. In conclusion, a better setup parameter and techniques has been discovered and achieve improvement to the classification accuracy of the system. Further improvement may include training with more patient data and use different techniques such as batch normalization and pseudo-labelling to improve the accuracy of the system.