Measuring What Matters: Synthetic Benchmarks for Concept Bottleneck Models

TL;DR

Synthetic benchmarks control data properties to evaluate concept bottleneck models' robustness and interpretability.

cs.LG 🔴 Advanced 2026-06-03 48 views
Julian Skirzynski Harry Cheon Shreyas Kadekodi Meredith Stewart Berk Ustun
Concept Bottleneck Models Synthetic Data Model Evaluation Interpretability Benchmarks

Key Findings

Methodology

This paper develops a synthetic data generation framework to evaluate CBMs across decision support and automation tasks. By tuning parameters such as data modality, concept relevance, annotation noise, and completeness, diverse datasets are created. Multiple architectures (CBM, CEM, ProbCBM, ECBM) are compared under varying conditions. Evaluation metrics include accuracy improvements, intervention effects, and coverage, enabling systematic diagnosis of failure modes and robustness. The framework facilitates controlled experiments to analyze how models respond to concept noise, incomplete labels, and different concept sources.

Key Results

  • In decision support, CBMs with interventions improved accuracy from 79.9% to 94.9%, outperforming black-box baselines. For automation, concept interventions increased automatic coverage by up to 20%. ProbCBM showed robustness under noisy and incomplete concept labels, maintaining high performance. Different architectures exhibited varying sensitivities to concept quality and noise, with CEM outperforming others in noisy environments. The synthetic benchmarks effectively revealed failure modes related to concept mis-specification, label noise, and architecture limitations, guiding targeted improvements.
  • Experiments demonstrated that synthetic data allows precise control over variables like concept relevance, modality, and noise levels. Results confirmed that models like CEM are more robust to concept definition biases, while ECBM struggles with partial interventions. The benchmarks highlighted the importance of concept quality and source, showing how automated concept annotations from LLM or CLIP impact performance. Overall, the framework provides a comprehensive toolset for evaluating CBMs' real-world applicability and limitations.

Significance

This work introduces a systematic, controllable evaluation framework for concept bottleneck models, addressing the limitations of real datasets constrained by annotation costs and biases. By enabling detailed analysis of model performance under diverse, simulated conditions, it advances understanding of model robustness, interpretability, and failure modes. The synthetic benchmarks facilitate rapid prototyping and testing, accelerating development of reliable, explainable AI systems for critical applications such as healthcare diagnostics, autonomous vehicles, and industrial inspection. This approach bridges the gap between theoretical research and practical deployment, fostering more trustworthy AI systems.

Technical Contribution

The paper proposes a multi-parameter synthetic data generator that allows precise control over data modality, concept relevance, noise, and annotation quality. It integrates multiple model architectures—CBM, CEM, ProbCBM, ECBM—and evaluates them across a suite of metrics. The framework supports scenario-specific testing, including concept mis-specification, label noise, and incomplete annotations. The methodology enables ablation studies, robustness analysis, and intervention effectiveness assessment, providing new insights into architecture-specific strengths and weaknesses. These innovations significantly extend the evaluation toolkit for concept-based models.

Novelty

This is the first comprehensive framework employing synthetic, parameterized data to evaluate CBMs across diverse scenarios systematically. Unlike prior work limited to real datasets with fixed labels, this approach offers fine-grained control over data properties, enabling targeted analysis of failure modes. The integration of multiple architectures and intervention strategies within a unified, reproducible platform represents a novel contribution, facilitating rigorous benchmarking and model comparison under controlled conditions. It sets a new standard for evaluating interpretability-focused models.

Limitations

  • Synthetic data, despite its controllability, may not fully capture the complexity and variability of real-world data, limiting direct transferability of results.
  • While the framework reveals model weaknesses under simulated conditions, real-world deployment may introduce unforeseen challenges not covered by synthetic scenarios.
  • Computational costs for large-scale scenario testing and multi-architecture evaluation remain significant, requiring further optimization for practical use.

Future Work

Future directions include integrating real-world datasets with synthetic benchmarks for hybrid evaluation, enhancing data complexity to better mimic real scenarios, and developing adaptive intervention strategies. Extending the framework to multi-modal, multi-task settings can broaden its applicability. Additionally, exploring active learning and human-in-the-loop approaches within this synthetic environment could improve model robustness and interpretability, accelerating deployment in high-stakes domains.

AI Executive Summary

The rapid advancement of AI has brought about remarkable capabilities but also raised concerns over interpretability and robustness, especially in high-stakes applications like healthcare and autonomous driving. Traditional evaluation methods rely heavily on real datasets, which are often limited by high annotation costs, bias, and lack of controllability. To address these challenges, this paper introduces a synthetic benchmarking framework designed specifically for concept bottleneck models (CBMs). By generating customizable datasets with adjustable properties such as data modality, concept relevance, noise levels, and annotation completeness, the framework enables systematic, scenario-specific evaluation.

This approach allows researchers to test multiple CBM architectures—including CBM, CEM, ProbCBM, and ECBM—under controlled conditions. Experiments demonstrate that interventions—human corrections or manipulations—can significantly improve model accuracy and automation coverage, with improvements up to 15-20%. The benchmarks reveal how different models respond to concept noise, incomplete labels, and source quality, providing insights into their robustness and failure modes. Notably, models like CEM show greater resilience to concept biases, while ECBM struggles with partial interventions.

The significance of this work lies in its ability to systematically diagnose model weaknesses, guide targeted improvements, and accelerate development of interpretable AI systems. The synthetic data's flexibility makes it suitable for a broad range of applications, from medical diagnosis to industrial inspection. Despite its advantages, the synthetic approach cannot fully replace real-world validation, and future work aims to combine both for comprehensive evaluation. Overall, this framework marks a substantial step toward more reliable, transparent, and adaptable AI models.

Deep Dive

Abstract

Concept bottleneck models predict outcomes from high-level concepts detected in inputs. Although concepts provide a simple way to reap benefits from interpretability, very few datasets include concept labels. This limits researchers' ability to determine which problems are suitable for these models, isolate the factors that drive their performance or lead to failures, or uncover which algorithms perform well. In this paper, we develop synthetic benchmarks for concept-bottleneck models, focusing on their two main use cases: decision support, in which models assist humans in making better decisions, and automation, in which models handle routine tasks without supervision. Our benchmarks can generate labeled datasets while controlling for properties that affect performance, including data modality, concept choice, annotation quality, and completeness. We demonstrate how the benchmarks can be used to evaluate representative classes of concept bottleneck models. Our demonstrations show how the benchmarks can diagnose failure modes and guide follow-up testing.

cs.LG cs.AI