跳转至

Free-Grained Hierarchical Visual Recognition

会议: CVPR 2026
arXiv: 2510.14737
代码: FreeGrainLearning
领域: Others / Visual Recognition
关键词: 层级分类, 混合粒度标注, 半监督学习, 文本引导, 分类法

一句话总结

提出"自由粒度"层级视觉识别(free-grained hierarchical recognition),允许训练标签出现在分类法的任意层级,并提出文本引导伪属性和分类法引导半监督学习两种方法来弥补缺失监督,推理时模型自适应选择预测深度。

研究背景与动机

传统层级分类假设每张训练图像在分类法的所有层级都有完整标注(如 Bird → Bird of prey → Bald eagle),但现实中标注往往不整齐: - 内在原因:图像可能没有足够视觉证据支持细粒度标签(如远处只能看到"鸟"但分不清种类) - 外在原因:标注受成本、专业水平或标注协议限制

本文定义了自由粒度学习设置:训练标签可以出现在分类法的任意层级,且不同样本的标注深度可以不同。模型需要从这种不完整、混合粒度的监督中学习一致的层级预测。

实验显示现有 SOTA 层级分类方法(H-CAST)在从完整标注转向自由粒度设置时,Full-Path Accuracy 暴跌 19-40 个百分点(如 iNat21-mini 从 64.9% 降至 25.6%),证明该设置的挑战性。

方法详解

整体框架

工作包含三个主要组成: 1. 基准数据集构建:将现有层级数据集适配为自由粒度设置 2. 两种训练方法:文本引导伪属性(Text-Attr)和分类法引导半监督学习(Taxon-SSL) 3. 自由粒度推理:模型自适应选择预测深度

关键设计

  1. ImageNet-3L 数据集构建: ImageNet 原始 WordNet 层级不规则(5-19 层深度、30% 类别有多条路径),不适合层级评估。本文将其重新组织为干净的三级分类法(20 个 basic / 127 个 subordinate / 505 个 fine-grained),遵循认知心理学分层原则。设计原则包括:移除单子女路径、最大化组内多样性、细化模糊类别、用 LLM+人工审核验证。

  2. Foundation-based Pruning(基于基础模型的标签剪枝): 用 CLIP/BioCLIP 的零样本预测来模拟真实的混合粒度标注。从粗到细逐层检查预测是否正确:如果 subordinate 预测正确则保留,如果 fine-grained 也正确则保留。错误的层级标签被移除。ImageNet-F 中 32.6% 保留全部三级标签,28.0% 保留两级,39.4% 仅保留 basic。

  3. Text-Attr(文本引导伪属性): 核心观察:虽然不同层级的类别标签不同,但很多视觉属性(如"短腿""尖耳朵")跨层级一致。使用冻结的 VLM(Llama-3.2-11B)为图像生成文本描述,用 CLIP 文本编码器编码,再通过对比学习将图像特征与文本嵌入对齐。这种文本监督不依赖类别标签,能在细粒度标注缺失时提供额外语义线索。

  4. Taxon-SSL(分类法引导半监督学习): 将缺失层级标签视为无标签数据。关键创新是分类法对齐的亲和度图:只有当两个样本在所有层级的伪标签都一致时才视为正样本对(公式 3)。这能有效过滤掉噪声伪标签,确保层级一致性。在此基础上用对比损失拉近正样本对、推远负样本对。

损失函数 / 训练策略

  • 自由粒度层级损失: \(\mathcal{L}_{hier} = \sum_l \mathbb{1}_{y_l \text{ exists}} \cdot \mathcal{L}(f_l(x), y_l)\),仅在有标签的层级应用监督
  • 文本对比损失: InfoNCE 损失对齐图像-文本嵌入
  • 分类法对齐对比损失: 基于全层级一致伪标签的对比学习
  • 骨干网络:ViT-Small (H-ViT) 或 H-CAST
  • 训练 100 epoch(ImageNet-F 200 epoch)

实验关键数据

主实验

