Our new X account is live! Follow @wizwand_team for updates
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
494
Image ClassificationFlowers102
Accuracy94
478
Image ClassificationFood101
Accuracy79.3
309
Image ClassificationImageNet (test)
Top-1 Accuracy70.5
291
Image ClassificationRESISC45--
263
Image ClassificationCUB-200 2011
Accuracy80.8
257
Image ClassificationOxford Flowers 102--
172
Image ClassificationImageNet
Acc76.5
45
Medical Image ClassificationHAM10000
Accuracy75.3
39
Image ClassificationFGVC Aircraft
Accuracy57.9
32
Showing 10 of 20 rows

Other info

Follow for update