A Conjugate Gradient-Based Efficient Algorithm for Training Single-Hidden-Layer Neural Networks

作者:Gong Xiaoling*; Wang Jian; Wang Yanjiang; Zurada Jacek M
来源:23rd International Conference on Neural Information Processing (ICONIP), 2016-10-16 to 2016-10-21.
DOI:10.1007/978-3-319-46681-1_56

摘要

A single hidden layer neural networks (SHLNNs) learning algorithm has been proposed which is called Extreme Learning Machine (ELM). It shows extremely faster than typical back propagation (BP) neural networks based on gradient descent method. However, it requires many more hidden neurons than BP neural networks to achieve assortive classification accuracy. This then leads more test time which plays an important role in practice. A novel learning algorithm (USA) for SHLNNs has been presented which updates the weights by using gradient method in the ELM framework. In this paper, we employ the conjugate gradient method to train the SHLNNs on the MNIST digit recognition problem. The simulated experiment demonstrates the better generalization and less required hidden neurons than the common ELM and USA.

全文