English

ToPT: Task-Oriented Prompt Tuning for Urban Region Representation Learning

Artificial Intelligence 2026-02-03 v1 Machine Learning

Abstract

Learning effective region embeddings from heterogeneous urban data underpins key urban computing tasks (e.g., crime prediction, resource allocation). However, prevailing two-stage methods yield task-agnostic representations, decoupling them from downstream objectives. Recent prompt-based approaches attempt to fix this but introduce two challenges: they often lack explicit spatial priors, causing spatially incoherent inter-region modeling, and they lack robust mechanisms for explicit task-semantic alignment. We propose ToPT, a two-stage framework that delivers spatially consistent fusion and explicit task alignment. ToPT consists of two modules: spatial-aware region embedding learning (SREL) and task-aware prompting for region embeddings (Prompt4RE). SREL employs a Graphormer-based fusion module that injects spatial priors-distance and regional centrality-as learnable attention biases to capture coherent, interpretable inter-region interactions. Prompt4RE performs task-oriented prompting: a frozen multimodal large language model (MLLM) processes task-specific templates to obtain semantic vectors, which are aligned with region embeddings via multi-head cross-attention for stable task conditioning. Experiments across multiple tasks and cities show state-of-the-art performance, with improvements of up to 64.2\%, validating the necessity and complementarity of spatial priors and prompt-region alignment. The code is available at https://github.com/townSeven/Prompt4RE.git.

Keywords

Cite

@article{arxiv.2602.01610,
  title  = {ToPT: Task-Oriented Prompt Tuning for Urban Region Representation Learning},
  author = {Zitao Guo and Changyang Jiang and Tianhong Zhao and Jinzhou Cao and Genan Dai and Bowen Zhang},
  journal= {arXiv preprint arXiv:2602.01610},
  year   = {2026}
}

Comments

The paper has been accepted by ICASSP 2026

R2 v1 2026-07-01T09:30:52.909Z