Work

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

Year: 2025
Type: preprint
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.
Language: English
Open Access status: green
OpenAlex ID: W4416611794
Cites:
Cited by:
Related to: