2026
Lightweight and Interpretable Transformer via Unrolling of Mixed Graph Algorithms for Traffic Forecast
ICML 2026poster
To forecast traffic with both spatial and temporal dimensions, we unroll a mixed-graph-based optimization algorithm into a lightweight and interpretable transformer-like neural net. Specifically, we construct two graphs: an undirected graph $\mathcal{G}^u$ capturing spatial correlations across geogr…