无需标注数据,测试时实时优化视觉语言模型性能。
TTRV: Test-Time Reinforcement Learning for Vision Language Models
- 基于输出频率设计奖励,在测试时多次推理动态调整模型。
- 图像识别最高提升52.4%,跨16数据集平均提升24.6%。
- 仅用一个无标签样本也能提升5.5%,适合资源受限场景。
现有强化学习奖励信号提取方法通常依赖标注数据和专门训练集,与人类直接从环境学习的方式相悖。本文提出TTRV,通过在推理时动态适应视觉语言模型,无需任何标注数据即可提升理解能力。具体而言,改进了组相对策略优化(GRPO)框架,基于基础模型输出频率设计奖励,并在每个测试样本上多次推理以生成反馈。同时,通过鼓励输出经验分布熵降低来控制输出多样性。该方法在物体识别和视觉问答(VQA)任务中均表现优异,分别实现最高52.4%和29.8%的提升,跨16个数据集平均提升24.6%和10.0%。值得注意的是,将TTRV应用于InternVL 8B模型后,在8个基准上平均超过GPT-4o 2.3%,且在VQA任务上仍具竞争力,表明测试时强化学习可媲美甚至超越最强闭源模型。此外,即使在极低数据条件下(仅使用一个随机选择的无标签测试样本),TTRV在识别任务中仍能带来最高5.5%的非平凡提升。
原文摘要 · Abstract (English)
Existing methods for extracting reward signals in Reinforcement Learning typically rely on labeled data and dedicated training splits, a setup that contrasts with how humans learn directly from their environment. In this work, we propose TTRV to enhance vision language understanding by adapting the model on the fly at inference time, without the need for any labeled data. Concretely, we enhance the Group Relative Policy Optimization (GRPO) framework by designing rewards based on the frequency of the base model's output, while inferring on each test sample multiple times. Further, we also propose to control the diversity of the model's output by simultaneously rewarding the model for obtaining low entropy of the output empirical distribution. Our approach delivers consistent gains across both object recognition and visual question answering (VQA), with improvements of up to 52.4% and 29.8%, respectively, and average boosts of 24.6% and 10.0% across 16 datasets. Remarkably, on image recognition, TTRV applied to InternVL 8B surpasses GPT-4o by an average of 2.3% over 8 benchmarks, while remaining highly competitive on VQA, demonstrating that test-time reinforcement learning can match or exceed the strongest proprietary models. Finally, we find many interesting properties of test-time RL for VLMs: for example, even in extremely data-constrained scenarios, where adaptation is performed on a single randomly chosen unlabeled test example, TTRV still yields non-trivial improvements of up to 5.5% in recognition tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。