On Distributionally Robust Optimization and Data Rebalancing

Abstract: Machine learning systems based on minimizing average error have been shown to perform inconsistently across notable subsets of the data, which is not exposed by a low average error for the entire dataset. Distributionally Robust Optimization (DRO) seemingly addresses this problem by minimizing the worst expected risk across subpopulations. We establish theoretical results that clarify the relation between DRO and the optimization of the same loss averaged on an adequately weighted training dataset. The results cover finite and infinite number of training distributions, as well as convex and non-convex loss functions. An implication of our results is that for each DRO problem there exists a data distribution such that learning this distribution is equivalent to solving the DRO problem. Yet, important problems that DRO seeks to address (for instance, adversarial robustness and fighting bias) cannot be reduced to finding the one ‘unbiased’ dataset. Our discussion section addresses this important discrepancy.

Agnieszka Słowik and Léon Bottou: On Distributionally Robust Optimization and Data Rebalancing, Proc. AIStats 2022, Feb 2022.

slowik-2022.djvu slowik-2022.pdf slowik-2022.ps.gz

@inproceedings{slowik-bottou-2022,
  author = {S\l{}owik, Agnieszka and Bottou, L\'{e}on},
  year = {2022},
  month = {Feb},
  booktitle = {Proc. AIStats 2022},
  title = {On Distributionally Robust Optimization and Data Rebalancing},
  note = {to appear},
  url = {http://leon.bottou.org/papers/slowik-bottou-2022},
}