English

Towards Reasonable Concept Bottleneck Models

Machine Learning 2026-04-14 v2 Artificial Intelligence Machine Learning

Abstract

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 (CCC-C) and concept-task (CYC \to Y) relationships within the model's reasoning when making predictions. The resulting C\textbf{C}oncept REA\textbf{REA}soning M\textbf{M}odels (CREAMs) architecturally encode arbitrary types of CCC-C relationships such as mutual exclusivity, hierarchical associations, and/or correlations, as well as potentially sparse CYC \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 CYC \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.

Keywords

Cite

@article{arxiv.2506.05014,
  title  = {Towards Reasonable Concept Bottleneck Models},
  author = {Nektarios Kalampalikis and Kavya Gupta and Georgi Vitanov and Isabel Valera},
  journal= {arXiv preprint arXiv:2506.05014},
  year   = {2026}
}

Comments

32 pages, 20 figures

R2 v1 2026-07-01T03:01:30.522Z