STMF: A Spatiotemporal Multimodal Fusion Model for Long-Term Water Quality Forecasting

Water quality forecasting is a time series analysis task involving estimating future water conditions, vital in environmental management and pollution control. However, existing time series analysis methods focus only on historical observational data, neglecting information from other modalities, leading to incomplete feature extraction and affecting forecasting accuracy and robustness. In addition, the complex spatial dependencies between water quality monitoring stations and the nonlinear fluctuations in water quality indicators caused by meteorological factors present additional challenges. This work proposes a Spatio-Temporal Multimodal Fusion architecture for long-term water quality forecasting, named STMF, to address these issues. It first captures spatio-temporal dependencies by integrating temporal features with upstream-downstream relationships among monitoring stations. Then, STMF further designs a Low-rank Cross-modal Interaction Fusion (LRCIF) method, which fuses spatio-temporal features with precipitation features from the remote sensing image, as an additional modality, effectively leveraging complementary information from multiple data sources to enhance the accuracy and stability of water quality forecasting. Experimental results on real-world water quality datasets demonstrate that the proposed STMF significantly outperforms existing state-of-the-art methods in prediction accuracy. In particular, for long-term forecasting tasks with a 192-step horizon, STMF improves MSE and MAE by 14% and 12%, respectively, compared to unimodal models. It further validates the effectiveness of the multimodal fusion strategy. Overall, STMF offers an effective solution for water quality monitoring and management.