42
1

DRO: A Python Library for Distributionally Robust Optimization in Machine Learning

Main:6 Pages
3 Figures
Bibliography:4 Pages
4 Tables
Appendix:6 Pages
Abstract

We introduce dro, an open-source Python library for distributionally robust optimization (DRO) for regression and classification problems. The library implements 14 DRO formulations and 9 backbone models, enabling 79 distinct DRO methods. Furthermore, dro is compatible with both scikit-learn and PyTorch. Through vectorization and optimization approximation techniques, dro reduces runtime by 10x to over 1000x compared to baseline implementations on large-scale datasets. Comprehensive documentation is available atthis https URL.

View on arXiv
@article{liu2025_2505.23565,
  title={ DRO: A Python Library for Distributionally Robust Optimization in Machine Learning },
  author={ Jiashuo Liu and Tianyu Wang and Henry Lam and Hongseok Namkoong and Jose Blanchet },
  journal={arXiv preprint arXiv:2505.23565},
  year={ 2025 }
}
Comments on this paper