DocumentCode
2746076
Title
Effective neural network pruning using cross-validation
Author
Huynh, Thuan Q. ; Setiono, Rudy
Author_Institution
Sch. of Comput., Nat. Univ. of Singapore, Singapore
Volume
2
fYear
2005
fDate
31 July-4 Aug. 2005
Firstpage
972
Abstract
This paper addresses the problem of finding neural networks with optimal topology such that their generalization capability is maximized. Our approach is to combine the use of a penalty function during network training and a subset of the training samples for cross-validation. The penalty is added to the error function so that the weights of network connections that are not useful have small magnitude. Such network connections can be pruned if the resulting accuracy of the network does not change beyond a preset level. Training samples in the cross-validation set are used to indicate when network pruning is terminated. Our results on 32 publicly available data sets show that the proposed method outperforms existing neural network and decision tree methods for classification.
Keywords
learning (artificial intelligence); neural nets; decision tree methods; error function; network training; neural network pruning; optimal topology; penalty function; Backpropagation algorithms; Classification tree analysis; Computer networks; Decision trees; Electronic mail; Feedforward neural networks; Heuristic algorithms; Multi-layer neural network; Network topology; Neural networks;
fLanguage
English
Publisher
ieee
Conference_Titel
Neural Networks, 2005. IJCNN '05. Proceedings. 2005 IEEE International Joint Conference on
Print_ISBN
0-7803-9048-2
Type
conf
DOI
10.1109/IJCNN.2005.1555984
Filename
1555984
Link To Document