TY - GEN
T1 - Mitigating Domain Shift in Concept Bottleneck Models for Diabetic Retinopathy Grading via Learnable Concept Projection
AU - Lee, Yunyoung
AU - Lee, Eungjoo
N1 - Publisher Copyright:
© 2026 SPIE. All rights reserved.
PY - 2026/4/3
Y1 - 2026/4/3
N2 - 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.
AB - 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.
KW - Concept Bottleneck Model
KW - Diabetic Retinopathy
KW - Few-Shot Domain Incremental Learning
UR - https://www.scopus.com/pages/publications/105039259217
UR - https://www.scopus.com/pages/publications/105039259217#tab=citedBy
U2 - 10.1117/12.3088213
DO - 10.1117/12.3088213
M3 - Conference contribution
AN - SCOPUS:105039259217
T3 - Progress in Biomedical Optics and Imaging - Proceedings of SPIE
BT - Medical Imaging 2026
A2 - Anastasio, Mark A.
A2 - Brankov, Jovan G.
PB - SPIE
T2 - Medical Imaging 2026: Image Perception, Observer Performance, and Technology Assessment
Y2 - 17 February 2026 through 19 February 2026
ER -