arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.08341stat.ME
收藏

Identifying Invariant Factors Across Multiple Environments with KL Regression

Jaime Roquero Gimenez, James Zou

英文摘要:

Many datasets are collected from multiple environments (e.g. different labs, perturbations, etc.), and it is often advantageous to learn models and relations that are invariant across environments. Invariance can improve robustness to unknown confounders and improve generalization to new domains. We develop a novel framework -- KL regression -- to reliably estimate regression coefficients in a challenging multi-environment setting, where latent confounders affect the data from each environment. KL regression is based on a new objective of simultaneously minimizing the KL- divergence between a parametric model and the observed data from each environment. We prove that KL regression recovers the true invariant factors under a flexible confounding setup. Moreover, it is computationally efficient as we derive an analytic solution for its global optimum. In systematic experiments, we validate the improved performance of KL regression compared to commonly used approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