用生成回放防遗忘,让医疗图像模型持续学习新数据不丢旧知识
Domain-incremental White Blood Cell Classification with Privacy-aware Continual Learning
- 用轻量生成器合成历史数据,隐私安全地回放旧任务
- 在4个数据集上验证,相比微调保留95%以上旧任务准确率
- 适合需要长期更新的临床影像诊断系统
白细胞分类在血液病诊断中至关重要,但因样本来源(如外周血或骨髓)和医院成像条件差异导致领域分布变化,传统深度学习模型易出现灾难性遗忘。尽管基础模型泛化性强,当推理数据分布与训练数据不同时性能仍会下降。为此,我们提出一种基于生成回放的持续学习策略,用于保护基础模型在白细胞分类中的稳定性。该方法采用轻量级生成器,通过合成潜在表示来模拟历史数据,实现隐私友好的数据回放。我们在包含四个数据集、四种任务顺序及四种主干网络(ResNet50、RetCCL、CTransPath、UNI)的实验中验证了其有效性。结果表明,传统微调方法在新任务上性能下降且无法应对领域偏移;而本方法有效缓解了灾难性遗忘,在不同领域间保持稳定性能。本工作为临床环境中数据分布动态变化下的可靠白细胞分类提供了实用解决方案。
原文摘要 · Abstract (English)
White blood cell (WBC) classification plays a vital role in hematology for diagnosing various medical conditions. However, it faces significant challenges due to domain shifts caused by variations in sample sources (e.g., blood or bone marrow) and differing imaging conditions across hospitals. Traditional deep learning models often suffer from catastrophic forgetting in such dynamic environments, while foundation models, though generally robust, experience performance degradation when the distribution of inference data differs from that of the training data. To address these challenges, we propose a generative replay-based Continual Learning (CL) strategy designed to prevent forgetting in foundation models for WBC classification. Our method employs lightweight generators to mimic past data with a synthetic latent representation to enable privacy-preserving replay. To showcase the effectiveness, we carry out extensive experiments with a total of four datasets with different task ordering and four backbone models including ResNet50, RetCCL, CTransPath, and UNI. Experimental results demonstrate that conventional fine-tuning methods degrade performance on previously learned tasks and struggle with domain shifts. In contrast, our continual learning strategy effectively mitigates catastrophic forgetting, preserving model performance across varying domains. This work presents a practical solution for maintaining reliable WBC classification in real-world clinical settings, where data distributions frequently evolve.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。