Timezone: »
Poster
Integrated Latent Heterogeneity and Invariance Learning in Kernel Space
Jiashuo Liu · Zheyuan Hu · Peng Cui · Bo Li · Zheyan Shen
The ability to generalize under distributional shifts is essential to reliable machine learning, while models optimized with empirical risk minimization usually fail on non-$i.i.d$ testing data. Recently, invariant learning methods for out-of-distribution (OOD) generalization propose to find causally invariant relationships with multi-environments. However, modern datasets are frequently multi-sourced without explicit source labels, rendering many invariant learning methods inapplicable. In this paper, we propose Kernelized Heterogeneous Risk Minimization (KerHRM) algorithm, which achieves both the latent heterogeneity exploration and invariant learning in kernel space, and then gives feedback to the original neural network by appointing invariant gradient direction. We theoretically justify our algorithm and empirically validate the effectiveness of our algorithm with extensive experiments.
Author Information
Jiashuo Liu (Tsinghua University)
Zheyuan Hu (Tsinghua University, Tsinghua University)
Peng Cui (Tsinghua University)
Bo Li (Tsinghua University)
Zheyan Shen (Tsinghua University)
More from the Same Authors
-
2020 Poster: Counterfactual Prediction for Bundle Treatment »
Hao Zou · Peng Cui · Bo Li · Zheyan Shen · Jianxin Ma · Hongxia Yang · Yue He -
2019 Poster: Learning Disentangled Representations for Recommendation »
Jianxin Ma · Chang Zhou · Peng Cui · Hongxia Yang · Wenwu Zhu