paper-with-me

Papers

Concept Bottleneck Large Language Models

2024-12-11 · Chung-En Sun, Tuomas Oikarinen, Berk Ustun, Tsui-Wei Weng

We introduce the Concept Bottleneck Large Language Model (CB-LLM), a pioneering approach to creating inherently interpretable Large Language Models (LLMs). Unlike traditional black-box LLMs that rely on post-hoc interpretation methods with limited neuron function insights, CB-LLM sets a new standard with its built-in interpretability, scalability, and ability to provide clear, accurate explanations. We investigate two essential tasks in the NLP domain: text classification and text generation. In text classification, CB-LLM narrows the performance gap with traditional black-box models and provides clear interpretability. In text generation, we show how interpretable neurons in CB-LLM can be used for concept detection and steering text generation. Our CB-LLMs enable greater interaction between humans and LLMs across a variety of tasks -- a feature notably absent in existing LLMs. Our code is available at https://github.com/Trustworthy-ML-Lab/CB-LLMs.

📄 PDF Abstract BibTeX arXiv:2412.07992

Code (1)

trustworthy-ml-lab/cb-llms 공식 구현 pytorch

Tasks

Language ModelingLanguage ModellingLarge Language Modeltext-classificationText ClassificationText Generation

Similar Papers 제목 키워드 기반

Language in a Bottle: Language Model Guided Concept Bottlenecks for Interpretable Image Classification

2022-11-21 · CVPR 2023 1 · Yue Yang, Artemis Panagopoulou, Shenghao Zhou, Daniel Jin 외

Concept Bottleneck Models (CBM) are inherently interpretable models that factor model decisions into human-readable concepts. They allow people to easily understand why a model is failing, a critical feature for high-sta…

image-classificationImage ClassificationLanguage ModelingLanguage Modelling

Concept Bottleneck Language Models For protein design

2024-11-09 · Aya Abdelsalam Ismail, Tuomas Oikarinen, Amy Wang, Julius Adebayo 외

We introduce Concept Bottleneck Protein Language Models (CB-pLM), a generative masked language model with a layer where each neuron corresponds to an interpretable concept. Our architecture offers three key benefits: i) …

Decision MakingDrug DiscoveryLanguage ModelingLanguage Modelling+1

VLG-CBM: Training Concept Bottleneck Models with Vision-Language Guidance

2024-07-18 · Divyansh Srivastava, Ge Yan, Tsui-Wei Weng

Concept Bottleneck Models (CBMs) provide interpretable prediction by introducing an intermediate Concept Bottleneck Layer (CBL), which encodes human-understandable concepts to explain models' decision. Recent works propo…

Avg

V2C-CBM: Building Concept Bottlenecks with Vision-to-Concept Tokenizer

2025-01-09 · Hangzhou He, Lei Zhu, Xinliang Zhang, Shuang Zeng 외

Concept Bottleneck Models (CBMs) offer inherent interpretability by initially translating images into human-comprehensible concepts, followed by a linear combination of these concepts for classification. However, the ann…

Uncertainty-aware Language Guidance for Concept Bottleneck Models

2026-02-26 · Yangyi Li, Mengdi Huai arxiv

Concept Bottleneck Models (CBMs) provide inherent interpretability by first mapping input samples to high-level semantic concepts, followed by a combination of these concepts for the final classification. However, the an…