Unveiling Stochasticity: Universal Multi-modal Probabilistic Modeling for Traffic Forecasting

• URL: http://arxiv.org/abs/2604.16084v1
• Date: Fri, 17 Apr 2026 14:10:16 GMT
• Title: Unveiling Stochasticity: Universal Multi-modal Probabilistic Modeling for Traffic Forecasting
• Abstract summary: This paper proposes an elegant yet universal approach that transforms existing models into probabilistic predictors.<n> Experiments on multiple traffic datasets show that our approach generalizes from classic to modern model architectures.<n>A more detailed study on a real-world dense urban traffic network is presented to examine the impact of data quality uncertainty.
• Score: 3.268970214362015
• License: http://creativecommons.org/licenses/by-nc-nd/4.0/
• Abstract: Traffic forecasting is a challenging spatio-temporal modeling task and a critical component of urban transportation management. Current studies mainly focus on deterministic predictions, with limited considerations on the uncertainty and stochasticity in traffic dynamics. Therefore, this paper proposes an elegant yet universal approach that transforms existing models into probabilistic predictors by replacing only the final output layer with a novel Gaussian Mixture Model (GMM) layer. The modified model requires no changes to the training pipeline and can be trained using only the Negative Log-Likelihood (NLL) loss, without any auxiliary or regularization terms. Experiments on multiple traffic datasets show that our approach generalizes from classic to modern model architectures while preserving deterministic performance. Furthermore, we propose a systematic evaluation procedure based on cumulative distributions and confidence intervals, and demonstrate that our approach is considerably more accurate and informative than unimodal or deterministic baselines. Finally, a more detailed study on a real-world dense urban traffic network is presented to examine the impact of data quality on uncertainty quantification and to show the robustness of our approach under imperfect data conditions. Code available at https://github.com/Weijiang-Xiong/OpenSkyTraffic

Related papers

• Demystifying Data-Driven Probabilistic Medium-Range Weather Forecasting [63.8116386935854]
  We demonstrate that state-of-the-art probabilistic skill requires neither intricate architectural constraints nor specialized trainings.<n>We introduce a scalable framework for learning multi-scale atmospheric dynamics by combining a directly downsampled latent space with a history-conditioned local projector.<n>We find that our framework design is robust to the choice of probabilistic estimators, seamlessly supporting interpolants, diffusion models, and CRPS-based ensemble training.
  arXiv  Detail & Related papers  (2026-01-26T03:52:16Z)
• ResAD: Normalized Residual Trajectory Modeling for End-to-End Autonomous Driving [64.42138266293202]
  ResAD is a Normalized Residual Trajectory Modeling framework.<n>It reframes the learning task to predict the residual deviation from an inertial reference.<n>On the NAVSIM benchmark, ResAD achieves a state-of-the-art PDMS of 88.6 using a vanilla diffusion policy.
  arXiv  Detail & Related papers  (2025-10-09T17:59:36Z)
• Revisiting Multivariate Time Series Forecasting with Missing Values [65.30332997607141]
  Missing values are common in real-world time series.<n>Current approaches have developed an imputation-then-prediction framework that uses imputation modules to fill in missing values, followed by forecasting on the imputed data.<n>This framework overlooks a critical issue: there is no ground truth for the missing values, making the imputation process susceptible to errors that can degrade prediction accuracy.<n>We introduce Consistency-Regularized Information Bottleneck (CRIB), a novel framework built on the Information Bottleneck principle.
  arXiv  Detail & Related papers  (2025-09-27T20:57:48Z)
• PhysVarMix: Physics-Informed Variational Mixture Model for Multi-Modal Trajectory Prediction [0.0]
  We present a novel hybrid approach that integrates learning-based with physics-based constraints to address the multi-modality inherent in trajectory prediction.<n>Our framework incorporates physical realism through sector-specific boundary conditions and Model Predictive Control (MPC)-based smoothing.<n>Our approach offers a robust and scalable solution for navigating the uncertainties of real-world urban environments.
  arXiv  Detail & Related papers  (2025-07-25T22:45:42Z)
• Solving Inverse Problems with FLAIR [68.87167940623318]
  We present FLAIR, a training-free variational framework that leverages flow-based generative models as prior for inverse problems.<n>Results on standard imaging benchmarks demonstrate that FLAIR consistently outperforms existing diffusion- and flow-based methods in terms of reconstruction quality and sample diversity.
  arXiv  Detail & Related papers  (2025-06-03T09:29:47Z)
• Scene-Aware Explainable Multimodal Trajectory Prediction [15.58042746234974]
  We introduce the Explainable Conditional Diffusion-based Multimodal Trajectory Prediction (DMTP) model.<n>Our model integrates a modified conditional diffusion approach to capture multimodal trajectory patterns and employs a revised Shapley Value model to assess the significance of global and scenario-specific features.<n> Experiments demonstrate that our explainable model excels in identifying critical inputs and significantly outperforms baseline models in accuracy.
  arXiv  Detail & Related papers  (2024-10-22T08:17:33Z)
• Physics-guided Active Sample Reweighting for Urban Flow Prediction [75.24539704456791]
  Urban flow prediction is a nuanced-temporal modeling that estimates the throughput of transportation services like buses, taxis and ride-driven models. Some recent prediction solutions bring remedies with the notion of physics-guided machine learning (PGML) We develop a atized physics-guided network (PN), and propose a data-aware framework Physics-guided Active Sample Reweighting (P-GASR)
  arXiv  Detail & Related papers  (2024-07-18T15:44:23Z)
• A Bayesian approach to quantifying uncertainties and improving generalizability in traffic prediction models [0.0]
  We propose a Bayesian recurrent neural network framework for uncertainty in traffic prediction with higher generalizability. We show that normalization alters the training process of deep neural networks by controlling the model's complexity. Our findings are especially relevant to traffic management applications, where predicting traffic conditions across multiple locations is the goal.
  arXiv  Detail & Related papers  (2023-07-12T06:23:31Z)
• Modeling Uncertain Feature Representation for Domain Generalization [49.129544670700525]
  We show that our method consistently improves the network generalization ability on multiple vision tasks. Our methods are simple yet effective and can be readily integrated into networks without additional trainable parameters or loss constraints.
  arXiv  Detail & Related papers  (2023-01-16T14:25:02Z)
• FloMo: Tractable Motion Prediction with Normalizing Flows [0.0]
  We model motion prediction as a density estimation problem with a normalizing flow between a noise sample and the future motion distribution. Our model, named FloMo, allows likelihoods to be computed in a single network pass and can be trained directly with maximum likelihood estimation. Our method achieves state-of-the-art performance on three popular prediction datasets, with a significant gap to most competing models.
  arXiv  Detail & Related papers  (2021-03-05T11:35:27Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.