用视觉反馈强化大模型,少样本下也能精准识别图像。
Visual-RFT: Visual Reinforcement Fine-Tuning

- 用多路推理生成+视觉可验证奖励函数优化模型。
- 少样本图像分类准确率提升24.3%,目标检测提升超15%。
- 适合数据稀缺场景,尤其适合需要推理的视觉任务。
大型推理模型如OpenAI o1通过答案反馈进行强化微调(RFT),在标注数据稀缺时尤为有效。近期开源工作DeepSeek-R1表明,基于可验证奖励的强化学习是复现o1的关键方向。尽管该方法在语言模型中取得成功,其在多模态领域的应用仍不充分。本文提出视觉强化微调(Visual-RFT),将RFT拓展至视觉任务。具体而言,先用大视觉语言模型(LVLM)生成含推理过程与最终答案的多个响应,再通过我们设计的视觉可验证奖励函数(如目标检测中的交并比IoU)结合组相对策略优化(GRPO)更新模型。针对不同感知任务设计专属奖励函数。在细粒度图像分类、少样本目标检测、推理定位及开放词汇目标检测等基准测试中,Visual-RFT表现优于监督微调(SFT)。例如,在仅100样本的一次样本细粒度分类中,准确率提升24.3%;在少样本目标检测上,于COCO两样本设置中提升21.9,在LVIS上提升15.4。Visual-RFT代表了微调大视觉语言模型的新范式,提供数据高效、奖励驱动的方法,显著增强模型在特定任务中的推理与适应能力。
原文摘要 · Abstract (English)
Reinforcement Fine-Tuning (RFT) in Large Reasoning Models like OpenAI o1 learns from feedback on its answers, which is especially useful in applications when fine-tuning data is scarce. Recent open-source work like DeepSeek-R1 demonstrates that reinforcement learning with verifiable reward is one key direction in reproducing o1. While the R1-style model has demonstrated success in language models, its application in multi-modal domains remains under-explored. This work introduces Visual Reinforcement Fine-Tuning (Visual-RFT), which further extends the application areas of RFT on visual tasks. Specifically, Visual-RFT first uses Large Vision-Language Models (LVLMs) to generate multiple responses containing reasoning tokens and final answers for each input, and then uses our proposed visual perception verifiable reward functions to update the model via the policy optimization algorithm such as Group Relative Policy Optimization (GRPO). We design different verifiable reward functions for different perception tasks, such as the Intersection over Union (IoU) reward for object detection. Experimental results on fine-grained image classification, few-shot object detection, reasoning grounding, as well as open-vocabulary object detection benchmarks show the competitive performance and advanced generalization ability of Visual-RFT compared with Supervised Fine-tuning (SFT). For example, Visual-RFT improves accuracy by $24.3\%$ over the baseline in one-shot fine-grained image classification with around 100 samples. In few-shot object detection, Visual-RFT also exceeds the baseline by $21.9$ on COCO's two-shot setting and $15.4$ on LVIS. Our Visual-RFT represents a paradigm shift in fine-tuning LVLMs, offering a data-efficient, reward-driven approach that enhances reasoning and adaptability for domain-specific tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。