Learning Optimal Features via Partial Invariance

Moulik Choraria, Ibtihal Ferwana, Ankur Mani +1

Learning models that are robust to distribution shifts is a key concern in the context of their real-life applicability. Invariant Risk Minimization (IRM) is a popular framework th…