arXiv cs.LG

CVaR anchor regression protects against rare shifts

· 2小时前
AI 摘要

arXiv:2609.27034v2 Announce Type: replace-cross Abstract: We study prediction in new environments when the training data contains rare, large distribution shifts. Anchor regression penalizes the average of the squared mean residual across environments. It protects against shifts in an ellipsoid determined by the second moment of the training shifts. Covering rare shifts may therefore require a large penalty $\gamma \gg 1$, expanding the ellipsoid in every direction and reducing accuracy on common environments. We propose CVaR anchor regression, which replaces the average of the squared mean residuals with a tail average. We prove an exact worst-case risk guarantee under a linear structural model that allows for heteroscedastic noise. CVaR anchor regression protects against shifts in a union of ellipsoids determined by second moments of the shifts in the worst $\alpha$-fraction of environments. For $\alpha$ small enough, this is the scaled convex hull of the training shifts and their negatives. It therefore covers all training shifts when $\gamma \geq 1$. Unlike CVaR or GroupDRO applied directly to prediction risks, CVaR anchor regression does not give environments more weight solely because their noise levels are high. Examples show how the method can improve protection against rare shifts while retaining accuracy on common environments. We illustrate the method on the New York City taxi dataset.

查看原文(arXiv cs.LG)