An improved extreme learning machine based on Variable-length Particle Swarm Optimization

Abstract

Extreme Learning Machine (ELM) for Single-hidden Layer Feedforward Neural Network (SLFN) has been attracting attentions because of its faster learning speed and better generalization performance than those of the traditional gradient-based learning algorithms. However, it has been proven that generalization performance of ELM classifier depends critically on the number of hidden neurons and the random determination of the input weights and hidden biases. In this paper, we propose Variable-length Particle Swarm Optimization algorithm (VPSO) for ELM to automatically select the number of hidden neurons as well as corresponding input weights and hidden biases for maximizing ELM classifier's generalization performance. Experimental results have verified that the proposed VPSO-ELM scheme significantly improves the testing accuracy of classification problems.