Boosting Fine-Grained Urban Flow Inference via Lightweight Architecture and Focalized Optimization

Authors

  • Yuanshao Zhu Southern University of Science and Technology City University of Hong Kong
  • Xiangyu Zhao City University of Hong Kong
  • Zijian Zhang Jilin University
  • Xuetao Wei Southern University of Science and Technology
  • James Jianqiao Yu Harbin Institute of Technology, Shenzhen

DOI:

https://doi.org/10.1609/aaai.v40i19.38696

Abstract

Fine-grained urban flow inference is crucial for urban planning and intelligent transportation systems, enabling precise traffic management and resource allocation. However, the practical deployment of existing methods is hindered by two key challenges: the prohibitive computational cost of over-parameterized models and the suboptimal performance of conventional loss functions on the highly skewed distribution of urban flows. To address these challenges, we propose a unified solution that synergizes architectural efficiency with adaptive optimization. Specifically, we first introduce PLGF, a lightweight yet powerful architecture that employs a Progressive Local-Global Fusion strategy to effectively capture both fine-grained details and global contextual dependencies. Second, we propose DualFocal Loss, a novel function that integrates dual-space supervision with a difficulty-aware focusing mechanism, enabling the model to adaptively concentrate on hard-to-predict regions. Extensive experiments on 4 real-world scenarios validate the effectiveness and scalability of our method. Notably, while achieving state-of-the-art performance, PLGF reduces the model size by up to 97% compared to current high-performing methods. Furthermore, under comparable parameter budgets, our model yields an accuracy improvement of over 10% against strong baselines.

Published

2026-03-14

How to Cite

Zhu, Y., Zhao, X., Zhang, Z., Wei, X., & Jianqiao Yu, J. (2026). Boosting Fine-Grained Urban Flow Inference via Lightweight Architecture and Focalized Optimization. Proceedings of the AAAI Conference on Artificial Intelligence, 40(19), 16557–16565. https://doi.org/10.1609/aaai.v40i19.38696

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management III