Domain generalization on medical imaging classification using episodic training with task augmentation. (February 2022)

• Record Type: Journal Article
• Authors: Li, Chenxin; Lin, Xin; Mao, Yijin; Lin, Wei; Qi, Qi; Ding, Xinghao; Huang, Yue; Liang, Dong; Yu, Yizhou
• Is Part Of: Computers in biology and medicine. Volume 141(2022)
• Publication Date: 2022-02
• Subjects: Medical imaging classification -- Domain generalization -- Model-agnostic meta-learning -- Episodic training -- Task-level overfitting; Medicine -- Data processing -- Periodicals; Biology -- Data processing -- Periodicals; 610.285
• DOI: 10.1016/j.compbiomed.2021.105144
• Languages: English
• ISSNs: 0010-4825

Abstract

Abstract: Medical imaging datasets usually exhibit domain shift due to the variations of scanner vendors, imaging protocols, etc. This raises the concern about the generalization capacity of machine learning models. Domain generalization (DG), which aims to learn a model from multiple source domains such that it can be directly generalized to unseen test domains, seems particularly promising to medical imaging community. To address DG, recent model-agnostic meta-learning (MAML) has been introduced, which transfers the knowledge from previous training tasks to facilitate the learning of novel testing tasks. However, in clinical practice, there are usually only a few annotated source domains available, which decreases the capacity of training task generation and thus increases the risk of overfitting to training tasks in the paradigm. In this paper, we propose a novel DG scheme of episodic training with task augmentation on medical imaging classification. Based on meta-learning, we develop the paradigm of episodic training to construct the knowledge transfer from episodic training-task simulation to the real testing task of DG. Motivated by the limited number of source domains in real-world medical deployment, we consider the unique task-level overfitting and we propose task augmentation to enhance the variety during training task generation to alleviate it. With the established learning framework, we further exploit a novel meta-objective to regularize the deep embedding of training domains. To validate the effectiveness of the proposed method, we perform experiments on histopathological images and abdominal CT images. Highlights: We propose a novel DG scheme of episodic training with task augmentation on medical imaging classification. Based on meta-learning, we develop the paradigm of episodic training to construct the knowledge transfer from episodic training-task simulation to the real testing task of DG. We consider the unique task-level overfitting and then propose task augmentation to enhance the variety during training. We further exploit a novel meta-objective to regularize the deep embedding of training domains. The method achieved the state-of-the-art performance on histopathological images and CT images.