Optimizing Number of Hidden Neurons in Neural Networks

Y. Liu, J.A. Starzyk, and Z. Zhu (USA)

Keywords

Neural network, network optimization, function approximation, overfitting, signaltonoise ratio figure

Abstract

In this paper, a novel and effective criterion based on the estimation of the signal-to-noise-ratio figure (SNRF) is proposed to optimize the number of hidden neurons in neural networks to avoid overfitting in the function approximation. SNRF can quantitatively measure the useful information left unlearned so that overfitting can be automatically detected from the training error only without use of a separate validation set. It is illustrated by optimizing the number of hidden neurons in a multi-layer perceptron (MLP) using benchmark datasets. The criterion can be further utilized in the optimization of other parameters of neural networks when overfitting needs to be considered.

Important Links:



Go Back