CNN-LSTM-attention water quality prediction hybrid model

Water quality prediction is a fundamental task to mitigate water resource crisis and promote water ecological protection and restoration. The traditional LSTM water quality prediction model has the problems of weak generalization performance and low prediction accuracy. In this paper, we take the water quality parameters of Fairy Lake in Jiangxi Province as the research object and propose a hybrid CNN-LSTM-Attention water quality prediction model combining CNN, LSTM and attention mechanism. The model first extracts short-term abstract features of water quality data by CNN, then selects LSTM model to capture long-term dependencies between variables, and finally uses Attention mechanism to determine the importance of different temporal features and assign weights, output prediction results. After experimental validation, the CNN-LSTM-Attention water quality prediction hybrid model has smaller prediction error, and the RMSE and MAE evaluation indexes are better than LSTM prediction model and CNN-LSTM prediction mode.