Title :
Metaheuristics for feature selection: Application to sepsis outcome prediction
Author :
Vieira, Susana M. ; Mendonça, Luis F. ; Farinha, Gonçalo J. ; Sousa, João M C
Author_Institution :
Dept. of Mech. Eng., Tech. Univ. of Lisbon, Lisbon, Portugal
Abstract :
This paper proposes the application of a new binary particle swarm optimization (BPSO) method to feature selection problems. Two enhanced versions of binary particle swarm optimization, designed to cope with premature convergence of the BPSO algorithm, are proposed. These methods control the swarm variability using the velocity and the similarity between best swarm solutions. The proposed PSO methods use neural networks, fuzzy models and support vector machines in a wrapper approach, and are tested in a benchmark database. It was shown that the proposed BPSO approaches require an inferior simulation time, less selected features and increase accuracy. The best BPSO is then compared with genetic algorithms (GA) and applied to a real medical application, a sepsis patient database. The objective is to predict the outcome (survived or deceased) of the sepsis patients. It was shown that the proposed BPSO approaches are similar in terms of model accuracy when compared to GA, while requiring an inferior simulation time and less selected features.
Keywords :
data mining; feature extraction; fuzzy set theory; medical information systems; neural nets; particle swarm optimisation; support vector machines; BPSO method; binary particle swarm optimization method; feature selection metaheuristics; feature selection problems; fuzzy models; genetic algorithms; inferior simulation time; medical application; neural networks; sepsis outcome prediction; sepsis patient database; support vector machines; swarm variability; wrapper approach; Accuracy; Databases; Feature extraction; Genetic algorithms; Neurons; Support vector machines; Vectors;
Conference_Titel :
Evolutionary Computation (CEC), 2012 IEEE Congress on
Conference_Location :
Brisbane, QLD
Print_ISBN :
978-1-4673-1510-4
Electronic_ISBN :
978-1-4673-1508-1
DOI :
10.1109/CEC.2012.6256651