Drought prediction based on an improved VMD-OS-QR-ELM model

PLoS One. 2022 Jan 6;17(1):e0262329. doi: 10.1371/journal.pone.0262329. eCollection 2022.

Abstract

To overcome the low accuracy, poor reliability, and delay in the current drought prediction models, we propose a new extreme learning machine (ELM) based on an improved variational mode decomposition (VMD). The model first redefines the output of the hidden layer of the ELM model with orthogonal triangular matrix decomposition (QR) to construct an orthogonal triangular ELM (QR-ELM), and then introduces an online sequence learning mechanism (OS) into the QR-ELM to construct an online sequence OR-ELM (OS-QR-ELM), which effectively improves the efficiency of the ELM model. The mutual information extension method was then used to extend both ends of the original signal to improve the VMD end effect. Finally, VMD and OS-QR-ELM were combined to construct a drought prediction method based on the VMD-OS-QR-ELM. The reliability and accuracy of the VMD-OS-QR-ELM model were improved by 86.19% and 93.20%, respectively, compared with those of the support vector regression model combined with empirical mode decomposition. Furthermore, the calculation efficiency of the OS-QR-ELM model was increased by 88.65% and 85.32% compared with that of the ELM and QR-ELM models, respectively.

Publication types

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

MeSH terms

  • Algorithms
  • Droughts / prevention & control*
  • Machine Learning
  • Neural Networks, Computer
  • Reproducibility of Results

Grants and funding

This work was supported in part by the National Key Research and Development Project under Grant Strategic Research Projects in Key Area 16 and the Water Conservancy Science and Technology Research Project in Henan Province Grant GG202042.