Framework for interpretable Bayesian soil moisture modelling

Abstract

: Soil moisture is a key driver of hydrological processes such as flooding during wet periods and vegetation growth in dry periods, but is highly variable in time and space. Soil moisture varies due to landcover, soil type, landscape position, and rainfall input. Disentangling the different drivers and spatial relationships in soil moisture are important to deliver forecasts. To facilitate this, the most widely-used approach to simulating soil moisture is numerical modelling. More recent approaches using neural networks display good predictive performance, but lose interpretability. In contrast, numerical models are often too rigid, and quantifying uncertainty can be difficult and computationally expensive. The objective of this paper is to demonstrate a Bayesian modelling framework that is flexible, quantifies uncertainties, disentangles the relative importance of different drivers, and is able to make forecasts. The data are derived from a dense soil moisture observation network at Llara farm in Narrabri (NSW), installed as part of a landscape rehydration project. This project consists of two 40 ha sites, each with control areas and treatment areas. The treatment involves the installation of contour banks with 1 m elevation to reduce over-land flow velocities and increase infiltration. Given their spatial distribution, each of the gauges has varying topological features and soil characteristics that may also influence the relationship of soil moisture to forcing variables (e.g., rainfall and evaptranspiration). For this modelling, we focused on 16 months of 10 minute interval data across 32 gauges at a common depth below the surface (200 mm). These data were aggregated to a daily mean to align with rainfall data from a nearby rain gauge at Llara farm and evapotranspiration data downloaded from SILO. The model is a Hierarchical Linear Model (HLM) fit using a Bayesian approach leveraging the No-U-Turn MCMC sampler implemented in NumPyro. HLMs are well suited to modelling nested data, such as in this case where gauges can be grouped by site and treatment, alongside gauge specific factors. Soil moisture at time t ( SM t ) is modelled at each gauge as a simple linear regression of the rainfall ( R ), evapotranspiration ( E ), and an autoregressive term (the soil moisture from the previous timestep, SM t − 1 ). The corresponding coefficients of the linear regression are β R , β E and β AR , respectively, along with an intercept term ( β 0 ). By introducing group level parameters via hierarchical priors on our β terms, we can explore the effects of site and treatment on the dynamics of soil moisture over time.