DRO：面向机器学习的分布式鲁棒优化 Python 库
机器学习
2025-05-30 v1 数学软件
数值分析
数值分析
摘要
我们介绍了 dro，一个用于回归和分类问题的分布式鲁棒优化 (DRO) 的开源 Python 库。该库实现了 14 种 DRO 公式和 9 种骨干模型，支持 79 种不同的 DRO 方法。此外，dro 兼容 scikit-learn 和 PyTorch。通过向量化和优化近似技术，与大规模数据集上的基线实现相比，dro 将运行时间减少了 10 倍至 1000 倍以上。完整的文档可在 https://python-dro.org 获取。
引用
@article{arxiv.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}
}