LGAICVMay 24, 2024

Editable Concept Bottleneck Models

arXiv:2405.15476v322 citationsh-index: 14ICML
Originality Incremental advance
AI Analysis

This addresses the problem of data and concept editing in CBMs for applications with privacy or annotation errors, though it is incremental as it builds on existing CBM frameworks.

The paper tackles the challenge of efficiently editing Concept Bottleneck Models (CBMs) to remove or insert data or concepts without retraining, proposing Editable CBMs (ECBMs) that use closed-form approximations from influence functions, achieving practical efficiency and adaptability in experiments.

Concept Bottleneck Models (CBMs) have garnered much attention for their ability to elucidate the prediction process through a humanunderstandable concept layer. However, most previous studies focused on cases where the data, including concepts, are clean. In many scenarios, we often need to remove/insert some training data or new concepts from trained CBMs for reasons such as privacy concerns, data mislabelling, spurious concepts, and concept annotation errors. Thus, deriving efficient editable CBMs without retraining from scratch remains a challenge, particularly in large-scale applications. To address these challenges, we propose Editable Concept Bottleneck Models (ECBMs). Specifically, ECBMs support three different levels of data removal: concept-label-level, concept-level, and data-level. ECBMs enjoy mathematically rigorous closed-form approximations derived from influence functions that obviate the need for retraining. Experimental results demonstrate the efficiency and adaptability of our ECBMs, affirming their practical value in CBMs.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes