Multivariate Stochastic Dominance via Optimal Transport and Applications to Models Benchmarking

Gabriel Rioux, Apoorva Nitsure, Mattia Rigotti, Kristjan Greenewald, Youssef Mroueh
6/10/2024
stat.MLcs.LGmath.ST

No official code found. Best implementation: IBM/stochastic-order-eval (1 star, Apache-2.0, last commit 2025-09-18).

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.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

Repo for the code of the paper G. Rioux, A. Nitsure, M. Rigotti, K. Greenewald, Y. Mroueh. "Multivariate Stochastic Dominance via Optimal Transport and Applications to Models Benchmarking", NeurIPS 2024

10Oct 29, 20241 years agoApache-2.0

Cite this paper

@article{rioux2024multivariate,
  title  = {Multivariate Stochastic Dominance via Optimal Transport and Applications to Models Benchmarking},
  author = {Gabriel Rioux and Apoorva Nitsure and Mattia Rigotti and Kristjan Greenewald and Youssef Mroueh},
  year   = {2024},
  eprint = {2406.06425},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2406.06425},
  url    = {http://arxiv.org/abs/2406.06425v1},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.06425.svg)](https://codeofpaper.com/paper/2406.06425)

Discussion