What is Wrong with Continual Learning in Medical Image Segmentation?

TL;DR

This paper introduces UNEG, a multi-model benchmark for continual medical image segmentation that maintains separate models for each training stage and uses reconstruction error for inference, challenging existing methods focused on catastrophic forgetting.

Contribution

The paper proposes UNEG, a simple, effective multi-model benchmark for continual learning in medical segmentation, emphasizing the importance of strong baselines over complex forgetting prevention techniques.

Findings

UNEG outperforms several continual learning methods across multiple regions.

Reconstruction error effectively selects appropriate models during inference.

The study highlights the limitations of current continual learning approaches in medical imaging.

Abstract

Continual learning protocols are attracting increasing attention from the medical imaging community. In continual environments, datasets acquired under different conditions arrive sequentially; and each is only available for a limited period of time. Given the inherent privacy risks associated with medical data, this setup reflects the reality of deployment for deep learning diagnostic radiology systems. Many techniques exist to learn continuously for image classification, and several have been adapted to semantic segmentation. Yet most struggle to accumulate knowledge in a meaningful manner. Instead, they focus on preventing the problem of catastrophic forgetting, even when this reduces model plasticity and thereon burdens the training process. This puts into question whether the additional overhead of knowledge preservation is worth it - particularly for medical image segmentation, where computation requirements are already high - or if maintaining separate models would be a better solution. We propose UNEG, a simple and widely applicable multi-model benchmark that maintains separate segmentation and autoencoder networks for each training stage. The autoencoder is built from the same architecture as the segmentation network, which in our case is a full-resolution nnU-Net, to bypass any additional design decisions. During inference, the reconstruction error is used to select the most appropriate segmenter for each test image. Open this concept, we develop a fair evaluation scheme for different continual learning settings that moves beyond the prevention of catastrophic forgetting. Our results across three regions of interest (prostate, hippocampus, and right ventricle) show that UNEG outperforms several continual learning methods, reinforcing the need for strong baselines in continual learning research.