General Quantification of Covariate and Concept Shifts
Hongbo Chen, Li Xia
Abstract
Generalization under distribution shift remains a core challenge in modern machine learning, yet existing learning bound theory is limited to narrow, idealized settings and is non-estimable from samples. In this paper, we bridge the gap between theory and practical applications. We first show that existing bounds become loose and nonestimable because their concept shift definition breaks when the source and target supports mismatch. Leveraging entropic optimal transport, we propose new support-agnostic definitions for covariate and concept shifts, and derive a novel unified error bound that applies to broad loss functions, label spaces, and stochastic labeling. We further develop estimators for these shifts with concentration guarantees, and the DataShifts algorithm, which can quantify distribution shifts and estimate the error bound in most applications -a rigorous and general tool for analyzing learning error under distribution shift.
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.
Builds on3
- Out-of-Distribution Generalization via Risk Extrapolation (REx)David Krueger, Ethan Caballero, Jörn-Henrik Jacobsen, Amy Zhang et al.ICML 2021 · 1,163 citations
- Towards a Theoretical Framework of Out-of-Distribution GeneralizationHaotian Ye, Chuanlong Xie, Tianle Cai, Ruichen Li et al.NeurIPS 2021 · 159 citations
- NICO++: Towards Better Benchmarking for Domain GeneralizationXingxuan Zhang, Yue He, Renzhe Xu, Han Yu et al.CVPR 2023
Related papers
- Non-exchangeable Conformal Prediction with Optimal Transport: Tackling Distribution Shift with Unlabeled DataAlvaro H. C. Correia, Christos LouizosNeurIPS 2025 · 5 citations
- Margin-aware Adversarial Domain Adaptation with Optimal TransportSofien Dhouib, Ievgen Redko, Carole LartizienICML 2020 · 17 citations
- Universal generalization guarantees for Wasserstein distributionally robust modelsTam Le, Jérôme MalickICLR 2025
- A new similarity measure for covariate shift with applications to nonparametric regressionReese Pathak, Cong Ma, Martin J. WainwrightICML 2022 · 40 citations
- Incorporating Importance Weighting in Optimal Transport Based Domain AlignmentOkan Koç, Alexander Soen, Shanglin Li, Masashi SugiyamaICML 2026
