Domain Disentangled Meta-Learning

Xin Zhang,Yanhua Li,Ziming Zhang,Zhi-Li Zhang
DOI: https://doi.org/10.1137/1.9781611977653.ch61
2023-01-01
Abstract:A key challenge with supervised learning ( e.g ., image clas-sification) is the shift of data distribution and domain from training to testing datasets, so-called “domain shift” (or “dis-tribution shift”), which usually leads to a reduction of model accuracy. Various meta-learning approaches have been proposed to prevent the accuracy loss by learning an adaptable model with training data, and adapting it to test time data from a new data domain. However, when the domain shift occurs in multiple domain dimensions ( e.g ., images may be transformed by rotations, transitions, and expansions), the average predictive power of the adapted model will deteriorate. To tackle this problem, we propose a domain disentangled meta-learning (DDML) framework. DDML disentangles the data domain by dimensions, learns the representations of domain dimensions independently, and adapts to the domain of test time data. We evaluate our DDML on image classifi-cation problems using three datasets with distribution shifts over multiple domain dimensions. Comparing to various base-lines in meta-learning and empirical risk minimization, our DDML approach achieves consistently higher classification accuracy with the test time data. These results demonstrate that domain disentanglement reduces the complexity of the model adaptation, thus increases the model generalizability, and prevents it from overfitting.
Computer Science
What problem does this paper attempt to address?