A coupled model to improve river water quality prediction towards addressing non-stationarity and data limitation.

Researchers

Journal

Modalities

Models

Abstract

Accurate predictions of river water quality are vital for sustainable water management. However, even the powerful deep learning model, i.e., long short-term memory (LSTM), has difficulty in accurately predicting water quality dynamics owing to the high non-stationarity and data limitation in a changing environment. To wiggle out of quagmires, wavelet analysis (WA) and transfer learning (TL) techniques were introduced in this study to assist LSTM modeling, termed WA-LSTM-TL. Total phosphorus, total nitrogen, ammonia nitrogen, and permanganate index were predicted in a 4 h step within 49 water quality monitoring sites in a coastal province of China. We selected suitable source domains for each target domain using an innovatively proposed regionalization approach that included 20 attributes to improve the prediction efficiency of WA-LSTM-TL. The coupled WA-LSTM facilitated capturing non-stationary patterns of water quality dynamics and improved the performance by 53 % during testing phase compared to conventional LSTM. The WA-LSTM-TL, aided by the knowledge of source domain, obtained a 17 % higher performance compared to locally trained WA-LSTM, and such improvement was more impressive when local data was limited (+66 %). The benefit of TL-based modeling diminished as data quantity increased; however, it outperformed locally direct modeling regardless of whether target domain data was limited or sufficient. This study demonstrates the reasoning for coupling WA and TL techniques with LSTM models and provides a newly coupled modeling approach for improving short-term prediction of river water quality from the perspectives of non-stationarity and data limitation.Copyright © 2023 Elsevier Ltd. All rights reserved.