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

Towards Reasonable Concept Bottleneck Models

About

We propose a novel, flexible, and efficient framework for designing Concept Bottleneck Models (CBMs) that enables practitioners to explicitly encode and extend their prior knowledge and beliefs about the concept-concept ($C-C$) and concept-task ($C \to Y$) relationships within the model's reasoning when making predictions. The resulting $\textbf{C}$oncept $\textbf{REA}$soning $\textbf{M}$odels (CREAMs) architecturally encode arbitrary types of $C-C$ relationships such as mutual exclusivity, hierarchical associations, and/or correlations, as well as potentially sparse $C \to Y$ relationships. Moreover, CREAM can optionally incorporate a regularized side-channel to complement the potentially {incomplete concept sets}, achieving competitive task performance while encouraging predictions to be concept-grounded. To evaluate CBMs in such settings, we introduce a $C \to Y$ agnostic metric that quantifies interpretability when predictions partially rely on the side-channel. In our experiments, we show that, without additional computational overhead, CREAM models support efficient interventions, can avoid concept leakage, and achieve black-box-level performance under missing concepts. We further analyze how an optional side-channel affects interpretability and intervenability. Importantly, the side-channel enables CBMs to remain effective even in scenarios where only a limited number of concepts are available.

Nektarios Kalampalikis, Kavya Gupta, Georgi Vitanov, Isabel Valera• 2025

Related benchmarks

TaskDatasetResultRank
ClassificationCelebA
Avg Accuracy80.92
185
ClassificationCUB--
93
Image ClassificationCelebA--
42
Image ClassificationiFMNIST
Peak Memory1
12
Image ClassificationcFMNIST
Peak Memory Usage1
12
ClassificationiFMNIST
Accuracy (Y)92.43
11
ClassificationcFMNIST
Accuracy (Y)92.38
11
Image ClassificationCelebA
Training Time1
8
Concept-based LearningCelebA
Training Time1.002
7
Concept-based LearningiFMNIST
Training Time1.585
7
Showing 10 of 13 rows

Other info

Follow for update