Title

Continual domain incremental learning for chest X-ray classification in low-resource clinical settings

Document Type

Conference Proceeding

Publication Title

Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)

Abstract

Within clinical practise, a shift in the distribution of data collected over time is commonly observed. This occurs generally due to deliberate changes in acquisition hardware but also through natural, unforeseen shifts in the hardware’s physical properties like scanner SNR and gradient non-linearities. These domain shifts thus may not be known a-priori, but may cause significant degradation in the diagnostic performance of machine learning models. A deployed diagnostic system must therefore be robust to such unpredictable and continuous domain shifts. However, given the infrastructure and resource constraints pervasive in clinical settings in developing countries, such robustness must be achieved under finite memory and data privacy constraints. In this work, we propose a domain-incremental learning approach that leverages vector quantization to efficiently store and replay hidden representations under limited memory constraints. Our proposed approach validated on well known large-scale public Chest X-ray datasets achieves significant reduction in catastrophic forgetting over previous approaches in medical imaging, while requiring no prior knowledge of domain shift boundaries and a constrained memory. Finally, we also formulate a more natural continual learning setting for medical imaging using a tapered uniform distribution schedule with gradual interleaved domain shifts.

First Page

226

Last Page

238

DOI

10.1007/978-3-030-87722-4_21

Publication Date

9-21-2021

Keywords

Catastrophic forgetting, Chest X-ray classification, Continuous learning, Domain adaptation

Comments

IR Deposit conditions:

  • OA version (pathway a)
  • Accepted version 12 month embargo
  • Must link to published article
  • Set statement to accompany deposit

Share

COinS