DRO: A Python Library for Distributionally Robust Optimization in Machine Learning
2025-05-29Code Available2· sign in to hype
Jiashuo Liu, Tianyu Wang, Henry Lam, Hongseok Namkoong, Jose Blanchet
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/namkoong-lab/droOfficialIn paperpytorch★ 156
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 at https://python-dro.org.