Skip to content

bruAristimunha/Re-Deep-Convolution-Neural-Network-and-Autoencoders-Based-Unsupervised-Feature-Learning-of-EEG

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Open In Colab Code style: black

Reproduction authors.

We have no affiliation with the original authors and our attempts to contact them have failed.

Bruno Aristimunha1, Diogo Eduardo Lima Alves1, Walter Hugo Lopez Pinaya 1,2, Raphael Y. de Camargo 1

1 Center for Mathematics, Computation and Cognition (CMCC), Federal Univesity of ABC (UFABC), Rua Arcturus, 03. Jardim Antares, São Bernardo do Campo, CEP 09606-070, SP, Brazil.

2 King’s College London, London, UK.

*[email protected], [email protected]


Original paper authors.

Tingxi Wen 2, Zhongnan Zhang* 2

2 Software School, Xiamen University, Xiamen, China.

*[email protected]

Abstract

This paper presents our efforts to reproduce and improve the results achieved by the authors of the original article. We follow the steps and models described in their article and the same public data sets of EEG Signals. Epilepsy affects more than 65 million people globally, and EEG Signals are critical to analyze and recognize epilepsy. Although the efforts in the last years, it is still challenging to extract useful information from these signals and select useful features in a diagnostic application. We construct a deep convolution network and autoencoders-based model (AE-CDNN) in order to perform unsupervised feature learning. We use the AE-CDNN to extract the features of the available data sets, and then we use some common classifiers to classify the features. The results obtained demonstrate that the proposed AE-CDNN outperforms the traditional feature extraction based classification techniques by achieving better accuracy of classification.

Prerequisites for Reprodubility

Clone the repository and the branch:

git clone --recurse-submodules -j8 https://github.com/bruAristimunha/Re-Deep-Convolution-Neural-Network-and-Autoencoders-Based-Unsupervised-Feature-Learning-of-EEG.git ReScience-submission

Install Conda, we recommend the tutorial.

Create a conda enviromnment as follows:

conda create --name eeg python=3.7 pip --yes
conda activate eeg

We recommend using a GPU, if you have it available. But if not, just ignore the first line below:

conda install -c anaconda tensorflow-gpu
y
pip install -r ReScience-submission/requirements.txt
cd ReScience-submission/notebook
jupyter notebook

Execute

Use Jupyter to open the notebook "Jupyter_Paper_Re_Deep_Convolution_Neural_Network_and_Autoencoders_Based_Unsupervised_Feature_Learning_of_EEG_Signals.ipynb" to reproduce the results