An Advanced Pruning Method in the Architecture of Extreme Learning Machines using L1-regularization and Bootstrapping

Paulo De Campos Souza, Luiz Carlos Bambirra Torres, Gustavo Rodrigues Lacerda Silva, Antonio De Padua Braga, Edwin Lughofer

Research output: Contribution to journalArticlepeer-review

Abstract

Extreme learning machines (ELMs) are efficient for classification, regression, and time series prediction, as well as being a clear solution to backpropagation structures to determine values in intermediate layers of the learning model. One of the problems that an ELM may face is due to a large number of neurons in the hidden layer, making the expert model a specific data set. With a large number of neurons in the hidden layer, overfitting is more likely and thus unnecessary information can deterioriate the performance of the neural network. To solve this problem, a pruning method is proposed, called Pruning ELM Using Bootstrapped Lasso BR-ELM, which is based on regularization and resampling techniques, to select the most representative neurons for the model response. This method is based on an ensembled variant of Lasso (achieved through bootstrap replications) and aims to shrink the output weight parameters of the neurons to 0 as many and as much as possible. According to a subset of candidate regressors having significant coefficient values (greater than 0), it is possible to select the best neurons in the hidden layer of the ELM. Finally, pattern classification tests and benchmark regression tests of complex real-world problems are performed by comparing the proposed approach to other pruning models for ELMs. It can be seen that statistically BR-ELM can outperform several related state-of-the-art methods in terms of classification accuracies and model errors (while performing equally to Pruning-ELM P-ELM), and this with a significantly reduced number of finally selected neurons.
Original languageEnglish
Article number811
Pages (from-to)811
Number of pages24
JournalElectronics
Volume9
Issue number5
DOIs
Publication statusPublished - 2020

Fields of science

  • 101 Mathematics
  • 101013 Mathematical logic
  • 101024 Probability theory
  • 102001 Artificial intelligence
  • 102003 Image processing
  • 102019 Machine learning
  • 102035 Data science
  • 603109 Logic
  • 202027 Mechatronics

JKU Focus areas

  • Digital Transformation

Cite this