Skip to yearly menu bar Skip to main content


Poster
in
Workshop: Compositional Learning: Perspectives, Methods, and Paths Forward

Compositional Risk Minimization

Divyat Mahajan · Mohammad Pezeshki · Ioannis Mitliagkas · Kartik Ahuja · Pascal Vincent

Keywords: [ Compositional Generalization ] [ Out of Distribution Generalization ] [ Compositional Shifts ]


Abstract:

In this work, we tackle a challenging and extreme form of subpopulation shift, which is termed compositional shift. Under compositional shifts, some combinations of attributes are totally absent from the training distribution but present in the test distribution. We model the data with flexible additive energy distributions, where each energy term represents an attribute, and derive a simple alternative to empirical risk minimization termed compositional risk minimization (CRM). We provide an extensive theoretical analysis of CRM, where we show that our proposal extrapolates to special affine hulls of seen attribute combinations. Empirical evaluations on benchmark datasets confirms the improved robustness of CRM compared to other popular methods designed to tackle various forms of subpopulation shifts.

Chat is not available.