WaveDroughtNet: A Multi-Modal Wavelet-Enhanced Temporal Convolutional Network for Multi-Horizon Drought Forecasting and Onset Analysis

Drought is a slowly evolving, multi-driver hydro-meteorological hazard whose accurate early prediction is a cornerstone of climate-smart agriculture and water-resource planning. Existing data-driven drought forecasting frameworks suffer from three persistent limitations: (i) most models concatenate heterogeneous climate variables into a single flat feature vector, implicitly assuming a single dominant driver such as precipitation, even though atmospheric moisture demand, radiation and wind-mediated evapotranspiration co-determine drought onset; (ii) wavelet preprocessing is typically applied to the full series, introducing future-information leakage that violates the operational causality requirement of forecasting; and (iii) most architectures predict a single horizon and provide no causal attribution explaining when, where and which climatic variables initiated the event. This study proposes WaveDroughtNet, a multi-modal, multi-horizon deep-learning framework that addresses these limitations through five integrated components: (a) a strictly causal Daubechies-4 wavelet decomposition computed in a rolling fashion; (b) six modality-specific encoders with stochastic modality dropout (p = 0.15); (c) cross-modal multi-head attention with four heads; (d) a four-layer temporal convolutional network (TCN) backbone with dilation factors yielding a 240-step receptive field; and (e) a post hoc DroughtOriginTracer that combines temporal attention, modal-attribution and inter-district propagation scans. The Standardised Precipitation Evapotranspiration Index (SPEI), used as the supervisory target, is computed following the canonical Vicente-Serrano formulation. water balance D=P−PET (Hargreaves PET) at a 4-week (≈1-month) timescale, fitted with a three-parameter log-logistic distribution via L-moments, validated by Kolmogorov–Smirnov goodness-of-fit testing (α=0.05) per district, and standardised through the inverse-normal cumulative distribution function. Trained on 18,304 weekly district records from NASA POWER reanalysis (2014–2025) covering all 32 districts of Tamil Nadu, India, WaveDroughtNet uses only 256,869 parameters and produces, in a single forward pass, four forecasts (1 week, 1 month, 3 months, 1 year). On the held-out 2024 test partition (N=1728), the model attains weighted F1=0.9221 and R2=0.8512 at the 1-week horizon, and weighted F1=0.8498 and R2=0.6812 at the 1-year horizon. Diebold–Mariano tests confirm that WaveDroughtNet significantly outperforms naive persistence, seasonal naive, LSTM, ConvLSTM and a vanilla Transformer at the 3-month and 1-year horizons (p < 0.001). The DroughtOriginTracer successfully back-projects 15 Coimbatore events to causal origins 29–41 weeks prior to onset. We explicitly acknowledge three limitations that constrain operational deployment in its current form—zero severe events in the 2024 test partition (F1severe = 0.000), static inter-district modelling, and absence of vegetation-index supervision—and propose concrete mitigation pathways in the Discussion.