Continual Learning for Domain Adaptation in Chest X-ray Classification

  • 2020-01-16 16:20:43
  • Matthias Lenga, Heinrich Schulz, Axel Saalbach
  • 11

Abstract

Over the last years, Deep Learning has been successfully applied to a broadrange of medical applications. Especially in the context of chest X-rayclassification, results have been reported which are on par, or even superiorto experienced radiologists. Despite this success in controlled experimentalenvironments, it has been noted that the ability of Deep Learning models togeneralize to data from a new domain (with potentially different tasks) isoften limited. In order to address this challenge, we investigate techniquesfrom the field of Continual Learning (CL) including Joint Training (JT),Elastic Weight Consolidation (EWC) and Learning Without Forgetting (LWF). Usingthe ChestX-ray14 and the MIMIC-CXR datasets, we demonstrate empirically thatthese methods provide promising options to improve the performance of DeepLearning models on a target domain and to mitigate effectively catastrophicforgetting for the source domain. To this end, the best overall performance wasobtained using JT, while for LWF competitive results could be achieved - evenwithout accessing data from the source domain.

 

Quick Read (beta)

loading the full paper ...