What Happens Before Decoding? Prefill Determines GUI Grounding in VLMs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lin, Jiaping, Shen, Fei, Li, Junzhe, Nie, Ping, Yu, Fei, Li, Ming, Li, Haizhou
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916006429982720
author Lin, Jiaping
Shen, Fei
Li, Junzhe
Nie, Ping
Yu, Fei
Li, Ming
Li, Haizhou
author_facet Lin, Jiaping
Shen, Fei
Li, Junzhe
Nie, Ping
Yu, Fei
Li, Ming
Li, Haizhou
contents Existing training-free approaches for GUI grounding often rely on multiple inference runs, such as iterative cropping or candidate aggregation, to identify target elements. Despite this additional computation, each forward pass still independently interprets the instruction and parses the visual layout, without enabling progressive interaction among visual tokens. In this paper, we study what happens during GUI grounding in Vision-Language Models (VLMs) and identify a previously overlooked bottleneck. We show that grounding follows a two-stage paradigm: the prefill stage determines candidate UI elements, while the decoding stage subsequently refines the final coordinates. This asymmetry establishes prefill as the critical step, as errors in candidate selection cannot be effectively corrected during decoding. Based on this observation, we propose Re-Prefill, a training-free method that revisits inference by introducing an attention-guided second prefill stage to refine target selection. Specifically, visual tokens that consistently receive high attention from the query position, i.e., the final token, across layers are extracted as a preliminary target hypothesis and appended to the input, together with the instruction hidden states, enabling the model to deeply re-think its decision before coordinate generation. Experiments across four VLMs and five benchmarks, including ScreenSpot-Pro, ScreenSpot-V2, OSWorld-G, UI-Vision, and MMBench-GUI, demonstrate consistent improvements without additional training, with gains of up to 4.3% on ScreenSpot-Pro. Code will be available at https://github.com/linjiaping1/Re-Prefill.
format Preprint
id arxiv_https___arxiv_org_abs_2605_12549
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle What Happens Before Decoding? Prefill Determines GUI Grounding in VLMs
Lin, Jiaping
Shen, Fei
Li, Junzhe
Nie, Ping
Yu, Fei
Li, Ming
Li, Haizhou
Computer Vision and Pattern Recognition
Existing training-free approaches for GUI grounding often rely on multiple inference runs, such as iterative cropping or candidate aggregation, to identify target elements. Despite this additional computation, each forward pass still independently interprets the instruction and parses the visual layout, without enabling progressive interaction among visual tokens. In this paper, we study what happens during GUI grounding in Vision-Language Models (VLMs) and identify a previously overlooked bottleneck. We show that grounding follows a two-stage paradigm: the prefill stage determines candidate UI elements, while the decoding stage subsequently refines the final coordinates. This asymmetry establishes prefill as the critical step, as errors in candidate selection cannot be effectively corrected during decoding. Based on this observation, we propose Re-Prefill, a training-free method that revisits inference by introducing an attention-guided second prefill stage to refine target selection. Specifically, visual tokens that consistently receive high attention from the query position, i.e., the final token, across layers are extracted as a preliminary target hypothesis and appended to the input, together with the instruction hidden states, enabling the model to deeply re-think its decision before coordinate generation. Experiments across four VLMs and five benchmarks, including ScreenSpot-Pro, ScreenSpot-V2, OSWorld-G, UI-Vision, and MMBench-GUI, demonstrate consistent improvements without additional training, with gains of up to 4.3% on ScreenSpot-Pro. Code will be available at https://github.com/linjiaping1/Re-Prefill.
title What Happens Before Decoding? Prefill Determines GUI Grounding in VLMs
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2605.12549