UrbanGraphEmbeddings: Learning and Evaluating Spatially Grounded Multimodal Embeddings for Urban Science
About
Learning transferable multimodal embeddings for urban environments is challenging because urban understanding is inherently spatial, yet existing datasets and benchmarks lack explicit alignment between street-view images and urban structure. We introduce UGData, a spatially grounded dataset that anchors street-view images to structured spatial graphs and provides graph-aligned supervision via spatial reasoning paths and spatial context captions, exposing distance, directionality, connectivity, and neighborhood context beyond image content. Building on UGData, we propose UGE, a two-stage training strategy that progressively and stably aligns images, text, and spatial structures by combining instruction-guided contrastive learning with graph-based spatial encoding. We finally introduce UGBench, a comprehensive benchmark to evaluate how spatially grounded embeddings support diverse urban understanding tasks -- including geolocation ranking, image retrieval, urban perception, and spatial grounding. We develop UGE on multiple state-of-the-art VLM backbones, including Qwen2-VL, Qwen2.5-VL, Phi-3-Vision, and LLaVA1.6-Mistral, and train fixed-dimensional spatial embeddings with LoRA tuning. UGE built upon Qwen2.5-VL-7B backbone achieves up to 44% improvement in image retrieval and 30% in geolocation ranking on training cities, and over 30% and 22% gains respectively on held-out cities, demonstrating the effectiveness of explicit spatial grounding for spatially intensive urban tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Geolocation Ranking | New York | Hit@558.85 | 32 | |
| Nearest POI Spatial Grounding | SG | Hit@570.77 | 32 | |
| Nearest POI Spatial Grounding | PA | Hit@548.85 | 32 | |
| Nearest POI Spatial Grounding | BJ | Hit@575.38 | 32 | |
| Distance-Direction Spatial Grounding | NY | Hit@544.35 | 16 | |
| Distance-Direction Spatial Grounding | SG | Hit@534.23 | 16 | |
| Distance-Direction Spatial Grounding | BJ | Hit@558.95 | 16 | |
| Distance-Direction Spatial Grounding | PA | Hit@550.38 | 16 | |
| Geolocation Ranking | Singapore SG | Hit@564.23 | 16 | |
| Geolocation Ranking | Beijing (BJ) | Hit@571.92 | 16 |