atlas

Research map / Vision-language and multimodal / Human and embodied modeling

VLG-CBM: Training Concept Bottleneck Models with Vision-Language Guidance

NeurIPS 2024 · Vision-language and multimodal › Human and embodied modeling · approach: Concept bottleneck models · 21 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes VLG-CBM, which grounds LLM-generated concepts with Grounding DINO bounding boxes to train a concept bottleneck layer, and introduces a Number of Effective Concepts (NEC) metric to control information leakage.

Problem

Concept bottleneck models trained with LLM-generated concepts and VLM supervision suffer from concept predictions that mismatch images and from information leakage, where random concept layers reach high accuracy.

Approach

Ground candidate concepts on training images with an open-vocabulary detector (Grounding DINO Swin-B) to create concept-labeled data for training the CBL, and evaluate CBMs by accuracy at a fixed number of effective concepts (NEC) enforced via a sparse final layer, supported by a theorem that random CBLs approximate linear classifiers as concept count grows.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use VLG-CBM instead of the strongest reported baselineCBMs with CLIP-RN50 · CIFAR10, CIFAR100, ImageNet, and CUBANEC-5: at least 4.27% and up to 51.09%Helped
use VLG-CBM instead of the strongest reported baselineCBMs with CLIP-RN50 · CIFAR10, CIFAR100, ImageNet, and CUBANEC-avg: at least 0.45% and up to 29.78%Helped
use VLG-CBM instead of LF-CBMCBMs with non-CLIP backbones · CUB, Places365, and ImageNetANEC-5Helped
set confidence threshold T to 0.10 rather than 0.20 (ablation)VLG-CBM · CUB200ANEC-5: 75.73% -> 75.75%Helped
set confidence threshold T to 0.10 rather than 0.20 (ablation)VLG-CBM · Places365ANEC-5: 41.25% -> 41.84%Helped
set crop-to-concept augmentation probability to 0.2 rather than 0.0 (ablation)VLG-CBM · CUBANEC-5: 75.73 -> 75.83Helped
set crop-to-concept augmentation probability to 0.4 rather than 0.2 (ablation)VLG-CBM · CUBANEC-5: 75.83 -> 75.71Hurt
set crop-to-concept augmentation probability to 0.6 rather than 0.2 (ablation)VLG-CBM · CUBANEC-5: 75.83 -> 75.57Hurt
set crop-to-concept augmentation probability to 0.8 rather than 0.2 (ablation)VLG-CBM · CUBANEC-5: 75.83 -> 75.52Hurt
set crop-to-concept augmentation probability to 1.0 rather than 0.2 (ablation)VLG-CBM · CUBANEC-5: 75.83 -> 72.29Hurt

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR10 CIFAR100 CUB Places365 ImageNet Waterbirds

Benchmarks: CIFAR10 CIFAR100 CUB Places365 ImageNet Waterbirds

Metrics: ANEC-5 ANEC-avg accuracy precision recall percent of changed decisions after top-5 pruning

Baselines: LF-CBM LaBo LM4CV Random CBL

Builds on: Label-free CBM (LF-CBM) GLM-SAGA Grounding DINO

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy