Urban Region Pre-training and Prompting: A Graph-based Approach

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Jin, Jiahui, Song, Yifan, Kan, Dong, Zhu, Haojia, Sun, Xiangguo, Li, Zhicheng, Sun, Xigang, Zhang, Jinghui
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916824488083456
author Jin, Jiahui
Song, Yifan
Kan, Dong
Zhu, Haojia
Sun, Xiangguo
Li, Zhicheng
Sun, Xigang
Zhang, Jinghui
author_facet Jin, Jiahui
Song, Yifan
Kan, Dong
Zhu, Haojia
Sun, Xiangguo
Li, Zhicheng
Sun, Xigang
Zhang, Jinghui
contents Urban region representation is crucial for various urban downstream tasks. However, despite the proliferation of methods and their success, acquiring general urban region knowledge and adapting to different tasks remains challenging. Existing work pays limited attention to the fine-grained functional layout semantics in urban regions, limiting their ability to capture transferable knowledge across regions. Further, inadequate handling of the unique features and relationships required for different downstream tasks may also hinder effective task adaptation. In this paper, we propose a $\textbf{G}$raph-based $\textbf{U}$rban $\textbf{R}$egion $\textbf{P}$re-training and $\textbf{P}$rompting framework ($\textbf{GURPP}$) for region representation learning. Specifically, we first construct an urban region graph and develop a subgraph-centric urban region pre-training model to capture the heterogeneous and transferable patterns of entity interactions. This model pre-trains knowledge-rich region embeddings using contrastive learning and multi-view learning methods. To further refine these representations, we design two graph-based prompting methods: a manually-defined prompt to incorporate explicit task knowledge and a task-learnable prompt to discover hidden knowledge, which enhances the adaptability of these embeddings to different tasks. Extensive experiments on various urban region prediction tasks and different cities demonstrate the superior performance of our framework.
format Preprint
id arxiv_https___arxiv_org_abs_2408_05920
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Urban Region Pre-training and Prompting: A Graph-based Approach
Jin, Jiahui
Song, Yifan
Kan, Dong
Zhu, Haojia
Sun, Xiangguo
Li, Zhicheng
Sun, Xigang
Zhang, Jinghui
Artificial Intelligence
Machine Learning
Urban region representation is crucial for various urban downstream tasks. However, despite the proliferation of methods and their success, acquiring general urban region knowledge and adapting to different tasks remains challenging. Existing work pays limited attention to the fine-grained functional layout semantics in urban regions, limiting their ability to capture transferable knowledge across regions. Further, inadequate handling of the unique features and relationships required for different downstream tasks may also hinder effective task adaptation. In this paper, we propose a $\textbf{G}$raph-based $\textbf{U}$rban $\textbf{R}$egion $\textbf{P}$re-training and $\textbf{P}$rompting framework ($\textbf{GURPP}$) for region representation learning. Specifically, we first construct an urban region graph and develop a subgraph-centric urban region pre-training model to capture the heterogeneous and transferable patterns of entity interactions. This model pre-trains knowledge-rich region embeddings using contrastive learning and multi-view learning methods. To further refine these representations, we design two graph-based prompting methods: a manually-defined prompt to incorporate explicit task knowledge and a task-learnable prompt to discover hidden knowledge, which enhances the adaptability of these embeddings to different tasks. Extensive experiments on various urban region prediction tasks and different cities demonstrate the superior performance of our framework.
title Urban Region Pre-training and Prompting: A Graph-based Approach
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2408.05920