Causality-Structured Deep Learning for Soil Moisture Predictions

The accurate prediction of surface soil moisture (SM) is crucial for understanding hydrological processes. Deep learning (DL) models such as the long short-term memory model (LSTM) provide a powerful method and have been widely used in SM prediction. However, few studies have notably high success rates due to lacking prior knowledge in forms such as causality. Here we present a new causality-structure-based LSTM model (CLSTM), which could learn time interdependency and causality information for hydrometeorological applications. We applied and compared LSTM and CLSTM methods for forecasting SM across 64 FLUXNET sites globally. The results showed that CLSTM dramatically increased the predictive performance compared with LSTM. The Nash-Sutcliffe efficiency (NSE) suggested that more than 67% of sites witnessed an improvement of SM simulation larger than 10%. It is highlighted that CLSTM had a much better generalization ability that can adapt to extreme soil conditions, such as SM response to drought and precipitation events. By incorporating causal relations, CLSTM increased predictive ability across different lead times compared to LSTM. We also highlighted the critical role of physical information in the form of causality structure to improve drought prediction. At the same time, CLSTM has the potential to improve predictions of other hydrometeorological variables.