English

Predictive Inference in Multi-environment Scenarios

Machine Learning 2024-11-14 v2 Machine Learning Statistics Theory Methodology Statistics Theory

Abstract

We address the challenge of constructing valid confidence intervals and sets in problems of prediction across multiple environments. We investigate two types of coverage suitable for these problems, extending the jackknife and split-conformal methods to show how to obtain distribution-free coverage in such non-traditional, potentially hierarchical data-generating scenarios. We demonstrate a novel resizing method to adapt to problem difficulty, which applies both to existing approaches for predictive inference and the methods we develop; this reduces prediction set sizes using limited information from the test environment, a key to the methods' practical performance, which we evaluate through neurochemical sensing and species classification datasets. Our contributions also include extensions for settings with non-real-valued responses, a theory of consistency for predictive inference in these general problems, and insights on the limits of conditional coverage.

Keywords

Cite

@article{arxiv.2403.16336,
  title  = {Predictive Inference in Multi-environment Scenarios},
  author = {John C. Duchi and Suyash Gupta and Kuanhao Jiang and Pragya Sur},
  journal= {arXiv preprint arXiv:2403.16336},
  year   = {2024}
}
R2 v1 2026-06-28T15:32:00.991Z