Robust Learning via Nested Distributionally Robust Optimization
Jinyi Huang, Jinlong Lei, Guodong Shi
摘要
Distributionally Robust Optimization (DRO) is widely used to improve model robustness, with existing methods addressing either geometric perturbations (e.g., input shifts) or statistical contamination (e.g., heavy-tailed noise and outliers) effectively. However, these uncertainty sources often co-exist, and coupling them through a single divergence or optimal transport constraint conflates geometric displacement with loss-based outlierness, frequently discarding informative high-leverage samples. We introduce nested DRO, a bilevel formulation that combines an outer optimistic -divergence cleaning step with an inner pessimistic optimal-transport robustification step, thereby decoupling geometric smoothing from statistical cleaning. We prove that this structure naturally induces a geometry-invariant, loss-based reweighting mechanism that separates outlier suppression from transport-induced regularization. We derive a tractable strong dual for the resulting non-convex problem and show its equivalence to variance-regularized risk minimization, leading to a clear statistical interpretation of the induced weights. Empirical results on synthetic and real datasets demonstrate that nested DRO consistently outperforms geometry-coupled DRO baselines, particularly under heavy-tailed contamination where preserving high-leverage structure is crucial.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper5
- Outlier-Robust Wasserstein DROSloan Nietert, Ziv Goldfeld, Soroosh ShafieeNeurIPS 2023 · 被引用 26 次
- Outlier-Robust Distributionally Robust Optimization via Unbalanced Optimal TransportZifan Wang, Yi Shen, Michael M. Zavlanos, Karl Henrik JohanssonNeurIPS 2024 · 被引用 16 次
- Efficient Algorithms for Empirical Group Distributionally Robust Optimization and BeyondDingzhi Yu, Yunuo Cai, Wei Jiang, Lijun ZhangICML 2024 · 被引用 9 次
- Distributionally Robust Optimization with Bias and Variance ReductionRonak Mehta, Vincent Roulet, Krishna Pillutla, Zaïd HarchaouiICLR 2024 · 被引用 6 次
- Revisiting Large-Scale Non-convex Distributionally Robust OptimizationQi Zhang, Yi Zhou, Simon Khan, Ashley Prater-Bennette 等ICLR 2025
相关 Paper
- Loss-aware distributionally robust optimization via trainable optimal transport ambiguity setsJonas Ohnemus, Marta Fochesato, Riccardo Zuliani, John LygerosICML 2026
- Bulk-Calibrated Credal Ambiguity Sets: Fast, Tractable Decision Making under Out-of-Sample ContaminationMengqi Chen, Thomas Berrett, Theodoros Damoulas, Michele CaprioICML 2026 · 被引用 1 次
- Provable Robust Overfitting Mitigation in Wasserstein Distributionally Robust OptimizationShuang Liu, Yihan Wang, Yifan Zhu, Yibo Miao 等ICLR 2025
- Geometry-Calibrated DRO: Combating Over-Pessimism with Free Energy ImplicationsJiashuo Liu, Jiayun Wu, Tianyu Wang, Hao Zou 等ICML 2024 · 被引用 5 次
- DORO: Distributional and Outlier Robust OptimizationRuntian Zhai, Chen Dan, J. Zico Kolter, Pradeep RavikumarICML 2021 · 被引用 74 次
