Boosting Knowledge Diversity, Accuracy, and Stability Via Tri-Enhanced Distillation for Domain Continual Medical Image Segmentation.

Zhanshi Zhu,Xinghua Ma,Wei Wang,Suyu Dong,Kuanquan Wang,Lianming Wu,Gongning Luo,Guohua Wang,Shuo Li
DOI: https://doi.org/10.1016/j.media.2024.103112
IF: 10.9
2024-01-01
Medical Image Analysis
Abstract:Domain continual medical image segmentation plays a crucial role in clinical settings. This approach enables segmentation models to continually learn from a sequential data stream across multiple domains. However, it faces the challenge of catastrophic forgetting. Existing methods based on knowledge distillation show potential to address this challenge via a three-stage process: distillation, transfer, and fusion. Yet, each stage presents its unique issues that, collectively, amplify the problem of catastrophic forgetting. To address these issues at each stage, we propose a tri-enhanced distillation framework. (1) Stochastic Knowledge Augmentation reduces redundancy in knowledge, thereby increasing both the diversity and volume of knowledge derived from the old network. (2) Adaptive Knowledge Transfer selectively captures critical information from the old knowledge, facilitating a more accurate knowledge transfer. (3) Global Uncertainty-Guided Fusion introduces a global uncertainty view of the dataset to fuse the old and new knowledge with reduced bias, promoting a more stable knowledge fusion. Our experimental results not only validate the feasibility of our approach, but also demonstrate its superior performance compared to state-of-the-art methods. We suggest that our innovative tri-enhanced distillation framework may establish a robust benchmark for domain continual medical image segmentation.
What problem does this paper attempt to address?