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.
Sang ngôn ngữ khác
từ nội dung nguồn
arxiv.org
Thông tin chi tiết chính được chắt lọc từ
by Tim G. J. Ru... lúc arxiv.org 03-18-2024
https://arxiv.org/pdf/2403.09869.pdfYêu cầu sâu hơn