Evolutionary Wavelet Neural Network ensembles for breast cancer and Parkinson's disease prediction

PLoS One. 2018 Feb 8;13(2):e0192192. doi: 10.1371/journal.pone.0192192. eCollection 2018.

Abstract

Wavelet Neural Networks are a combination of neural networks and wavelets and have been mostly used in the area of time-series prediction and control. Recently, Evolutionary Wavelet Neural Networks have been employed to develop cancer prediction models. The present study proposes to use ensembles of Evolutionary Wavelet Neural Networks. The search for a high quality ensemble is directed by a fitness function that incorporates the accuracy of the classifiers both independently and as part of the ensemble itself. The ensemble approach is tested on three publicly available biomedical benchmark datasets, one on Breast Cancer and two on Parkinson's disease, using a 10-fold cross-validation strategy. Our experimental results show that, for the first dataset, the performance was similar to previous studies reported in literature. On the second dataset, the Evolutionary Wavelet Neural Network ensembles performed better than all previous methods. The third dataset is relatively new and this study is the first to report benchmark results.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Breast Neoplasms / diagnostic imaging
  • Breast Neoplasms / pathology*
  • Female
  • Humans
  • Male
  • Mammography
  • Neural Networks, Computer*

Grants and funding

The first author would like to acknowledge the support through an Australian Government Research Training Program Scholarship.