Continual Active Learning Using Pseudo-Domains for Limited Labelling Resources and Changing Acquisition Characteristics

Matthias Perkonigg1, Johannes Hofmanninger1, Christian Herold1, Helmut Prosch1, Georg Langs1
1: Medical University of Vienna
IPMI 2021 special issue
Publication date: 2022/03/16
PDF · arXiv · Video · Code

Abstract

Machine learning in medical imaging during clinical routine is impaired by changes in scanner protocols, hardware, or policies resulting in a heterogeneous set of acquisition settings. When training a deep learning model on an initial static training set, model performance and reliability suffer from changes of acquisition characteristics as data and targets may become inconsistent. Continual learning can help to adapt models to the changing environment by training on a continuous data stream. However, continual manual expert labelling of medical imaging requires substantial effort. Thus, ways to use labelling resources efficiently on a well chosen sub-set of new examples is necessary to render this strategy feasible. Here, we propose a method for continual active learning operating on a stream of medical images in a multi-scanner setting. The approach automatically recognizes shifts in image acquisition characteristics – new domains –, selects optimal examples for labelling and adapts training accordingly. Labelling is subject to a limited budget, resembling typical real world scenarios. In order to avoid catastrophic forgetting while learning on new domains the proposed method utilizes a rehearsal memory. To demonstrate generalizability, we evaluate the effectiveness of our method on three tasks: cardiac segmentation, lung nodule detection and brain age estimation. Results show that the proposed approach outperforms other active learning methods on a continuous data stream with domain shifts.

Keywords

Continual learning · Active learning · Domain adaptation

Bibtex @article{melba:2022:007:perkonigg, title = "Continual Active Learning Using Pseudo-Domains for Limited Labelling Resources and Changing Acquisition Characteristics", author = "Perkonigg, Matthias and Hofmanninger, Johannes and Herold, Christian and Prosch, Helmut and Langs, Georg", journal = "Machine Learning for Biomedical Imaging", volume = "1", issue = "IPMI 2021 special issue", year = "2022", issn = "2766-905X", url = "https://melba-journal.org/papers/2022:007.html" }
RISTY - JOUR AU - Perkonigg, Matthias AU - Hofmanninger, Johannes AU - Herold, Christian AU - Prosch, Helmut AU - Langs, Georg PY - 2022 TI - Continual Active Learning Using Pseudo-Domains for Limited Labelling Resources and Changing Acquisition Characteristics T2 - Machine Learning for Biomedical Imaging VL - 1 IS - IPMI 2021 special issue SN - 2766-905X UR - https://www.melba-journal.org/papers/2022:007.html ER -

2022:007 cover