SafeGround:通过不确定性校准了解何时信任GUI grounding模型 SafeGround: Know When to Trust GUI Grounding Models via Uncertainty Calibration
不确定性感知框架,通过校准实现GUI grounding的风险控制
前置知识
GUI grounding
GUI grounding是将自然语言指令转换为图形用户界面上可执行屏幕坐标的技术。给定一个UI截图和用户指令(如“点击提交按钮”),模型需要预测目标UI元素在屏幕上的精确位置坐标。这是构建自主GUI代理的核心组件,使得视觉语言模型能够理解并操作数字界面。传统方法将问题建模为基于文本的坐标预测任务,模型生成点坐标作为输出。
理解GUI grounding的基本概念是理解本文研究问题的前提,因为本文正是针对现有GUI grounding模型缺乏不确定性估计这一关键局限提出解决方案。
不确定性量化(Uncertainty Quantification)
不确定性量化是评估模型预测可靠性的方法,通过量化模型对其输出的信心程度来识别可能错误的预测。在深度学习中,不确定性通常分为认知不确定性(模型参数的不确定性)和偶然不确定性(数据本身的噪声)。常见方法包括基于模型概率的置信度、蒙特卡洛采样、集成方法等。不确定性量化对于高风险决策场景至关重要,可以帮助系统知道何时应该拒绝预测或寻求外部帮助。
本文的核心创新就是为GUI grounding模型引入有效的不确定性量化方法,使得系统能够识别不可靠的预测并采取相应的风险控制措施。
选择性预测(Selective Prediction)
选择性预测是一种机器学习范式,允许模型在面对不确定预测时选择“拒绝回答”而非给出可能错误的输出。模型设定一个置信度阈值,只有当预测置信度超过阈值时才输出结果,否则拒绝预测或将其转交给更强大的模型。这种方法通过牺牲部分覆盖率来提高预测的可靠性,在医疗诊断、金融决策等高风险场景中尤为重要。
本文采用选择性预测机制来实现GUI grounding的风险控制,通过校准阈值确保接受的预测中错误率不超过用户指定的风险水平。
Learn Then Test (LTT) 校准
Learn Then Test是一种后处理校准范式,将模型学习与统计风险控制分离。给定一个固定的预测模型,LTT将决策制定视为在低维决策空间上的假设检验问题,使用留出的校准数据来识别满足用户指定风险约束的参数,并提供有限样本保证。分割保形预测(Split Conformal Prediction)是LTT的一种实现,利用数据分割和基于置信区间的边界来进行有效的风险控制。
本文采用LTT校准框架来实现统计上严格的错误发现率(FDR)控制,为GUI grounding提供了有限样本的风险保证。
错误发现率(False Discovery Rate, FDR)
错误发现率是统计学中用于多重假设检验的风险度量,定义为被错误接受的预测在所有被接受预测中的比例。在GUI grounding场景中,FDR表示在所有被系统接受并执行的预测中,错误预测所占的比例。控制FDR对于高风险GUI操作(如支付确认、系统配置更改)至关重要,可以确保系统执行的操作中错误率不超过用户可接受的水平。
本文的核心目标之一就是通过校准阈值实现严格的FDR控制,确保在实际部署中GUI代理的错误操作率在统计上有保证。
研究动机
在实际的GUI交互场景中,一个错误的grounding预测可能导致昂贵且难以撤销的操作,例如错误的支付批准或不可逆的系统配置更改。然而,现有的GUI grounding模型通常只输出确定性的点预测,没有提供任何关于预测可靠性或不确定性估计的信息。当模型面对模糊或复杂的界面元素时(如多个相似的按钮、密集的UI布局),它无法识别自己的不确定性,导致用户可能错误地信任不正确的预测。例如,在ScreenSpot-Pro这样的专业级高分辨率GUI基准测试中,现有模型在密集UI布局和细粒度元素区分方面表现不佳,但无法告知用户何时应该谨慎对待其输出。此外,现有的不确定性量化技术(如基于模型概率或口头化自我评估)在GUI grounding场景中存在系统性偏差,无法提供可靠的不确定性信号。
本文的目标是本文旨在为GUI grounding模型引入一个不确定性感知框架,使得系统能够可靠地评估预测的不确定性,并基于用户指定的风险水平进行选择性预测。具体目标包括:(1)开发一种仅依赖模型输出分布(而非内部状态)的不确定性量化方法,适用于黑盒视觉语言模型;(2)通过校准过程实现统计上严格的错误发现率控制,提供有限样本保证;(3)在保持系统可用性的同时,显著提高高风险GUI交互场景中的系统级准确率。通过不确定性感知的级联推理,系统可以在本地模型不确定时调用更强大的外部模型,从而在控制风险的前提下提升整体性能。
与已有工作不同的是,本文的独特切入角度在于认识到现有GUI grounding研究的两个关键盲点:首先,现有工作主要关注提高点预测的准确性,而忽视了预测可靠性评估的重要性;其次,现有不确定性量化方法要么需要访问模型内部状态(对黑盒模型不可行),要么依赖模型的口头化自信度(在GUI场景中不可靠),或者只产生不确定性分数而没有指定如何在部署时采取行动。本文抓住了“空间分布”这一被忽视的信号——通过多次随机采样构建预测的空间分布,利用空间离散程度来量化不确定性。这种方法不需要模型内部访问,不依赖模型的显式置信度输出,而是从预测本身的变异性中提取可靠性信息,为GUI grounding提供了一种实用且原则性的风险控制机制。
核心方法
SAFEGROUND的核心思想可以用一个直觉的比喻来理解:就像人类在不确定时会犹豫不决、寻求更多信息一样,SAFEGROUND通过多次“询问”同一模型(通过随机采样)来观察其“犹豫程度”。如果模型对同一个问题给出了一致的空间位置(预测集中在小区域内),说明模型很有信心;如果预测分散在屏幕的不同区域,说明模型不确定。技术路线上,SAFEGROUND首先通过蒙特卡洛采样获取多个随机预测点,然后将这些点投影到离散化的屏幕网格上构建空间密度分布。接着,通过密度聚类将高密度区域聚合为对象级的候选区域,并计算三种互补的不确定性度量:候选区域间的歧义性(TA)、信息离散度(IE)和浓度缺失度(CD)。最后,采用Learn Then Test校准框架,在留出的校准集上确定决策阈值,确保接受的预测中错误率不超过用户指定的风险水平α。在测试时,低不确定性预测直接执行,高不确定性预测则被拒绝或转交给更强大的外部模型。
SAFEGROUND最本质的创新在于将GUI grounding从点预测范式转变为分布感知范式。传统方法将模型输出视为一个确定性的坐标点$\hat{y} = (\hat{u}, \hat{v}) \in \mathbb{R}^2$,而SAFEGROUND将其视为一个空间分布的采样结果。这种视角转换带来了几个关键优势:首先,它允许我们从预测的变异性中提取不确定性信息,而不需要访问模型内部状态或依赖模型的显式置信度输出;其次,三种互补的不确定性度量分别捕捉了不同的失败模式——候选区域间的歧义性(当存在多个相似目标时)、全局信息离散度(当模型信心分散时)和空间浓度缺失度(当缺乏明确焦点时);最重要的是,通过LTT校准框架,不确定性阈值的选择具有统计上的严格保证,而不是经验性的调整。这种采样-分布-校准的范式使得SAFEGROUND能够作为即插即用的模块,集成到任何现有的GUI grounding模型中,而无需修改模型架构或重新训练。
方法步骤详情
SAFEGROUND的方法包含四个主要步骤。第一步是基于采样的空间分布构建:对于每个输入(截图$x$和指令$q$),使用温度为1.0的随机解码进行$K=10$次随机前向传播,生成坐标集合$S=\{\hat{y}^{(i)}\}_{i=1}^{K}$。这些采样坐标被投影到离散化的屏幕网格(patch size=14)上,构建归一化的局部密度图$P$。第二步是区域提取与评分:通过密度聚类将连续的高密度patch聚合为不相交的区域$R=\{R_m\}_{m=1}^{M}$,每个区域$R_m$的得分$S_m$为其平均概率密度,然后按得分降序排列。第三步是不确定性度量计算:基于区域得分分布,计算三种互补的不确定性分数——Top-Candidate Ambiguity ($U_{TA}$)测量前两个候选区域的边际差异,Informational Dispersion ($U_{IE}$)计算归一化熵,Concentration Deficit ($U_{CD}$)计算1减去概率分布的二次集中度。最终的不确定性分数$U_{COM}$是这三个分数的加权组合:$U_{COM} = w_{CD} \cdot U_{CD} + w_{IE} \cdot U_{IE} + w_{TA} \cdot U_{TA}$。第四步是阈值校准与选择性预测:在校准集$D_{cal}$上,对每个样本计算不确定性分数$u_i$和错误指示函数$err_i$,然后遍历所有可能的阈值$\tau$,使用Clopper-Pearson置信区间计算FDR的上界,选择满足$\hat{FDR}^{upper}_{1-\delta}(\tau) \leq \alpha$的最大阈值$\hat{\tau}$。在测试时,不确定性分数$\leq \hat{\tau}$的预测被接受执行,$> \hat{\tau}$的预测被拒绝或转交给更强模型。
技术新颖性
SAFEGROUND的技术新颖性体现在三个层面。首先,在不确定性量化层面,与现有依赖模型概率或口头化自信度的方法不同,SAFEGROUND提出了一种基于空间分布的纯输出级不确定性度量,不需要访问模型内部状态。这种方法特别适合黑盒视觉语言模型,且通过三种互补度量的组合,能够捕捉不同类型的预测失败模式。其次,在风险控制层面,SAFEGROUND首次将Learn Then Test校准框架引入GUI grounding领域,提供了严格的有限样本FDR保证。与经验性阈值选择不同,这种方法在统计上确保了接受预测的错误率不超过用户指定的风险水平,并且在校准阶段就能判断目标风险水平是否可达。第三,在系统设计层面,SAFEGROUND实现了不确定性感知的级联推理机制,当本地模型不确定时自动转交给更强的外部模型(如Gemini)。这种设计不仅提高了系统级准确率(最高提升5.38%),还通过减少对昂贵外部模型的调用降低了计算成本。这种“采样-分布-校准-级联”的完整框架是GUI grounding领域的首创,为安全部署GUI代理提供了原则性的解决方案。
实验结果
SAFEGROUND在多个维度上展示了显著的效果。在不确定性估计质量方面,Table 2显示$U_{COM}$在所有6个GUI grounding模型上都达到了较高的AUROC值(0.63-0.82),在Holo1.5-7B上从基线的0.6983提升到0.7526,在GUI-Actor-2VL-7B上达到0.8155。对于无法使用概率置信度(PC)的模型(如GUI-Actor变体),$U_{COM}$仍然表现出强大的区分能力。Table 3的AUARC结果进一步证实了$U_{COM}$在选择性预测中的有效性,在Holo1.5-3B上从0.6444提升到0.6576。在FDR控制方面,Figure 3显示在所有测试模型和风险水平下,实际FDR都严格低于理论上限,验证了SAFEGROUND提供的严格安全保证。Power比较(Figure 4)表明$U_{COM}$在相同风险水平下比PC保留了更多正确的预测,特别是在严格风险水平(如$\alpha=0.38$)下PC往往无法产生有效预测,而$U_{COM}$仍能保持较高的Power。最引人注目的是系统级准确率提升:Table 1显示在ScreenSpot-Pro上,Holo1.5-7B在风险水平$\alpha=0.34$下通过级联推理达到58.66%准确率,比Gemini-only推理提升5.38%;UI-TARS-1.5-7B和GUI-Actor-2VL-7B的提升更为显著,分别达到13.12%和14.39%。这些结果表明SAFEGROUND不仅提供了可靠的不确定性估计,还能在实际部署中带来实质性的性能提升。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| GUI grounding (ScreenSpot-Pro) | 系统级准确率 (%) | 58.66% (Holo1.5-7B, α=0.34) | 53.28% (Gemini-only) | +5.38% |
| GUI grounding (ScreenSpot-Pro) | 系统级准确率 (%) | 53.68% (UI-TARS-1.5-7B, α=0.34) | 41.58% | +12.10% |
| GUI grounding (ScreenSpot-Pro) | 系统级准确率 (%) | 55.18% (GUI-Actor-2VL-7B, α=0.34) | 40.79% | +14.39% |
| 不确定性区分 | AUROC | 0.8155 (GUI-Actor-2VL-7B) | 0.6983 (Holo1.5-7B PC) | +0.1172 |
| 选择性预测 | AUARC | 0.7166 (GUI-Actor-2VL-7B) | 0.6444 (Holo1.5-3B PC) | +0.0722 |
局限与改进
尽管SAFEGROUND取得了显著成果,但仍存在几个值得关注的局限性。首先,不确定性估计依赖于采样预测的变异性来表征空间歧义性,对于高度确定性、采样多样性有限的模型,产生的空间分布可能信息量不足。其次,当前方法需要对每个输入进行K=10次随机采样,这增加了推理时间和计算成本,虽然Figure 6显示从K=5到K=10有明显改进,但进一步增加到K=15只有边际收益。第三,SAFEGROUND的FDR控制依赖于校准集的质量和代表性,如果校准数据与测试数据分布不匹配,风险保证可能会受到影响。第四,虽然论文展示了在ScreenSpot-Pro基准上的有效性,但该基准相对较小(1581个样本),在更大规模、更多样化的GUI场景中的泛化能力还需要进一步验证。此外,当前的不确定性度量权重(wCD=0.6, wIE=0.2, wUTA=0.2)是固定且跨模型通用的,虽然消融研究(Table 4)显示最优权重因模型而异,但这种固定权重方案可能不是所有情况下的最优选择。最后,论文主要关注点坐标的grounding,对于更复杂的GUI操作(如拖拽、多步操作)的不确定性量化尚未探索。
独立分析的弱点
基于独立分析,SAFEGROUND存在几个可以改进的弱点。首先,采样效率方面,当前需要10次随机采样来构建不确定性估计,这在实时GUI交互场景中可能带来不可接受的延迟。改进方向包括开发更高效的采样策略(如重要性采样)或使用轻量级代理模型来近似不确定性。其次,权重调优方面,虽然论文声称使用固定权重是“即插即用”的,但消融研究显示不同模型的最优权重组合不同。一个改进方向是开发自适应权重学习机制,根据模型特性和任务特征动态调整权重。第三,校准数据需求方面,当前方法需要专门的校准集,这在资源受限或数据稀缺的场景中可能不切实际。可以探索少样本校准或在线自适应校准方法。第四,区域提取方面,当前使用固定的patch size=14和阈值β=0.3,这些超参数可能不适用于所有分辨率和UI密度。可以开发自适应的区域提取策略,根据界面复杂度动态调整参数。最后,级联策略方面,当前只是简单地将高不确定性样本转交给更强模型,没有考虑不同类型的不确定性可能需要不同的处理策略(如请求用户澄清、提供多个候选等)。
未来方向
论文作者和基于本文成果可以延伸的未来研究方向包括多个维度。作者提出的方面:SAFEGROUND为不确定性感知的GUI grounding奠定了基础,未来可以探索将不确定性估计集成到模型训练过程中,而不仅仅是后处理;可以研究如何将SAFEGROUND扩展到其他GUI交互任务,如元素识别、操作序列规划等。基于成果可延伸的方向:首先,多模态不确定性融合,将视觉不确定性(如界面模糊、遮挡)与语言不确定性(如指令歧义)相结合,提供更全面的可靠性评估。其次,主动学习与不确定性引导的数据收集,利用不确定性识别模型最需要改进的案例,指导数据标注和模型微调。第三,层次化风险控制,根据不同操作的严重程度(如浏览vs支付)动态调整风险水平,实现更细粒度的安全控制。第四,跨平台泛化,研究SAFEGROUND在不同操作系统(Windows、macOS、移动端)和不同应用类型(Web、桌面应用、移动应用)中的适应性。最后,人机协作不确定性解决,当系统检测到高不确定性时,不仅是拒绝或转交,还可以与用户进行交互式澄清,构建更智能的人机协作系统。
复现评估
从复现评估角度来看,SAFEGROUND具有较好的可复现性。代码和数据方面,论文基于公开的ScreenSpot-Pro基准测试,该基准包含1581个UI截图和自然语言指令,数据相对容易获取。论文使用的6个GUI grounding模型(Holo1.5-3B/7B、GUI-Actor-2VL-7B/2.5VL-7B、UI-TARS-1.5-7B、GTA1-7B)都是开源的,可以在HuggingFace等平台下载。算力需求方面,主要计算成本来自K=10次随机采样和阈值校准过程,对于单个模型在ScreenSpot-Pro上的完整评估,估计需要几小时到一天的GPU时间(具体取决于硬件)。复现难度中等,因为论文提供了详细的算法描述(Algorithm 1)和超参数设置(patch size=14、β=0.3、δ=0.05),但一些实现细节(如密度聚类的具体参数、区域提取的边界处理)可能需要参考附录或源代码。主要挑战在于准确理解论文中的数学符号和公式,特别是Clopper-Pearson置信区间的实现。总体而言,只要具备基本的深度学习实验环境和GUI grounding背景知识,复现SAFEGROUND的主要结果应该是可行的。
论文图表
展示了现有GUI grounding模型与SAFEGROUND在风险控制方面的对比。左侧显示现有模型可能在“删除购物车物品”这样的高风险操作上产生错误的点击预测(235, 834),而SAFEGROUND通过检测到高不确定性而拒绝立即预测,转而咨询外部知识,最终产生正确的预测(232, 733)。
这张图直观地展示了本文的核心动机——现有模型缺乏不确定性估计可能导致昂贵的错误操作,而SAFEGROUND通过不确定性感知的风险控制可以避免这种情况。