Skip to main navigation Skip to search Skip to main content

Mitigating Domain Shift in Concept Bottleneck Models for Diabetic Retinopathy Grading via Learnable Concept Projection

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution

Abstract

Concept bottleneck models (CBMs) offer interpretable diabetic retinopathy (DR) grading by routing predictions through clinically meaningful lesion concepts, yet they assume that training and deployment data share the same distribution. In practice, fundus images vary substantially across hospital sites due to differences in camera hardware, field of view, and patient demographics, causing concept predictors to misfire on out-of-distribution inputs. To address this limitation, we propose DICAN, a few-shot domain incremental learning framework that freezes all diagnostic components (backbone, prototype bank, and classification head) and trains only a low-rank residual projector (262K parameters). This projector realigns backbone features with frozen concept prototypes when a new hospital domain arrives with as few as 10 labeled images per grade. A multi-cluster prototype bank with best-match-then-aggregate scoring compresses sub-cluster similarities into three fixed statistics per concept, decoupling prototype granularity from classifier complexity and stabilizing few-shot adaptation. In this study, we evaluate DICAN across four sequential sessions using five multi-national DR datasets (DDR, FGADR, APTOS, Messidor-2, DRAC 2022). The proposed method achieves 54.37% average accuracy and 0.48 quadratic weighted kappa after all sessions, outperforming the strongest baseline by 10 percentage points in accuracy while halving forgetting. The concept bottleneck remains fully inspectable and correctable across all sessions, allowing clinicians to override lesion-concept scores and observe deterministic changes in the diagnostic output.

Original languageEnglish (US)
Title of host publicationMedical Imaging 2026
Subtitle of host publicationImage Perception, Observer Performance, and Technology Assessment
EditorsMark A. Anastasio, Jovan G. Brankov
PublisherSPIE
ISBN (Electronic)9781510697935
DOIs
StatePublished - Apr 3 2026
EventMedical Imaging 2026: Image Perception, Observer Performance, and Technology Assessment - Vancouver, Canada
Duration: Feb 17 2026 → Feb 19 2026

Publication series

NameProgress in Biomedical Optics and Imaging - Proceedings of SPIE
Volume13928
ISSN (Print)1605-7422
ISSN (Electronic)2410-9045

Conference

ConferenceMedical Imaging 2026: Image Perception, Observer Performance, and Technology Assessment
Country/TerritoryCanada
CityVancouver
Period2/17/26 → 2/19/26

Keywords

  • Concept Bottleneck Model
  • Diabetic Retinopathy
  • Few-Shot Domain Incremental Learning

ASJC Scopus subject areas

  • Electronic, Optical and Magnetic Materials
  • Atomic and Molecular Physics, and Optics
  • Biomaterials
  • Radiology Nuclear Medicine and imaging

Fingerprint

Dive into the research topics of 'Mitigating Domain Shift in Concept Bottleneck Models for Diabetic Retinopathy Grading via Learnable Concept Projection'. Together they form a unique fingerprint.

Cite this