← 返回 2026-03-05

面向细粒度开放世界分类的特异性感知强化学习 Specificity-aware reinforcement learning for fine-grained open-world classification

Samuele Angheben, Davide Berasi, Alessandro Conti, Elisa Ricci, Yiming Wang 📅 2026-03-03 👍 16 2026-07-13 08:35
图像分类 多模态学习 大语言模型 开放世界识别 强化学习

通过动态奖励的强化学习提升多模态模型在开放世界分类中的特异性

前置知识

开放世界图像分类

与传统封闭世界分类不同,开放世界分类不预定义固定的类别标签集。模型需要能够识别任意语义概念,包括训练时未见过的类别。这种设定更贴近现实应用,因为现实世界中新的概念和类别会不断出现,模型需要具备泛化到未知类别的能力。

本文的核心任务场景,理解开放世界与封闭世界的区别是理解论文动机的基础。

大型多模态模型(LMM)

结合视觉编码器和大型语言模型的架构,能够处理图像和文本的联合理解与生成。这类模型通过大规模预训练获得强大的视觉-语言对齐能力,可以进行开放式的视觉问答和推理。代表模型包括Qwen2.5VL、InternVL等。

本文研究的对象就是这类模型在分类任务中的行为,特别是它们的推理能力。

强化学习与可验证奖励(RLVR)

一种使用规则或程序可验证的奖励信号来训练模型的方法。与依赖人类偏好的RLHF不同,RLVR通过直接检查模型输出与正确答案来计算奖励,特别适合数学、编程等有明确正确答案的任务。GRPO(Group Relative Policy Optimization)是其中一种高效算法。

本文的技术基础,需要理解RLVR如何工作以及为什么适合本文的设定。

特异性(Specificity)与正确性(Correctness)

特异性衡量预测的精细程度,例如预测'鸟'的特异性低于预测'金翅莺'。正确性衡量预测是否正确。在开放世界中,一个预测可以是正确但泛化的(如预测'花'而实际是'雏菊'),也可以是精细但错误的。本文的核心挑战是如何同时提升两者。

这是本文定义的两个核心评估指标,理解它们的权衡关系是理解论文贡献的关键。

Best-of-N(BoN)采样

对同一输入进行N次独立采样,然后选择其中最佳的一个结果作为最终输出。这种方法可以近似估计模型的上限能力,因为即使单次采样可能失败,多次采样中总有可能产生正确结果。在本文中,BoN-64被用作模型潜在能力的参考基准。

用于分析模型是否具备更精细的预测能力,以及作为SpeciaRL方法设计的灵感来源。

研究动机

大型多模态模型(如Qwen2.5VL)在开放世界图像分类任务中表现出明显的泛化倾向。当被问及'图像中的主要物体是什么'时,模型倾向于给出过于笼统的答案。例如,面对一张雏菊的图片,模型可能只回答'花'而不是'雏菊';面对一只金翅莺,可能只回答'鸟'而不是具体的物种。这种现象在细粒度分类任务中尤为严重,研究表明在非常细粒度的数据集(如StanfordCars、FGVCAircraft)上,模型产生泛化预测的比例高达74.5%。现有方法试图通过提示工程(如要求模型'be specific')或监督微调来改善这一问题,但往往会损害预测的正确性,导致更多错误输出。

本文的目标是本文的目标是设计一种有效的方法,能够在不损害预测正确性的前提下,提升大型多模态模型在开放世界细粒度图像分类中的特异性。具体而言,希望模型能够给出尽可能精细但同时正确的预测,例如对于雏菊图片应该预测'雏菊'而不是'花',对于金翅莺应该预测具体的鸟类物种而不是泛泛的'鸟'。目标是在特异性和正确性之间达到最佳平衡,用调和平均数(HM)来综合衡量这一平衡。

