We study the problem of multiclass classification for settings where data features $\mathbf{x}$ and their labels $\mathbf{y}$ are uncertain. We identify that distributionally robust one-vs-all (OVA) classifiers often struggle in settings with imbalanced data. To address this issue, we use Wasserstein distributionally robust optimization to develop a robust version of the multiclass support vector machine (SVM) characterized by the Crammer-Singer (CS) loss. First, we prove that the CS loss is bounded from above by a Lipschitz continuous function for all $\mathbf{x} \in \mathcal{X}$ and $\mathbf{y} \in \mathcal{Y}$, then we exploit strong duality results to express the dual of the worst-case risk problem, and we show that the worst-case risk minimization problem admits a tractable convex reformulation due to the regularity of the CS loss. Moreover, we develop a kernel version of our proposed model to account for nonlinear class separation, and we show that it admits a tractable convex upper bound. We also propose a projected subgradient method algorithm for a special case of our proposed linear model to improve scalability. Our numerical experiments demonstrate that our model outperforms state-of-the art OVA models in settings where the training data is highly imbalanced. We also show through experiments on popular real-world datasets that our proposed model often outperforms its regularized counterpart as the first accounts for uncertain labels unlike the latter.
Improving Mixup Calibration with Wasserstein Distributionally Robust Optimization
Department of Mathematics & Statistics(数学与统计学系);Boston University(波士顿大学);Department of Electrical and Computer Engineering, Division of Systems Engineering(电气与计算机工程系,系统工程 division);Department of Biomedical Engineering(生物医学工程系);Faculty of Computing & Data Sciences(计算与数据科学学院)
FDR-SVM: A Federated Distributionally Robust Support Vector Machine via a Mixture of Wasserstein Balls Ambiguity Set
H. Milton Stewart School of Industrial and Systems Engineering(H. Milton Stewart工业与系统工程学院);Georgia Institute of Technology(佐治亚理工学院);Department of Electrical Engineering(电气工程系);University of Chile(智利大学)
Institute for Math and AI(数学与人工智能研究所);Wuhan University(武汉大学);School of Artificial Intelligence(人工智能学院);Department of Statistics and Data Science(统计与数据科学系);The Chinese University of Hong Kong(香港中文大学);The Hong Kong Polytechnic University(香港理工大学)
Columbia University(哥伦比亚大学);Northwestern University(西北大学);Columbia Mailman School of Public Health(哥伦比亚大学梅尔曼公共卫生学院);Hong Kong University of Science and Technology(香港科技大学)
Wasserstein Distributionally Robust Nash Equilibrium Seeking with Heterogeneous Data: A Lagrangian Approach
Division of Decision and Control Systems, School of Electrical Engineering and Computer Science, KTH Royal Institute of Technology(决策与控制系统系,电气工程与计算机科学学院,皇家理工学院);Delft Center for Systems and Control, Delft University of Technology(系统与控制中心,代尔夫特理工大学);Department of Mechanical Engineering and Materials Science, Duke University(机械工程与材料科学系,杜克大学)
Wasserstein Distributionally Robust Optimization Through the Lens of Structural Causal Models and Individual Fairness
Max Planck Institute for Intelligent Systems(马克斯·普朗克智能系统研究所);Tübingen AI Center(图宾根人工智能中心);Mila Québec AI Institute(魁北克人工智能研究所);McGill University(麦吉尔大学)
Wasserstein Distributionally Robust Regret Optimization for Reinforcement Learning from Human Feedback
Department of Statistics and Operations Research, University of North Carolina(统计与运筹学系,北卡罗来纳大学);Imperial Business School, Imperial College London(帝国理工学院伦敦商学院);Department of Management Science and Engineering, Stanford University(管理科学与工程系,斯坦福大学)
A stochastic smoothing framework for nonconvex-nonconcave minEmax problems with applications to Wasserstein distributionally robust optimization
Department of Applied Mathematics, Hong Kong Polytechnic University, Hong Kong, China(应用数学系,香港理工大学);Institute for Math & AI, Wuhan (IMAI), Wuhan university, China(数学与人工智能研究院,武汉大学);Department of Mathematical Sciences Rensselaer Polytechnic Institute, Troy, NY, USA(数学科学系,伦斯勒理工学院)
School of Mathematics, Sichuan University(四川大学数学学院);School of Statistics and Data Science, Southwestern University of Finance and Economics(西南财经大学统计与数据科学学院)
Certified Safety Radii in Forecast-Error Space for Wasserstein Distributionally Robust Small Signal Stability-Constrained AC Optimal Power Flow via Lifted Spectrahedral Containment
College of Automation Engineering, Nanjing University of Aeronautics and Astronautics(南京航空航天大学自动化学院)
Provable Robust Overfitting Mitigation in Wasserstein Distributionally Robust Optimization
Academy of Mathematics and Systems Science, Chinese Academy of Sciences(中国科学院数学与系统科学研究院);University of Chinese Academy of Sciences(中国科学院大学);Kaiyuan International Mathematical Sciences Institute(开元国际数学科学研究院)
Coresets for Wasserstein Distributionally Robust Optimization Problems
School of Data Science(数据科学学院);School of Computer Science and Technology(计算机科学与技术学院);City University of Hong Kong(香港城市大学);University of Science and Technology of China(中国科学技术大学)
Wasserstein Distributionally Robust Estimation in High Dimensions: Performance Analysis and Optimal Hyperparameter Tuning
Automatic Control Lab, ETH Zürich, Switzerland(苏黎世联邦理工学院自动控制实验室);Operations Research and Information Engineering, Cornell University, USA(康奈尔大学运筹学与信息工程)
School of Industrial and Systems Engineering, Georgia Institute of Technology(佐治亚理工学院工业与系统工程学院);Department of Information, Risk, and Operations Management, University of Texas at Austin(德克萨斯大学奥斯汀分校信息、风险与运营管理系)