Granger-Guided AI Predicts Traffic Flow With Causal Clues

Every morning, millions of commuters sit in traffic that was, in principle, predictable hours earlier. Now a team of researchers at the University of Pisa, working with collaborators in France, has unveiled a deep learning framework that promises to make those predictions sharper, longer-ranged, and more interpretable than ever before. In a study published in Discover Artificial Intelligence, Chenxi Wang, Chiara Riccardi, Nicholas Fiorentini, and Massimo Losa introduce the Granger Causality-Guided Multi-Graph Transformer, or GC-MGT, a model that borrows a classic statistical idea from economics and welds it to the most powerful sequence-processing architecture in modern machine learning.

The core insight is deceptively simple. Most traffic forecasting models learn correlations: when sensor A reads a slowdown, sensor B often does too. But correlations say nothing about direction. Granger causality, a technique dating back to the 1960s, asks a more useful question: does the past of one time series improve predictions of another? If knowing the history of sensor A lets you forecast sensor B better than B’s own history alone, then A carries directional predictive information about B. The Pisa team treats this statistical signal as a prior, a structural hint about how congestion actually propagates along a road network.

GC-MGT is built from four cooperating modules. The first is a hierarchical dynamic dependency module that runs a linear Granger test over sliding windows of one day’s observations, sampled every five minutes, with a lag order of twelve steps. Because the model runs thousands of pairwise tests, the authors apply the Benjamini-Hochberg procedure to control the false discovery rate at five percent, keeping only statistically defensible directional links. That fixed Granger graph is then multiplied by a sigmoid-bounded neural refinement that adapts edge strengths to the current traffic state, capturing nonlinear, time-varying interactions that a purely linear test would miss. Regularization terms encourage sparsity and temporal smoothness so the dependency graph does not thrash from one input window to the next.

The second module fuses three different views of the network. A physical topology graph encodes actual road connections, weighted by shortest-path distance with a five-kilometer connectivity threshold. A semantic graph, learned through multi-head attention over recent observations, captures sensors whose traffic patterns behave alike even if they sit far apart. The causality-inspired graph from the first module supplies directionality. A small neural network generates learnable, state-dependent weights that blend all three into a single fused adjacency matrix at every time step. During free-flowing conditions the semantic graph may dominate; during a merge-zone jam, the physical and directional graphs take over.

The third module is the Transformer itself, but modified in two crucial ways. The fused graph is injected directly into the attention logits as a bias term, so the model’s attention literally leans along the directions where traffic influence flows. In parallel, four convolutional branches with kernel sizes of one, three, five, and seven extract patterns at multiple temporal scales, from abrupt braking events to daily commute rhythms. A learned gating mechanism balances the convolutional and attention streams, and a lightweight non-autoregressive decoder produces all forecast horizons simultaneously, up to sixty minutes ahead.

The fourth module is the most unusual: a reality check performed inside a traffic simulator. Using the Simulation of Urban Mobility platform, or SUMO, the team reconstructed the Florence-Pisa-Livorno highway corridor, calibrating it with a Greenshields fundamental diagram until simulated speeds matched observations with a mean absolute percentage error of just 8.6 percent. They then ran three intervention experiments, reducing a mainline speed limit from 110 to 80 kilometers per hour, cutting ramp inflow by 30 percent, and closing an acceleration lane, each repeated ten times with different random seeds. By comparing the model’s predicted responses to SUMO’s simulated propagation, and measuring the overlap of affected sensors with a Jaccard index, they tested whether the learned dependencies actually reflect how disturbances travel through a real highway.

The results are striking. Across three datasets, the California PeMS08 benchmark with 170 sensors, the San Francisco Bay Area PEMS-BAY dataset with 325 loop detectors, and the FI-PI-LI Tuscan highway with its notoriously tight Ginestra Fiorentina interchange, GC-MGT beat every baseline at every horizon. At the hardest sixty-minute forecast, relative to PDFormer, the strongest competing model, GC-MGT cut mean absolute error by 13.3, 21.3, and 12.0 percent across the three datasets, and root mean squared error by 3.2, 24.5, and 19.2 percent. The advantage was largest on FI-PI-LI, where complex merging zones produce chaotic, disturbance-driven fluctuations that conventional models handle poorly. Paired t-tests on five independent runs confirmed the improvements were statistically significant, not artifacts of random initialization.

Ablation experiments revealed exactly where the performance comes from. Removing the entire directional-dependency mechanism was devastating, inflating MAE by 23.3 percent on PeMS08, 38.1 percent on PEMS-BAY, and 54.6 percent on FI-PI-LI. The Granger prior and the neural refinement each contributed independently, and adaptive multi-graph fusion consistently outperformed fixed weighting. Crucially, in the SUMO intervention tests, GC-MGT achieved the lowest prediction error under intervention and the highest affected-sensor similarity in all three scenarios, reaching a Jaccard index of 0.87 in the speed-limit experiment, compared with 0.80 for Graph WaveNet and 0.75 for PDFormer. The model’s internal dependency structure, in other words, behaves like genuine traffic physics.

The authors are careful about what this means. Granger analysis, they stress, demonstrates predictive precedence, not structural causation; the learned graph is a map of directional predictive dependencies, not a proof of mechanism. Still, the practical implications are considerable. Because the framework is both accurate and interpretable, the researchers envision integrating it with digital twins and reinforcement learning controllers to drive proactive traffic management: variable speed limits, adaptive ramp metering, and scenario-based optimization that responds to predicted congestion before it materializes. Congestion costs cities billions in fuel, time, and greenhouse gas emissions each year, and the TomTom traffic index shows most major cities wrestling with worsening gridlock.

What makes this work resonate beyond transportation is its methodological message. Transformers, the architecture behind modern language models, are superb at capturing long-range patterns but notoriously blind to why things happen. By anchoring attention in statistically tested directional dependencies, and then validating those dependencies against simulated interventions, the Pisa team has sketched a template for making black-box forecasting models legible without sacrificing accuracy. It is a rare case where an economic statistic from 1969 and a cutting-edge neural architecture find common cause on a Tuscan highway, and both come out ahead.

Subject of Research: Granger causality-guided deep learning framework for spatio-temporal traffic flow forecasting

Article Title: A Granger causality-guided multi-graph transformer framework for traffic flow forecasting

Article References: Wang, C., Riccardi, C., Fiorentini, N., & Losa, M. (2026). A Granger causality-guided multi-graph transformer framework for traffic flow forecasting. Discover Artificial Intelligence, 6(1), Article 1219. https://doi.org/10.1007/s44163-026-02230-y

Image Credits: AI Generated

DOI: 10.1007/s44163-026-02230-y

Keywords: traffic flow forecasting, Granger causality, transformer, multi-graph neural networks, spatio-temporal prediction, directional dependency, SUMO simulation, deep learning, traffic management, digital twins, intelligent transportation systems, interpretability