与已有工作不同的是,本文的独特视角在于对模型行为的深入分析。作者发现,模型之所以产生泛化预测,并非因为缺乏领域知识。通过Best-of-64采样实验,模型在64次尝试中能找到更精细且正确的预测,这表明模型内部确实具备相关的精细知识,只是在单次推理中无法可靠地表达出来。这一发现启发了一种新的思路:与其向模型注入新知识,不如通过强化学习改善模型的'采样效率',引导其在能力范围内产生最精细的预测。

核心方法

SpeciaRL的核心思想可以用一个类比来理解:想象一个学生在考试中,他知道答案的细节,但总是倾向于给出保守的、笼统的答案。与其惩罚他答错(这会让他更不敢尝试细节),不如根据他能力范围内能达到的最佳水平来设定奖励标准。如果他在多次尝试中能答出'金翅莺',那么只答'鸟'就应该被扣分;但如果他最多只能答出'鸟',那么答'鸟'就应该被奖励。技术路线上,SpeciaRL采用GRPO算法框架,对每个样本生成N个预测,然后通过LLM评判器对每个预测进行分类(错误、泛化、较少特异性、特异性、更特异性等),最后基于当前模型的最佳表现动态调整奖励信号。

与现有方法最本质的区别在于奖励设计的动态性和样本特异性。传统RLVR方法使用静态奖励:只有完全匹配才给奖励1,否则给0。这种方法在开放世界中过于严格,因为正确但不够精细的预测也会被惩罚,导致模型为了安全而更倾向于泛化。SpeciaRL的创新在于:对于每个训练样本,先用当前模型进行N次采样,找到能达到的最高特异性水平作为'参考水平',然后只有当预测达到或超过这个参考水平时才给予正奖励。这样,模型被鼓励在能力范围内追求最大特异性,而不是被鼓励去冒险追求超出能力的精细预测。

方法步骤详情

SpeciaRL的完整流程包括:1)对于每个训练样本(图像I和标签y),使用当前模型生成N个预测{p1,...,pN};2)使用LLM评判器(Qwen3-30B)对每个预测进行分类,得到类别c∈{W(错误), A(放弃), G(泛化), S−(较少特异性), S(特异性), S+(更特异性)};3)在N个预测中找到最佳类别cbest;4)根据cbest确定参考水平c*:如果cbest=S+则c*=S,如果cbest=W则c*=A,否则c*=cbest;5)计算奖励:如果预测的类别c满足c⪰c*则奖励为1,否则为0;6)使用GRPO算法更新模型参数。整个过程是在线的,奖励基于当前模型的能力动态调整。

技术新颖性

SpeciaRL的技术新颖性体现在三个方面:首先是动态奖励机制,这是首次在开放世界分类中引入基于模型当前能力的自适应奖励;其次是样本特异性的奖励设计,不同样本根据其难度有不同的奖励标准,避免了'一刀切'的静态奖励;最后是与GRPO的无缝集成,利用GRPO本身的N个rollouts来估计模型能力,不需要额外的采样开销。与Visual-RFT等现有方法相比,SpeciaRL不要求预测完全匹配标签,而是允许一定程度的语义近似,这更符合开放世界的实际情况。

Qwen2.5VL-7B模型在细粒度和非常细粒度数据集上的预测分布
Figure 2: Qwen2.5VL-7B模型在细粒度和非常细粒度数据集上的预测分布
SpeciaRL方法框架概览
Figure 3: SpeciaRL方法框架概览

实验结果

实验结果表明SpeciaRL在特异性和正确性之间达到了最佳平衡。在细粒度数据集平均上,SpeciaRL的特异性为0.920,正确性为0.848,调和平均数(HM)为0.883,显著优于基础模型Qwen2.5VL-7B(HM=0.790)和其他基线方法。在非常细粒度数据集上,SpeciaRL同样表现最佳,HM达到0.830,而基础模型仅为0.692。值得注意的是,SpeciaRL不仅提升了特异性,还同时提升了正确性,这与'be specific'提示方法形成鲜明对比——后者虽然提升了特异性(0.816),但损害了正确性(0.832),最终HM仅为0.822。在跨域泛化方面,尽管只在鸟类数据集(CUB)上训练,SpeciaRL在花卉、食物、宠物、汽车、飞机等完全不同的领域都展现出强大的泛化能力。与监督微调(SFT)和标准强化微调(RFT)相比,SpeciaRL在大多数评估指标上都表现更优。

