Environment-Conditioned Tail Reweighting for Total Variation Invariant Risk Minimization

Yuanchao Wang, Zhao-Rong Lai, Tianqi Zhong +1

Out-of-distribution (OOD) generalization remains challenging when models simultaneously encounter correlation shifts across environments and diversity shifts driven by rare or hard…