Psychiatry Investig.  2014 Jul;11(3):243-250.

Ant Colony Optimization Based Feature Selection Method for QEEG Data Classification

• Erguzel TT
• Ozekes S
• Gultekin S
• Tarhan N

Abstract

OBJECTIVE
Many applications such as biomedical signals require selecting a subset of the input features in order to represent the whole set of features. A feature selection algorithm has recently been proposed as a new approach for feature subset selection.
METHODS
Feature selection process using ant colony optimization (ACO) for 6 channel pre-treatment electroencephalogram (EEG) data from theta and delta frequency bands is combined with back propagation neural network (BPNN) classification method for 147 major depressive disorder (MDD) subjects.
RESULTS
BPNN classified R subjects with 91.83% overall accuracy and 95.55% subjects detection sensitivity. Area under ROC curve (AUC) value after feature selection increased from 0.8531 to 0.911. The features selected by the optimization algorithm were Fp1, Fp2, F7, F8, F3 for theta frequency band and eliminated 7 features from 12 to 5 feature subset.
CONCLUSION
ACO feature selection algorithm improves the classification accuracy of BPNN. Using other feature selection algorithms or classifiers to compare the performance for each approach is important to underline the validity and versatility of the designed combination.

Keyword

Neural networks; Feature selection; QEEG; Major depressive disorder

MeSH Terms

Ants*
Classification*
Depressive Disorder, Major
Electroencephalography
ROC Curve