Thinking with Map: Reinforced Parallel Map-Augmented Agent for Geolocalization

Yuxiang Ji, Yong Wang, Ziyu Ma, Yiming Hu, Hailang Huang +4 more
1/8/2026
cs.CVcs.AIcs.CL

No official code found. Best implementation: AMAP-ML/Thinking-with-Map (177 stars, Apache-2.0, last commit 2026-03-09).

Abstract

The image geolocalization task aims to predict the location where an image was taken anywhere on Earth using visual clues. Existing large vision-language model (LVLM) approaches leverage world knowledge, chain-of-thought reasoning, and agentic capabilities, but overlook a common strategy used by humans -- using maps. In this work, we first equip the model \textit{Thinking with Map} ability and formulate it as an agent-in-the-map loop. We develop a two-stage optimization scheme for it, including agentic reinforcement learning (RL) followed by parallel test-time scaling (TTS). The RL strengthens the agentic capability of model to improve sampling efficiency, and the parallel TTS enables the model to explore multiple candidate paths before making the final prediction, which is crucial for geolocalization. To evaluate our method on up-to-date and in-the-wild images, we further present MAPBench, a comprehensive geolocalization training and evaluation benchmark composed entirely of real-world images. Experimental results show that our method outperforms existing open- and closed-source models on most metrics, specifically improving Acc@500m from 8.0\% to 22.1\% compared to \textit{Gemini-3-Pro} with Google Search/Map grounded mode.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

[ACL 2026 Findings] Thinking with Map: Reinforced Parallel Map-Augmented Agent for Geolocalization

1777Jinja, ShellJan 7, 20267 months agoApache-2.0
agentagentic-rlgeo-localizationmllmreasoning

Similar Papers

Cite this paper

@article{ji2026thinking,
  title  = {Thinking with Map: Reinforced Parallel Map-Augmented Agent for Geolocalization},
  author = {Yuxiang Ji and Yong Wang and Ziyu Ma and Yiming Hu and Hailang Huang and Xuecai Hu and Guanhua Chen and Liaoni Wu and Xiangxiang Chu},
  year   = {2026},
  eprint = {2601.05432},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.05432v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.05432.svg)](https://codeofpaper.com/paper/2601.05432)

Discussion