CI-CBM: Class-Incremental Concept Bottleneck Model for Interpretable Continual Learning

TL;DR

The paper introduces CI-CBM, a concept bottleneck model for class-incremental learning that maintains interpretability and achieves high accuracy, addressing catastrophic forgetting.

Contribution

It proposes a novel approach combining concept regularization and pseudo-concept generation to preserve interpretability during continual learning.

Findings

CI-CBM outperforms previous interpretable methods in accuracy by 36% on average.

It maintains interpretable decision processes throughout incremental learning phases.

The approach is effective with both pretrained and from-scratch models.

Abstract

Catastrophic forgetting remains a fundamental challenge in continual learning, in which models often forget previous knowledge when fine-tuned on a new task. This issue is especially pronounced in class incremental learning (CIL), which is the most challenging setting in continual learning. Existing methods to address catastrophic forgetting often sacrifice either model interpretability or accuracy. To address this challenge, we introduce ClassIncremental Concept Bottleneck Model (CI-CBM), which leverage effective techniques, including concept regularization and pseudo-concept generation to maintain interpretable decision processes throughout incremental learning phases. Through extensive evaluation on seven datasets, CI-CBM achieves comparable performance to black-box models and outperforms previous interpretable approaches in CIL, with an average 36% accuracy gain. CICBM provides interpretable decisions on individual inputs and understandable global decision rules, as shown in our experiments, thereby demonstrating that human understandable concepts can be maintained during incremental learning without compromising model performance. Our approach is effective in both pretrained and non-pretrained scenarios; in the latter, the backbone is trained from scratch during the first learning phase. Code is publicly available at github.com/importAmir/CI-CBM.