Multivariate Stochastic Dominance via Optimal Transport and Applications to Models Benchmarking
Gabriel Rioux, Apoorva Nitsure, Mattia Rigotti, Kristjan H. Greenewald, Youssef Mroueh
Abstract
Stochastic dominance is an important concept in probability theory, econometrics and social choice theory for robustly modeling agents' preferences between random outcomes. While many works have been dedicated to the univariate case, little has been done in the multivariate scenario, wherein an agent has to decide between different multivariate outcomes. By exploiting a characterization of multivariate first stochastic dominance in terms of couplings, we introduce a statistic that assesses multivariate almost stochastic dominance under the framework of Optimal Transport with a smooth cost. Further, we introduce an entropic regularization of this statistic, and establish a central limit theorem (CLT) and consistency of the bootstrap procedure for the empirical statistic. Armed with this CLT, we propose a hypothesis testing framework as well as an efficient implementation using the Sinkhorn algorithm. We showcase our method in comparing and benchmarking Large Language Models that are evaluated on multiple metrics. Our multivariate stochastic dominance test allows us to capture the dependencies between the metrics in order to make an informed and statistically significant decision on the relative performance of the models.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 2af72e4a-cf7e-47a9-a9eb-21757777c6d3Cited by top-tier papers2
- Imitation Beyond Expectation Using Pluralistic Stochastic DominanceAli Farajzadeh, Danyal Saeed, Syed M. Abbas, Rushit N. Shah et al.NeurIPS 2025 · 2 citations
- Center-Outward q-Dominance: A Sample-Computable Proxy for Strong Stochastic Dominance in Stochastic Multi-Objective OptimisationRobin van der Laag, Hao Wang, Thomas Bäck, Yingjie FanAAAI 2026
Builds on4
- PyTorch 2: Faster Machine Learning Through Dynamic Python Bytecode Transformation and Graph CompilationJason Ansel, Edward Z. Yang, Horace He, Natalia Gimelshein et al.ASPLOS 2024 · 693 citations
- LLM-Blender: Ensembling Large Language Models with Pairwise Ranking and Generative FusionDongfu Jiang, Xiang Ren, Bill Yuchen LinACL 2023 · 95 citations
- Smooth p-Wasserstein Distance: Structure, Empirical Approximation, and Statistical ApplicationsSloan Nietert, Ziv Goldfeld, Kengo KatoICML 2021 · 39 citations
- Inherent Trade-Offs between Diversity and Stability in Multi-Task BenchmarksGuanhua Zhang, Moritz HardtICML 2024 · 22 citations
Related papers
- Risk Aware Benchmarking of Large Language ModelsApoorva Nitsure, Youssef Mroueh, Mattia Rigotti, Kristjan H. Greenewald et al.ICML 2024 · 3 citations
- Sinkhorn Treatment Effects: A Causal Optimal Transport MeasureMedha Agarwal, Alex LuedtkeICML 2026
- Bisimulation Metrics are Optimal Transport Distances, and Can be Computed EfficientlySergio Calo, Anders Jonsson, Gergely Neu, Ludovic Schwartz et al.NeurIPS 2024 · 9 citations
- Online Sinkhorn: Optimal Transport distances from sample streamsArthur Mensch, Gabriel PeyréNeurIPS 2020 · 35 citations
- Learning with Stochastic OrdersCarles Domingo-Enrich, Yair Schiff, Youssef MrouehICLR 2023
