arXiv:2602.07605cs.CVcs.AI2026-02被引 11

用思维链训练让多模态大模型精准识别细粒度视觉类别

Fine-R1: Make Multi-modal LLMs Excel in Fine-Grained Visual Recognition by Chain-of-Thought Reasoning

  • 构建带推理过程的细粒度视觉数据集,引导模型逐步分析图像特征
  • 仅用4样本训练即超越通用大模型和对比学习模型,对未见类别也有效
  • 适合需要专家标注但类别繁多的知识密集型场景

视觉世界中的实体可按共性分层归类至细粒度子类别。尽管多模态大语言模型(MLLMs)在粗粒度视觉任务上表现良好,但在细粒度视觉识别(FGVR)上仍存在困难。现有方法需大量标注数据,成本高,且模型易过拟合已见类别,泛化能力差。为此,我们提出Fine-R1,基于R1训练框架:(1) 思维链监督微调,构建包含‘视觉分析、候选类别、对比推理、预测’的高质量CoT数据集,使模型具备强开放世界分类能力;(2) 三元组增强策略优化,通过类内增强融合同类图像轨迹提升抗类内差异能力,类间增强强化跨类别响应区分度,增强判别力。仅需4样本训练,Fine-R1在已见与未见子类别识别上均优于现有通用MLLM、推理型MLLM及对比学习模型(如CLIP),在难以获取全部类别专家标注的知识密集型领域具有应用潜力。代码已开源。

原文摘要 · Abstract (English)

Any entity in the visual world can be hierarchically grouped based on shared characteristics and mapped to fine-grained sub-categories. While Multi-modal Large Language Models (MLLMs) achieve strong performance on coarse-grained visual tasks, they often struggle with Fine-Grained Visual Recognition (FGVR). Adapting general-purpose MLLMs to FGVR typically requires large amounts of annotated data, which is costly to obtain, leaving a substantial performance gap compared to contrastive CLIP models dedicated for discriminative tasks. Moreover, MLLMs tend to overfit to seen sub-categories and generalize poorly to unseen ones. To address these challenges, we propose Fine-R1, an MLLM tailored for FGVR through an R1-style training framework: (1) Chain-of-Thought Supervised Fine-tuning, where we construct a high-quality FGVR CoT dataset with rationales of "visual analysis, candidate sub-categories, comparison, and prediction", transition the model into a strong open-world classifier; and (2) Triplet Augmented Policy Optimization, where Intra-class Augmentation mixes trajectories from anchor and positive images within the same category to improve robustness to intra-class variance, while Inter-class Augmentation maximizes the response distinction conditioned on images across sub-categories to enhance discriminative ability. With only 4-shot training, Fine-R1 outperforms existing general MLLMs, reasoning MLLMs, and even contrastive CLIP models in identifying both seen and unseen sub-categories, showing promise in working in knowledge-intensive domains where gathering expert annotations for all sub-categories is arduous. Code is available at https://github.com/PKU-ICST-MIPL/FineR1_ICLR2026.

细粒度识别多模态思维链小样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。