Uncertainty Quantification for Image-based Traffic Prediction across Cities
Not on the map. Why?
Abstract (arXiv)
Despite the strong predictive performance of deep learning models for traffic prediction, their widespread deployment in real-world intelligent transportation systems has been restrained by a lack of interpretability. Uncertainty quantification (UQ) methods provide an approach to induce probabilistic reasoning, improve decision-making and enhance model deployment potential. To gain a comprehensive picture of the usefulness of existing UQ methods for traffic prediction and the relation between obtained uncertainties and city-wide traffic dynamics, we investigate their application to a large-scale image-based traffic dataset spanning multiple cities and time periods. We compare two epistemic and two aleatoric UQ methods on both temporal and spatio-temporal transfer tasks, and find that meaningful uncertainty estimates can be recovered. We further demonstrate how uncertainty estimates can be employed for unsupervised outlier detection on changes in city traffic dynamics. We find that our approach can capture both temporal and spatial effects on traffic behaviour in a representative case study for the city of Moscow. Our work presents a further step towards boosting uncertainty awareness in traffic prediction tasks, and aims to highlight the value contribution of UQ methods to a better understanding of city traffic dynamics.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 55 of its 132 references that are in the map. First posted on arXiv 11 Aug 2023.
Builds on
55 of 132 references.
References · 55
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
- 2020A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challengescited by 813
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2019UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentationcited by 583
38 more
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2020BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learningcited by 368
- 2017Snapshot Ensembles: Train 1, get M for freecited by 349
- 2018Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networkscited by 278
- 2017Concrete Dropoutcited by 227
- 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 227
- 2019Evaluating Scalable Bayesian Deep Learning Methods for Robust Computer Visioncited by 182
- 2020U-Net and its variants for medical image segmentation: theory and applicationscited by 169
- 2018Understanding Batch Normalizationcited by 157
- 2020Hyperparameter Ensembles for Robustness and Uncertainty Quantificationcited by 149
- 2017Batch Renormalization: Towards Reducing Minibatch Dependence in Batch-Normalized Modelscited by 127
- 2020Better Aggregation in Test-Time Augmentationcited by 121
- 2019A General Framework for Uncertainty Estimation in Deep Learningcited by 119
- 2019DropConnect Is Effective in Modeling Uncertainty of Bayesian Deep Networkscited by 92
- 2019Evaluating and Calibrating Uncertainty Prediction in Regression Taskscited by 85
- 2020Bayesian Deep Learning via Subnetwork Inferencecited by 85
- 2021Uncertainty Baselines: Benchmarks for Uncertainty & Robustness in Deep Learningcited by 81
- 2018Lightweight Probabilistic Deep Networkscited by 77
- 2019Sampling-free Epistemic Uncertainty Estimation Using Approximated Variance Propagationcited by 74
- 2018Bayesian Uncertainty Estimation for Batch Normalized Deep Networkscited by 70
- 2018Safe Reinforcement Learning with Model Uncertainty Estimatescited by 70
- 2022Deep Ensembles Work, But Are They Necessary?cited by 63
- 2019STG2Seq: Spatial-temporal Graph to Sequence Model for Multi-step Passenger Demand Forecastingcited by 59
- 2021Uncertainty Toolbox: an Open-Source Library for Assessing, Visualizing, and Improving Uncertainty Quantificationcited by 57
- 2021Quantifying Uncertainty in Deep Spatiotemporal Forecastingcited by 48
- 2018Bayesian QuickNAT: Model Uncertainty in Deep Whole-Brain Segmentation for Structure-wise Quality Controlcited by 39
- 2020How to Build a Graph-Based Deep Learning Architecture in Traffic Domain: A Surveycited by 38
- 2018Automatic Brain Tumor Segmentation using Convolutional Neural Networks with Test-Time Augmentationcited by 36
- 2019Anomaly Detection in Road Traffic Using Visual Surveillance: A Surveycited by 30
- 2021Global canopy height regression and uncertainty estimation from GEDI LIDAR waveforms with deep ensemblescited by 26
- 2020Empirical Frequentist Coverage of Deep Learning Uncertainty Quantification Procedurescited by 19
- 2020A general framework for ensemble distribution distillationcited by 15
- 2022Traffic4cast at NeurIPS 2021 -- Temporal and Spatial Few-Shot Transfer Learning in Gridded Geo-Spatial Processescited by 15
- 2020Interval Neural Networks: Uncertainty Scorescited by 14
- 2021Applications of deep learning in traffic congestion detection, prediction and alleviation: A surveycited by 14
- 2018Uncertainty Estimation via Stochastic Batch Normalizationcited by 11
- 2022Deep-Ensemble-Based Uncertainty Quantification in Spatiotemporal Graph Neural Networks for Traffic Forecastingcited by 10
- 2021Learning to Transfer for Traffic Forecasting via Multi-task Learningcited by 7
50 most cited of 55 All 55 (JSON)
Built on by
Citing papers per year: 2024: 1.
1 paper.