arXiv:2510.17200cs.CV2025-10被引 1

针对内镜图像分类的持续学习难题,提出新框架缓解遗忘与类别不平衡。

EndoCIL: A Class-Incremental Learning Framework for Endoscopic Image Classification

  • 用分布对齐策略精选代表性样本,提升回放效果
  • 结合先验分布设计损失函数,缓解跨阶段与组内类别不平衡
  • 调整全连接层梯度以平衡新旧类别学习偏差,适合临床持续学习

内镜图像分析中的类别增量学习对真实临床应用至关重要,要求诊断模型能持续适应新数据的同时保持对已有类别的性能。然而,现有基于回放的增量学习方法因内镜图像固有的域差异和类别不平衡问题,难以有效缓解灾难性遗忘。为此,本文提出专为内镜诊断设计的统一增量学习框架EndoCIL,包含三个核心组件:基于最大均值差异的回放(MDBR),采用分布对齐的贪心策略选取多样且有代表性的样本;先验正则化类别平衡损失(PRCBL),通过引入先验类别分布与平衡权重来缓解跨阶段及组内类别不平衡;全连接梯度校准(CFG),调节分类器梯度以减少对新类别的偏倚。在四个公开内镜数据集上的大量实验表明,EndoCIL在不同缓存大小和评估指标下普遍优于当前最优增量学习方法。该框架有效平衡了长期诊断中的稳定性与可塑性,展现出良好的临床可扩展性与部署潜力。

原文摘要 · Abstract (English)

Class-incremental learning (CIL) for endoscopic image analysis is crucial for real-world clinical applications, where diagnostic models should continuously adapt to evolving clinical data while retaining performance on previously learned ones. However, existing replay-based CIL methods fail to effectively mitigate catastrophic forgetting due to severe domain discrepancies and class imbalance inherent in endoscopic imaging. To tackle these challenges, we propose EndoCIL, a novel and unified CIL framework specifically tailored for endoscopic image diagnosis. EndoCIL incorporates three key components: Maximum Mean Discrepancy Based Replay (MDBR), employing a distribution-aligned greedy strategy to select diverse and representative exemplars, Prior Regularized Class Balanced Loss (PRCBL), designed to alleviate both inter-phase and intra-phase class imbalance by integrating prior class distributions and balance weights into the loss function, and Calibration of Fully-Connected Gradients (CFG), which adjusts the classifier gradients to mitigate bias toward new classes. Extensive experiments conducted on four public endoscopic datasets demonstrate that EndoCIL generally outperforms state-of-the-art CIL methods across varying buffer sizes and evaluation metrics. The proposed framework effectively balances stability and plasticity in lifelong endoscopic diagnosis, showing promising potential for clinical scalability and deployment.

增量学习内镜图像分类医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。