1. One Improved Wasserstein GAN with Gradient Penalty for Grain Consumption Prediction.
- Author
-
Pei Li and Chunhua Zhu
- Subjects
CONVOLUTIONAL neural networks ,GENERATIVE adversarial networks ,FEATURE extraction ,SUPPLY & demand - Abstract
Prediction of grain consumption is crucial for analyzing the changing trend and balancing the grain supply and demand in China. Recently, the use of generative adversarial networks (GAN) to capture the distribution of historical data for generating future data has gained attention in time-series prediction. In order to enhance prediction performance and address model instability, an improved Wasserstein GAN with gradient penalty, referred to as IWGAN-GP, is proposed. The IWGAN-GP utilizes a bidirectional long short-term memory neural network (BiLSTM) as the generator and a convolutional neural network (CNN) as the discriminator, combining the memory capabilities of LSTM with the nonlinear feature extraction capabilities of CNN. Specifically, the loss function of the generator incorporates the mean square error (MSE) between real and generated samples to optimize the LSTM network, while the loss function of the discriminator includes the L
1 norm as the gradient penalty term to enhance sparsity and robustness, in contrast to the L2 norm used in existing WGAN-GP models. Experimental results on grain consumption data from 1981 to 2020 demonstrate that the proposed IWGAN-GP improves prediction accuracy compared to BiLSTM, GAN, and WGAN-GP models. [ABSTRACT FROM AUTHOR]- Published
- 2024