In this paper, the authors introduce a family of group-aware prior distributions over neural network parameters to enhance generalization under subpopulation shifts. The research focuses on achieving group robustness by designing data-driven priors that favor models with high group robustness. Unlike previous approaches, the authors tackle group robustness from a Bayesian perspective, enabling models to fit training data while respecting soft constraints imposed by the prior distribution. By constructing an example of a simple data-driven group-aware prior distribution, they demonstrate improved performance on benchmarking tasks related to subpopulation shifts. The study highlights the importance of probabilistic formulations in enhancing group robustness and leveraging Bayesian inference methods for higher levels of robustness.
Egy másik nyelvre
a forrásanyagból
arxiv.org
Mélyebb kérdések