Guider l'allocation des données pour une généralisation robuste aux sous-populations

Les modèles de machine learning affichent souvent une bonne précision moyenne à l'entraînement mais se dégradent sur les données de test en cas de subpopulation shift. Les auteurs étendent un cadre de modèle de features gaussiennes pour caractériser la performance selon la distribution des sous-groupes. Ils montrent que plusieurs configurations, y compris très déséquilibrées, peuvent égaler la configuration équilibrée. Ils dérivent aussi des directions fondées sur le gradient pour ajuster les proportions de sous-groupes et améliorer l'adjusted accuracy et la worst-group accuracy. Le code est disponible.