Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Estimating and Explaining Model Performance When Both Covariates and Labels Shift

About

Deployed machine learning (ML) models often encounter new user data that differs from their training data. Therefore, estimating how well a given model might perform on the new data is an important step toward reliable ML applications. This is very challenging, however, as the data distribution can change in flexible ways, and we may not have any labels on the new data, which is often the case in monitoring settings. In this paper, we propose a new distribution shift model, Sparse Joint Shift (SJS), which considers the joint shift of both labels and a few features. This unifies and generalizes several existing shift models including label shift and sparse covariate shift, where only marginal feature or label distribution shifts are considered. We describe mathematical conditions under which SJS is identifiable. We further propose SEES, an algorithmic framework to characterize the distribution shift under SJS and to estimate a model's performance on new data without any labels. We conduct extensive experiments on several real-world datasets with various ML models. Across different datasets and distribution shifts, SEES achieves significant (up to an order of magnitude) shift estimation error improvements over existing approaches.

Lingjiao Chen, Matei Zaharia, James Zou• 2022

Related benchmarks

TaskDatasetResultRank
Model Improvement EstimationSparse Covariate Shift (test)
Average Improvement56.679
28
Shift EstimationSparse Covariate Shifts Linear Models
Estimation Inaccuracy (Bin 1)1.4
10
Shift EstimationSparse Joint Shift
Estimation Inaccuracy (0.159, 0.167)5.7
9
ImprovementSparse Covariate Shift
Average Improvement (B1)14.05
7
ImprovementSparse Joint Shift
Avg Improvement (B1)16.59
7
Model Improvementsparse joint shift bin (0.159, 0.167) (test)
Average Improvement42.273
6
Model Improvementsparse joint shift bin (0.167, 0.174) (test)
Average Improvement39.636
6
Model Improvementsparse joint shift bin (0.174, 0.182) (test)
Average Improvement45.562
6
Model Improvementsparse joint shift bin (0.182, 0.189) (test)
Average Improvement39.977
6
Model ImprovementNatural shifts Bin (-0.082, -0.041)
Average Improvement25.821
6
Showing 10 of 25 rows

Other info

Follow for update