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 ClassificationFood-101
Accuracy92
570
Image ClassificationFlowers102
Accuracy94
558
Image ClassificationRESISC45--
472
Image ClassificationFood101
Accuracy79.3
457
Image ClassificationCUB-200 2011
Accuracy80.8
374
Image ClassificationImageNet (test)
Top-1 Accuracy70.5
299
Image ClassificationOxford Flowers 102--
234
Image ClassificationImageNet
Acc76.5
45
Image ClassificationFGVC Aircraft
Accuracy57.9
39
Medical Image ClassificationHAM10000
Accuracy75.3
39
Showing 10 of 20 rows

Other info

Follow for update