Lifelong nnUNet: a framework for standardized medical continual learning

Abstract

<title>Abstract</title> As the acceptability of Deep Learning solutions grows, both medical practitioners and regulatory bodies are exploring ways to safely introduce image segmentation solutions in clinical practice. One frontier to overcome when translating promising research into the clinical open-world is the shift from static to lifelong learning. Continual learning, the practice of training models sequentially with shifting population dynamics, is seeing growing interest but is still at its infancy in healthcare. We present Lifelong nnUNet, a standardized framework for medical image segmentation that places state-of-the art continual segmentation at the hands of researchers and clinicians. Build on top of the popular nnUNet and equipped with all necessary modules for training and testing continual models, we ensure broad applicability and lower the barrier for evaluating new solutions in a sequential fashion. We introduce benchmark results on three medical segmentation use cases with open-source datasets, and provide the tools to quickly test five continual learning methods on further applications.

Ähnliche Arbeiten