开放世界图像分类主要结果对比
Table 1: 开放世界图像分类主要结果对比
不同静态奖励规则与SpeciaRL动态奖励的对比
Table 3: 不同静态奖励规则与SpeciaRL动态奖励的对比
基础模型与SpeciaRL的定性比较示例
Figure 4: 基础模型与SpeciaRL的定性比较示例
查看结构化数据
任务指标本文基线提升
细粒度分类(平均) 调和平均数(HM) 0.883 Qwen2.5VL-7B (0.790) +11.8%
非常细粒度分类(平均) 调和平均数(HM) 0.830 Qwen2.5VL-7B (0.692) +19.9%
Flowers102花卉分类 调和平均数(HM) 0.934 Qwen2.5VL-7B (0.823) +13.5%
StanfordCars汽车分类 调和平均数(HM) 0.818 Qwen2.5VL-7B (0.693) +18.0%
通用评估协议(细粒度) 概念相似度(CS) 81.1 Qwen2.5VL-7B (78.9) +2.8%

局限与改进

尽管SpeciaRL取得了显著成果,但仍存在一些局限性。首先,训练和评估使用不同的LLM评判器(训练用Qwen3-30B,评估用Llama3-72B),虽然这是为了避免家族偏差,但评判器之间的不一致性可能影响评估的可靠性。其次,方法在某些情况下会产生'过度特异性'的预测,例如对普通家猫预测'Felis Catus'(学名)而不是'Cat',这种预测虽然在技术上是正确的,但可能不符合人类的使用习惯。第三,在非常细粒度的任务上(如StanfordCars),SpeciaRL的正确性提升相对有限(0.918 vs 0.906),表明在极端细粒度场景下,模型能力可能已接近上限。此外,论文只在7B规模的模型上进行了实验,更大规模模型的表现尚不清楚。

独立分析的弱点

从技术角度分析,SpeciaRL存在几个可以改进的弱点。首先,当前的奖励设计完全依赖LLM评判器进行分类,但评判器本身可能存在偏差或错误,特别是在处理领域特定术语时。改进方向可以是引入多评判器投票机制或人类验证环节。其次,Best-of-N采样在训练阶段增加了计算开销,虽然利用了GRPO本身的rollouts,但N=10的设置可能限制了能力估计的准确性。可以探索更高效的能力估计方法,例如使用小型代理模型或缓存历史预测。第三,方法对训练数据的选择敏感,论文只在CUB鸟类数据集上训练,但不同领域的迁移效果可能不同。可以研究自动化的训练数据选择策略。

未来方向

基于本文的成果,有几个有前景的研究方向。首先,可以将SpeciaRL扩展到其他多模态任务,如视觉问答、图像描述生成等,这些任务同样面临特异性和准确性的权衡。其次,可以研究更细粒度的奖励设计,例如引入连续的特异性分数而不是离散的类别,或者考虑概念之间的层次关系。第三,可以探索SpeciaRL与人类反馈的结合,让人类用户直接定义什么是'足够特异',使系统更符合实际应用需求。最后,可以研究SpeciaRL在更大规模模型(如70B参数)上的表现,以及在多语言环境下的泛化能力。

复现评估

论文的复现条件相对友好。作者公开了代码和模型(GitHub: s-angheben/SpeciaRL),使用的数据集(CUB、Flowers102等)都是公开可用的标准数据集。训练使用6块A100 GPU,总训练时间约12小时,这对于学术研究来说是可接受的。主要的技术挑战在于LLM评判器的部署(需要Qwen3-30B模型)和GRPO算法的实现(论文使用了Verl框架)。对于计算资源有限的研究者,可以考虑使用更小的评判器模型或减少rollout数量N,但可能会影响性能。总体而言,复现难度中等,主要瓶颈在于GPU资源和LLM推理效率。