English

Zero-shot Concept Bottleneck Models

Machine Learning 2026-04-06 v2 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Concept bottleneck models (CBMs) are inherently interpretable and intervenable neural network models, which explain their final label prediction by the intermediate prediction of high-level semantic concepts. However, they require target task training to learn input-to-concept and concept-to-label mappings, incurring target dataset collections and training resources. In this paper, we present zero-shot concept bottleneck models (Z-CBMs), which predict concepts and labels in a fully zero-shot manner without training neural networks. Z-CBMs utilize a large-scale concept bank, which is composed of millions of vocabulary extracted from the web, to describe arbitrary input in various domains. For the input-to-concept mapping, we introduce concept retrieval, which dynamically finds input-related concepts by the cross-modal search on the concept bank. In the concept-to-label inference, we apply concept regression to select essential concepts from the retrieved concepts by sparse linear regression. Through extensive experiments, we confirm that our Z-CBMs provide interpretable and intervenable concepts without any additional training. Code will be available at https://github.com/yshinya6/zcbm.

Keywords

Cite

@article{arxiv.2502.09018,
  title  = {Zero-shot Concept Bottleneck Models},
  author = {Shin'ya Yamaguchi and Kosuke Nishida and Daiki Chijiwa and Yasutoshi Ida},
  journal= {arXiv preprint arXiv:2502.09018},
  year   = {2026}
}

Comments

Accepted to IEEE ICME 2026

R2 v1 2026-06-28T21:42:40.165Z