English

Robust Regression over Averaged Uncertainty

Machine Learning 2024-10-10 v2 Optimization and Control

Abstract

We propose a new formulation of robust regression by integrating all realizations of the uncertainty set and taking an averaged approach to obtain the optimal solution for the ordinary least squares regression problem. We show that this formulation recovers ridge regression exactly and establishes the missing link between robust optimization and the mean squared error approaches for existing regression problems. We further demonstrate that the condition of this equivalence relies on the geometric properties of the defined uncertainty set. We provide exact, closed-form, in some cases, analytical solutions to the equivalent regularization strength under uncertainty sets induced by p\ell_p norm, Schatten pp-norm, and general polytopes. We then show in synthetic datasets with different levels of uncertainties, a consistent improvement of the averaged formulation over the existing worst-case formulation in out-of-sample performance. In real-world regression problems obtained from UCI datasets, similar improvements are seen in the out-of-sample datasets.

Keywords

Cite

@article{arxiv.2311.06960,
  title  = {Robust Regression over Averaged Uncertainty},
  author = {Dimitris Bertsimas and Yu Ma},
  journal= {arXiv preprint arXiv:2311.06960},
  year   = {2024}
}
R2 v1 2026-06-28T13:18:43.329Z