acceptodds
Under review as a conference paper at ICLR 2027

Robust -Aggregation under Covariate Shift with Pairwise Overlap

Abstract

We study squared-loss aggregation of a fixed dictionary of predictors under covariate shift: a labeled sample of size comes from a source population with input law , while the target risk is under another input law . With known density ratios, the difficulty of a comparison depends on the weights where its two predictors disagree. We capture this dependence through an original-pair overlap coefficient and construct a robust -aggregate from median-of-means (MOM) estimates of risk differences and squared distances. Coordinatewise robust estimates can produce comparison cycles and non-Euclidean distance matrices. Our method repairs the distances by a root-distance projection onto the Euclidean distance matrices, then combines the predictors by solving a monotone variational inequality on the simplex. With exact population quantities, this rule is the classical uniform-prior -aggregate with mixing parameter . With noisy pairwise quantities, the variational form also accepts nonzero comparison cycles. The inputs are the density-ratio weights, a failure probability , and a bound on predictions such that, for a source input-response pair , the source conditional second moment satisfies wherever candidate predictors disagree. Without a supplied overlap bound, the resulting convex mixture achieves a leading-one oracle inequality of order , including settings in which the global second moment of the density ratio is infinite. Matching lower bounds show the worst-case dependence on overlap, dictionary size, and confidence.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.