Quantifying Uncertainty in Deep Spatiotemporal Forecasting
Dongxia Wu, Liyao Gao, Matteo Chinazzi, Xinyue Xiong, Alessandro Vespignani, Yi-An Ma, Rose Yu
Abstract
Deep learning is gaining increasing popularity for spatiotemporal forecasting. However, prior works have mostly focused on point estimates without quantifying the uncertainty of the predictions. In high stakes domains, being able to generate probabilistic forecasts with confidence intervals is critical to risk assessment and decision making. Hence, a systematic study of uncertainty quantification (UQ) methods for spatiotemporal forecasting is missing in the community. In this paper, we describe two types of spatiotemporal forecasting problems: regular grid-based and graph-based. Then we analyze UQ methods from both the Bayesian and the frequentist point of view, casting in a unified framework via statistical decision theory. Through extensive experiments on real-world road network traffic, epidemics, and air quality forecasting tasks, we reveal the statistical and computational trade-offs for different UQ methods: Bayesian methods are typically more robust in mean prediction, while confidence levels obtained from frequentist methods provide more extensive coverage over data variations. Computationally, quantile regression type methods are cheaper for a single confidence interval but require re-training for different intervals. Sampling based methods generate samples that can form multiple confidence intervals, albeit at a higher computational cost.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ea41c3d5-27f9-4796-8f9c-9a637102b3caCited by top-tier papers15
- DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal ForecastingSalva Rühling Cachay, Bo Zhao, Hailey Joren, Rose YuNeurIPS 2023 · 164 citations
- Spatial-Temporal Hypergraph Self-Supervised Learning for Crime PredictionZhonghang Li, Chao Huang, Lianghao Xia, Yong Xu et al.ICDE 2022 · 82 citations
- Copula Conformal prediction for multi-step time series predictionSophia Huiwen Sun, Rose YuICLR 2024 · 36 citations
- Uncertainty Quantification for Traffic Forecasting: A Unified ApproachWeizhu Qian, Dalin Zhang, Yan Zhao, Kai Zheng et al.ICDE 2023 · 27 citations
- Looks Too Good To Be True: An Information-Theoretic Analysis of Hallucinations in Generative Restoration ModelsRegev Cohen, Idan Kligvasser, Ehud Rivlin, Daniel FreedmanNeurIPS 2024 · 26 citations
Builds on6
- Bayesian Deep Learning and a Probabilistic Perspective of GeneralizationAndrew Gordon Wilson, Pavel IzmailovNeurIPS 2020 · 845 citations
- Why Gradient Clipping Accelerates Training: A Theoretical Justification for AdaptivityJingzhao Zhang, Tianxing He, Suvrit Sra, Ali JadbabaieICLR 2020 · 598 citations
- Efficient and Scalable Bayesian Neural Nets with Rank-1 FactorsMichael Dusenberry, Ghassen Jerfel, Yeming Wen, Yi-An Ma et al.ICML 2020 · 239 citations
- Quantifying Point-Prediction Uncertainty in Neural Networks via Residual Estimation with an I/O KernelXin Qiu, Elliot Meyerson, Risto MiikkulainenICLR 2020 · 60 citations
- Preserving Dynamic Attention for Long-Term Spatial-Temporal PredictionHaoxing Lin, Rufan Bai, Weijia Jia, Xinyu Yang et al.KDD 2020 · 56 citations
Related papers
- RNN with Particle Flow for Probabilistic Spatio-temporal ForecastingSoumyasundar Pal, Liheng Ma, Yingxue Zhang, Mark CoatesICML 2021 · 26 citations
- Relational Conformal Prediction for Correlated Time SeriesAndrea Cini, Alexander Jenkins, Danilo P. Mandic, Cesare Alippi et al.ICML 2025
- STUaNet: Understanding Uncertainty in Spatiotemporal Collective Human MobilityZhengyang Zhou, Yang Wang, Xike Xie, Lei Qiao et al.WWW 2021 · 29 citations
- When in Doubt: Neural Non-Parametric Uncertainty Quantification for Epidemic ForecastingHarshavardhan Kamarthi, Lingkai Kong, Alexander Rodríguez, Chao Zhang et al.NeurIPS 2021 · 26 citations
- DutyTTE: Deciphering Uncertainty in Origin-Destination Travel Time EstimationXiaowei Mao, Yan Lin, Shengnan Guo, Yubin Chen et al.AAAI 2025 · 4 citations