数据集 方法 FPA ↑ Fine ↑ Sub ↑ Basic ↑ TICE ↓
ImageNet-F H-CAST (full→free) 57.59 59.02 82.69 93.53 21.81
ImageNet-F Text-Attr (H-CAST) 63.20 64.91 84.47 93.56 18.58
ImageNet-F Taxon-SSL 48.40 52.34 65.74 82.96 19.87
iNat21-mini-F H-CAST 25.63 28.61 67.20 83.62 47.17
iNat21-mini-F Taxon-SSL + Text-Attr 31.93 37.08 69.76 82.20 37.04
iNat21-mini-F Taxon-SSL 31.74 37.11 69.53 82.02 37.31

消融实验

设置 关键指标 说明
H-CAST full → free (CUB) FPA: 84.9% → 45.1% 缺失标注导致 39.8pp 暴跌
H-CAST full → free (iNat) FPA: 64.9% → 25.6% 缺失标注导致 39.3pp 暴跌
Text-Attr 稀疏标签 优于 Taxon-SSL 文本在标签稀缺时弥补监督
Taxon-SSL 充足标签 优于 Text-Attr 有足够数据时 SSL 更有效

关键发现

  1. 现有方法在自由粒度设置下严重退化:H-CAST 的 FPA 下降 19-40 pp,证明该设置的研究必要性。

  2. Text-Attr 和 Taxon-SSL 各有优势:Text-Attr 在大规模多样化数据集(ImageNet-F)上表现更强,因文本描述提供了丰富的语义线索;Taxon-SSL 在细粒度生物数据集(iNat21-mini-F)上更优,因类间外观相似,视觉一致性更重要。

  3. 一致性优于置信度的推理策略:consistency-based stopping(当层级一致性被打破时停止)比 confidence-based stopping(当 softmax 置信度低于阈值时停止)产生更可靠、更深的正确预测,且不需要调节阈值。

  4. 文本引导改善语义聚焦:saliency map 显示 Text-Attr 帮助模型关注语义相关区域(如乐器而非人),而 Taxon-SSL 可能被视觉显著但语义无关区域误导。

亮点与洞察

  • 定义了一个重要的新设置:自由粒度层级识别。这比传统的完整层级标注假设更贴近现实。
  • ImageNet-3L 基准的构建本身就是有价值的贡献,为层级分类提供了大规模、干净的评估平台。
  • 两种方法的互补性是深刻的洞察:当标签稀缺时用外部语义知识(文本)弥补,当标签适中时用结构化 SSL 利用层级一致性。这为实际应用中选择策略提供了指导。
  • Consistency-based inference 是一个优雅的无参数推理策略。

局限与展望

  • 类级别和层级级别的不平衡未被显式处理
  • 标签剪枝依赖 CLIP,可能引入偏差,更好的剪枝方法(如基于多模型集成)有待探索
  • 提出的两种方法虽有效但提升仍有限(5-25%),表明自由粒度学习仍有很大改进空间
  • 未将方法扩展到更深的层级分类法(超过 3 级)
  • 推理时仅考虑何时"停止预测",未考虑层级间的信息传播和纠错

相关工作与启发

  • H-CAST (CVPR'23) 是层级分类 SOTA,鼓励跨层级的一致视觉分组
  • HRN (CVPR'22) 通过最大化树约束空间的边际概率处理多层级监督
  • CHMatch 将粗标签用于改进伪标签,但限于两级设置
  • 本文将长尾识别、半监督学习、弱监督学习和层级一致性统一在一个框架下

评分

  • 新颖性: ⭐⭐⭐⭐⭐ (定义了重要的新问题设置,数据集和方法均有创新)
  • 实验充分度: ⭐⭐⭐⭐⭐ (多数据集、多设置、详尽分析和可视化)
  • 写作质量: ⭐⭐⭐⭐⭐ (问题定义清晰,图表出色,组织结构好)
  • 价值: ⭐⭐⭐⭐⭐ (开辟新研究方向,提供基准和基线,实际意义强)

相关论文