arXiv:2501.04975cs.CV2025-01AAAI被引 14

用视觉模型直接生成可解释的图像概念,提升分类准确率与可读性。

V2C-CBM: Building Concept Bottlenecks with Vision-to-Concept Tokenizer

  • 用通用词汇构建概念词典,通过无标签图像训练视觉到概念的映射
  • 在多个图像分类任务上达到或超过大语言模型指导的模型性能
  • 概念更聚焦视觉特征,避免冗余信息,适合需要可解释性的场景

概念瓶颈模型(CBMs)通过将图像转化为人类可理解的概念再进行分类,实现内在可解释性。但人工标注概念成本高,限制了其广泛应用。现有方法利用大语言模型构建概念,虽能自动建模,但生成概念常冗长且包含非视觉属性,影响准确率和可解释性。本文提出直接从多模态模型中构建CBM的方法:以常见词汇为基础概念词表,借助无标签图像训练一个视觉到概念(V2C)分词器,显式将图像量化为最相关的视觉概念,形成紧耦合于多模态模型的视觉导向概念瓶颈。由此提出的V2C-CBM兼具训练高效、高准确率与强可解释性。在多个视觉分类基准上,其性能匹配甚至超越由大语言模型监督的CBMs,验证了该方法的有效性。

原文摘要 · Abstract (English)

Concept Bottleneck Models (CBMs) offer inherent interpretability by initially translating images into human-comprehensible concepts, followed by a linear combination of these concepts for classification. However, the annotation of concepts for visual recognition tasks requires extensive expert knowledge and labor, constraining the broad adoption of CBMs. Recent approaches have leveraged the knowledge of large language models to construct concept bottlenecks, with multimodal models like CLIP subsequently mapping image features into the concept feature space for classification. Despite this, the concepts produced by language models can be verbose and may introduce non-visual attributes, which hurts accuracy and interpretability. In this study, we investigate to avoid these issues by constructing CBMs directly from multimodal models. To this end, we adopt common words as base concept vocabulary and leverage auxiliary unlabeled images to construct a Vision-to-Concept (V2C) tokenizer that can explicitly quantize images into their most relevant visual concepts, thus creating a vision-oriented concept bottleneck tightly coupled with the multimodal model. This leads to our V2C-CBM which is training efficient and interpretable with high accuracy. Our V2C-CBM has matched or outperformed LLM-supervised CBMs on various visual classification benchmarks, validating the efficacy of our approach.

概念瓶颈可解释性多模态视觉分词

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