Optimal Aggregation of Prediction Intervals under Unsupervised Domain Shift
Jiawei Ge, Debarghya Mukherjee, Jianqing Fan
Abstract
As machine learning models are increasingly deployed in dynamic environments, it becomes paramount to assess and quantify uncertainties associated with distribution shifts. A distribution shift occurs when the underlying data-generating process changes, leading to a deviation in the model's performance. The prediction interval, which captures the range of likely outcomes for a given prediction, serves as a crucial tool for characterizing uncertainties induced by their underlying distribution. In this paper, we propose methodologies for aggregating prediction intervals to obtain one with minimal width and adequate coverage on the target domain under unsupervised domain shift, under which we have labeled samples from a related source domain and unlabeled covariates from the target domain. Our analysis encompasses scenarios where the source and the target domain are related via i) a bounded density ratio, and ii) a measure-preserving transformation. Our proposed methodologies are computationally efficient and easy to implement. Beyond illustrating the performance of our method through real-world datasets, we also delve into the theoretical details. This includes establishing rigorous theoretical guarantees, coupled with finite sample bounds, regarding the coverage and width of our prediction intervals. Our approach excels in practical applications and is underpinned by a solid theoretical framework, ensuring its reliability and effectiveness across diverse contexts.
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 b86dfdbc-acfa-4574-9940-dca32ba3b62eCited by top-tier papers2
- Multi-Condition Conformal SelectionQingyang Hao, Wenbo Liao, Bingyi Jing, Hongxin WeiICLR 2026 · 5 citations
- Parametric Scaling Law of Tuning Bias in Conformal PredictionHao Zeng, Kangdao Liu, Bingyi Jing, Hongxin WeiICML 2025
Builds on5
- Adaptive Conformal Inference Under Distribution ShiftIsaac Gibbs, Emmanuel J. CandèsNeurIPS 2021 · 665 citations
- Optimal transport mapping via input convex neural networksAshok Vardhan Makkuva, Amirhossein Taghvaei, Sewoong Oh, Jason D. LeeICML 2020 · 254 citations
- Probable Domain Generalization via Quantile Risk MinimizationCian Eastwood, Alexander Robey, Shashank Singh, Julius von Kügelgen et al.NeurIPS 2022 · 99 citations
- Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric ProjectionsNabarun Deb, Promit Ghosal, Bodhisattva SenNeurIPS 2021 · 96 citations
- Reliable Weighted Optimal Transport for Unsupervised Domain AdaptationRenjun Xu, Pelen Liu, Liyan Wang, Chao Chen et al.CVPR 2020
Related papers
- Adaptive Conformal Prediction Intervals for Invariant LearningShuxin Liang, Yihan Xiao, Linglong Kong, Wenlu TangKDD 2025
- PAC Prediction Sets Under Covariate ShiftSangdon Park, Edgar Dobriban, Insup Lee, Osbert BastaniICLR 2022 · 54 citations
- Non-exchangeable Conformal Prediction with Optimal Transport: Tackling Distribution Shift with Unlabeled DataAlvaro H. C. Correia, Christos LouizosNeurIPS 2025 · 5 citations
- Distributionally Robust Classification for Multi-source Unsupervised Domain AdaptationSeonghwi Kim, Sungho Jo, Wooseok Ha, Minwoo ChaeICLR 2026 · 4 citations
- Learning Prediction Intervals for Model PerformanceBenjamin Elder, Matthew Arnold, Anupama Murthi, Jirí NavrátilAAAI 2021 · 14 citations
