Tree of Concepts: Interpretable Continual Learners in Non-Stationary Clinical Domains

arXiv cs.LG / 4/21/2026

📰 NewsIdeas & Deep AnalysisModels & Research

Key Points

  • The paper addresses the difficulty of combining continual learning under distribution shift with interpretability, especially in high-stakes domains like healthcare.
  • It proposes “Tree of Concepts,” which uses a shallow, rule-based decision-tree concept interface and a concept bottleneck model to map raw features to stable concepts.
  • During continual updates, the method updates the concept extractor and label head while keeping concept semantics fixed, aiming to prevent explanation drift across time.
  • Across multiple tabular healthcare continual-learning benchmarks, the approach improves the stability–plasticity trade-off over existing baselines, including replay-based variants.
  • The authors conclude that structured concept interfaces can enable continual adaptation while maintaining a consistent, auditable explanation interface in non-stationary clinical settings.

Abstract

Continual learning aims to update models under distribution shift without forgetting, yet many high-stakes deployments, such as healthcare, also require interpretability. In practice, models that adapt well (e.g., deep networks) are often opaque, while models that are interpretable (e.g., decision trees) are brittle under shift, making it difficult to achieve both properties simultaneously. In response, we propose Tree of Concepts, an interpretable continual learning framework that uses a shallow decision tree to define a fixed, rule-based concept interface and trains a concept bottleneck model to predict these concepts from raw features. Continual updates act on the concept extractor and label head while keeping concept semantics stable over time, yielding explanations that do not drift across sequential updates. On multiple tabular healthcare benchmarks under continual learning protocols, our method achieves a stronger stability-plasticity trade-off than existing baselines, including replay-enhanced variants. Our results suggest that structured concept interfaces can support continual adaptation while preserving a consistent audit interface in non-stationary, high-stakes domains.