Take One Gram of Neural Features, Get Enhanced Group Robustness

Simon Roburin, Charles Corbière, Gilles Puy +4

Predictive performance of machine learning models trained with empirical risk minimization (ERM) can degrade considerably under distribution shifts. The presence of spurious correl…