Stability Evaluation via Distributional Perturbation Analysis

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Blanchet, Jose, Cui, Peng, Li, Jiajin, Liu, Jiashuo
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914785435582464
author Blanchet, Jose
Cui, Peng
Li, Jiajin
Liu, Jiashuo
author_facet Blanchet, Jose
Cui, Peng
Li, Jiajin
Liu, Jiashuo
contents The performance of learning models often deteriorates when deployed in out-of-sample environments. To ensure reliable deployment, we propose a stability evaluation criterion based on distributional perturbations. Conceptually, our stability evaluation criterion is defined as the minimal perturbation required on our observed dataset to induce a prescribed deterioration in risk evaluation. In this paper, we utilize the optimal transport (OT) discrepancy with moment constraints on the \textit{(sample, density)} space to quantify this perturbation. Therefore, our stability evaluation criterion can address both \emph{data corruptions} and \emph{sub-population shifts} -- the two most common types of distribution shifts in real-world scenarios. To further realize practical benefits, we present a series of tractable convex formulations and computational methods tailored to different classes of loss functions. The key technical tool to achieve this is the strong duality theorem provided in this paper. Empirically, we validate the practical utility of our stability evaluation criterion across a host of real-world applications. These empirical studies showcase the criterion's ability not only to compare the stability of different learning models and features but also to provide valuable guidelines and strategies to further improve models.
format Preprint
id arxiv_https___arxiv_org_abs_2405_03198
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Stability Evaluation via Distributional Perturbation Analysis
Blanchet, Jose
Cui, Peng
Li, Jiajin
Liu, Jiashuo
Machine Learning
Optimization and Control
The performance of learning models often deteriorates when deployed in out-of-sample environments. To ensure reliable deployment, we propose a stability evaluation criterion based on distributional perturbations. Conceptually, our stability evaluation criterion is defined as the minimal perturbation required on our observed dataset to induce a prescribed deterioration in risk evaluation. In this paper, we utilize the optimal transport (OT) discrepancy with moment constraints on the \textit{(sample, density)} space to quantify this perturbation. Therefore, our stability evaluation criterion can address both \emph{data corruptions} and \emph{sub-population shifts} -- the two most common types of distribution shifts in real-world scenarios. To further realize practical benefits, we present a series of tractable convex formulations and computational methods tailored to different classes of loss functions. The key technical tool to achieve this is the strong duality theorem provided in this paper. Empirically, we validate the practical utility of our stability evaluation criterion across a host of real-world applications. These empirical studies showcase the criterion's ability not only to compare the stability of different learning models and features but also to provide valuable guidelines and strategies to further improve models.
title Stability Evaluation via Distributional Perturbation Analysis
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2405.03198