Modified binary PSO for feature selection using SVM applied to mortality prediction of septic patients |
| |
Authors: | Susana M. Vieira Luís F. Mendonça Gonçalo J. Farinha João M.C. Sousa |
| |
Affiliation: | 1. Technical University of Lisbon, Instituto Superior Técnico, Department of Mechanical Engineering, Center of Intelligent Systems, IDMEC Av. Rovisco Pais, 1049-001 Lisbon, Portugal;2. Escola Superior Náutica Infante D. Henrique, Department of Marine Engineering, Lisbon, Portugal |
| |
Abstract: | This paper proposes a modified binary particle swarm optimization (MBPSO) method for feature selection with the simultaneous optimization of SVM kernel parameter setting, applied to mortality prediction in septic patients. An enhanced version of binary particle swarm optimization, designed to cope with premature convergence of the BPSO algorithm is proposed. MBPSO control the swarm variability using the velocity and the similarity between best swarm solutions. This paper uses support vector machines in a wrapper approach, where the kernel parameters are optimized at the same time. The approach is applied to predict the outcome (survived or deceased) of patients with septic shock. Further, MBPSO is tested in several benchmark datasets and is compared with other PSO based algorithms and genetic algorithms (GA). The experimental results showed that the proposed approach can correctly select the discriminating input features and also achieve high classification accuracy, specially when compared to other PSO based algorithms. When compared to GA, MBPSO is similar in terms of accuracy, but the subset solutions have less selected features. |
| |
Keywords: | Feature selection Wrapper methods Particle swarm optimization Premature convergence Sepsis Support vector machines |
本文献已被 ScienceDirect 等数据库收录! |
|