Explainable Quantile CNN-LSTM model for uncertainty-aware multi-layer soil moisture prediction in tropical cocoa plantations

Citation

Shawon, Sarowar Morshed and Zaman, Mukter and Maniam, Shamala and Al Qwaid, Marran and Sarker, Md Tanjil and Kheng, Tee Yei and Wong, Hin Yong (2026) Explainable Quantile CNN-LSTM model for uncertainty-aware multi-layer soil moisture prediction in tropical cocoa plantations. Scientific Reports, 16 (1). ISSN 2045-2322

[img] Text
s41598-026-48517-z.pdf - Published Version
Restricted to Repository staff only

Download (7MB)

Abstract

Accurate prediction of multi-layer soil moisture within the root zone is critical for cocoa plantations, where water availability directly influences root development, nutrient uptake, flowering and yield stability. In tropical systems, strong rainfall variability, heterogeneous soils, and delayed subsurface responses make depth-resolved moisture forecasting particularly challenging. This study proposes an improved Quantile Convolutional Neural Networks (CNN)-Long Short-Term Memory (LSTM) framework for robust and interpretable multi-layer soil moisture prediction. The model integrates CNN for localized temporal feature extraction with stacked LSTM long short-term memory networks for sequential dependency modelling, while incorporating quantile regression to provide probabilistic forecasts. Autocorrelation-guided lag optimization identified lag-7 temporal window as optimal. Zone 1 data were used exclusively for training and validation, whereas Zones 2 and 3 were reserved for independent testing to ensure spatial generalization and robustness. The proposed model achieved consistently high predictive accuracy across five soil depths (5–105 cm), with an overall average R2 of 0.948, low RMSE (0.39–0.79 across layers), and MAPE generally below 3%. Independent testing in Zones 2 and 3 demonstrated minimal performance degradation, confirming strong transferability under varying field conditions. For uncertainty quantification, quantile regression produced reliable 50 and 80% prediction intervals. Low mean pinball loss values, Prediction Interval Coverage Probability (PICP) close to nominal levels, moderate Mean Prediction Interval Width (MPIW), and favorable Winkler scores indicate well-calibrated and sharp probabilistic forecasts. Explainability analysis using SHapley Additive exPlanations (SHAP) and Integrated Gradients Attribution (IGA) revealed coherent depth-dependent climatic controls, transitioning from short-term atmospheric drivers in shallow layers to slower-varying seasonal influences in deeper horizons. Overall, this research delivers a robust, uncertainty-aware, and interpretable quantile deep learning framework for multi-layer soil moisture forecasting, supporting smart irrigation and climate-adaptive water management in tropical precision agriculture.

Item Type: Article
Uncontrolled Keywords: eXplainable AI, soil
Subjects: T Technology > TA Engineering (General). Civil engineering (General) > TA703-712 Engineering geology. Rock mechanics. Soil mechanics. Underground construction
Divisions: Faculty of Artificial Intelligence & Engineering (FAIE)
Depositing User: Ms Rosnani Abd Wahab
Date Deposited: 30 Jun 2026 07:15
Last Modified: 30 Jun 2026 07:15
URII: http://shdl.mmu.edu.my/id/eprint/16150

Downloads

Downloads per month over past year

View ItemEdit (login required)