cd /news/machine-learning/stagformer-a-spatio-temporal-agent-g… · home topics machine-learning article
[ARTICLE · art-52043] src=arxiv.org ↗ pub= topic=machine-learning verified=true sentiment=↑ positive

STAGformer: A Spatio-temporal Agent Graph Transformer for Micro Mobility Demand Forecasting

Researchers propose STAGformer, a Spatio-Temporal Agent Graph Transformer for micro mobility demand forecasting in bike-sharing systems. The model achieves efficient global modeling with linear computational complexity by using learnable agent tokens to capture long-range interactions. Experiments on NYC Citi-Bike and Chicago Divvy-Bike datasets show it outperforms state-of-the-art baselines in prediction accuracy.

read1 min views1 publishedJul 9, 2026

arXiv:2607.06614v1 Announce Type: new Abstract: Accurate station-level demand forecasting is essential for the efficient operation of bike-sharing systems, yet it remains challenging due to complex spatio-temporal dependencies and the large scale of urban networks. This paper presents STAGformer, a Spatio-Temporal Agent Graph Transformer that achieves efficient global modeling with linear computational complexity. The model introduces a two-step agent attention mechanism, where a small set of learnable spatial and temporal agent tokens first aggregate global information and then broadcast it back to individual stations and time steps, effectively capturing long-range interactions while reducing the quadratic cost of standard self-attention to O(NT). STAGformer integrates four core modules: a spatio-temporal encoder that fuses dynamic node features with external contextual factors (weather, time, points of interest), a graph propagation module for spatial neighbor aggregation, a temporal convolution module for local pattern extraction, and the agent attention module for global dependency modeling. Extensive experiments on two real-world datasets -- NYC Citi-Bike and Chicago Divvy-Bike -- demonstrate that STAGformer consistently outperforms state-of-the-art baselines across multiple prediction horizons, achieving significant improvements in both RMSE and MAE. Ablation studies validate the contribution of each component, with the agent attention mechanism proving critical for modeling global spatio-temporal dependencies.

── more in #machine-learning 4 stories · sorted by recency
── more on @stagformer 3 stories trending now
sponsored brought to you by zahid.host 4,200+ EU-deployed projects
reading about agents? ship yours in a single git push.

Run your AI side-project on zahid.host

EU-based hosting, git-push deploys, automatic HTTPS, no cold starts. Free tier with a custom domain — perfect for shipping the agent you just read about.

$git push zahid main
Live at https://your-agent.zahid.host
Get free account → Pricing
from €0/mo · no card required
LIVE [news/stagformer-a-spatio-…] indexed:0 read:1min 2026-07-09 ·