Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DCBM: Data-Efficient Visual Concept Bottleneck Models

About

Concept Bottleneck Models (CBMs) enhance the interpretability of neural networks by basing predictions on human-understandable concepts. However, current CBMs typically rely on concept sets extracted from large language models or extensive image corpora, limiting their effectiveness in data-sparse scenarios. We propose Data-efficient CBMs (DCBMs), which reduce the need for large sample sizes during concept generation while preserving interpretability. DCBMs define concepts as image regions detected by segmentation or detection foundation models, allowing each image to generate multiple concepts across different granularities. This removes reliance on textual descriptions and large-scale pre-training, making DCBMs applicable for fine-grained classification and out-of-distribution tasks. Attribution analysis using Grad-CAM demonstrates that DCBMs deliver visual concepts that can be localized in test images. By leveraging dataset-specific concepts instead of predefined ones, DCBMs enhance adaptability to new domains.

Katharina Prasse, Patrick Knab, Sascha Marton, Christian Bartelt, Margret Keuper• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 (test)
Accuracy88.7
1063
Image ClassificationImageNet 1k (test)
Top-1 Accuracy80.2
939
Image ClassificationFood-101
Accuracy92
590
Image ClassificationFlowers102
Accuracy94
558
Image ClassificationRESISC45--
539
Image ClassificationFood101
Accuracy79.3
457
Image ClassificationCUB-200 2011
Accuracy80.8
381
Image ClassificationCIFAR-100
Accuracy86.4
362
Image ClassificationCUB
Accuracy81
351
Image ClassificationCUB-200-2011 (test)
Top-1 Acc81
316
Showing 10 of 34 rows

Other info

Follow for update