用注意力对齐提升小样本下GUI指令理解准确率
GUI-AIMA: Aligning Intrinsic Multimodal Attention with a Context Anchor for GUI Grounding
- 通过注意力对齐实现无需坐标的高效视觉定位
- 仅用50.9万样本训练,3B模型在多个数据集上达最优
- 适合研究轻量级人机交互与多模态模型可解释性
图形化用户界面(GUI)定位是计算机使用智能体的关键能力,将自然语言指令映射到屏幕上的可操作区域。现有基于多模态大模型(MLLM)的方法通常将GUI定位建模为文本驱动的坐标生成任务,但直接从视觉输入生成精确坐标既具挑战性又依赖大量数据。更直观的策略是先识别与指令相关的视觉块,再在其中确定点击位置。受近期发现的通用MLLM在其注意力图中内嵌定位能力的启发,我们提出GUI-AIMA:一种基于注意力且无坐标的监督微调框架,用于高效GUI定位。GUI-AIMA将MLLM的内在多模态注意力与分块定位信号对齐,这些信号通过简化查询-视觉注意力矩阵的多头聚合自适应计算,适用于多样指令。此外,其无坐标设计可轻松集成即插即用的放大阶段。GUI-AIMA-3B仅用50.9万样本(约10.1万张截图)训练,验证了轻量训练即可激发MLLM的原始定位能力,成为3B模型中的最新最佳表现,在ScreenSpot-Pro上平均准确率达61.5%,ScreenSpot-v2为92.1%,OSWorld-G为68.1%,MMBench-GUI-L2为79.1%,UI-Vision为60.0%。
原文摘要 · Abstract (English)
Graphical user interface (GUI) grounding is a key capability for computer-use agents, mapping natural-language instructions to actionable regions on the screen. Existing Multimodal Large Language Model (MLLM) approaches typically formulate GUI grounding as a text-based coordinate generation task. However, directly generating precise coordinates from visual inputs is challenging and often data-intensive. A more intuitive strategy is to first identify instruction-relevant visual patches and then determine the exact click location within them. Motivated by recent observations that general MLLMs exhibit native grounding ability embedded in their attention maps, we propose GUI-AIMA, an attention-based and coordinate-free supervised fine-tuning framework for efficient GUI grounding. GUI-AIMA aligns the intrinsic multimodal attention of MLLMs with patch-wise grounding signals. These signals are calculated adaptively for diverse user instructions by multi-head aggregation on simplified query-visual attention matrices. Besides, its coordinate-free manner can easily integrate a plug-and-play zoom-in stage. GUI-AIMA-3B was trained with only 509k samples (around 101k screenshots), demonstrating exceptional data efficiency and verifying that light training can trigger the native grounding capability of MLLMs. It achieves state-of-the-art performance among 3B models, attaining an average accuracy of 61.5% on ScreenSpot-Pro, 92.1% on ScreenSpot-v2, 68.1% on OSWorld-G, 79.1% on MMBench-GUI-L2, and 60.0% on UI-Vision. Project page: https://github.com/sjz5202/GUI-AIMA .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。