cs.LGOct 6, 2026

Detecting a Shift Is Not Enough: Exact Minimax Limits of Linear Representation Repair

Authors: Anuar Aimoldin, Yankai Chen, Ayana Mussabayeva, Xue Liu

Organizations: Mohamed bin Zayed University of Artificial Intelligence (MBZUAI)

Abstract

A mean shift between two data sources can be easy to detect but hard to remove without substantially changing their representations. We cast its removal as a statistical decision problem: from noisy differences between paired calibration measurements in Rd\mathbb{R}^d, learn one linear map, applied to both sources under a hard distortion budget, that leaves as little of the shift as possible on fresh data. We derive the exact finite-sample minimax risk over all such maps, (d−k)E[1/(d+2J)](d-k) \mathbb{E}[1/(d+2J)] with J∼Pois(κ/2)J\sim\mathrm{Pois}(κ/2), where the budget allows deleting kk directions and κκ is the calibration signal-to-noise ratio. Projecting out the mean calibration difference attains it without knowing κκ or the noise scale. This exposes a detection-repair gap: detecting the shift needs only κ≫dκ\gg\sqrt d, whereas removing a fixed fraction of it at constant distortion needs κ≍dκ\asymp d, as for estimating its direction. Standard linear concept erasers (MP, SAL, LEACE) remove the same calibration difference, so the formula gives, before fitting, exactly how much shift they leave on fresh data and how much calibration a target requires. The limit is robust: pairing keeps it exact for non-Gaussian shared content, the projection keeps its guarantee under anisotropic noise, and selective abstention cannot close the gap. On paired clinical and wearable sleep EEG, where differences between participants act as calibration noise, the formula predicts the device shift left in new participants, and more recordings per person soon stop helping. Together, these results tell whether a correction that falls short needs a better method, more recordings, or more participants.

Figures & tables

Appendix figures & tables9 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Target-Aware Linear Regression Under Distribution Shift

    Jun 22, 2026Zhewen Hou, Tian ZhengCovariate ShiftDistribution Shifts

  2. Anytime-Valid Confirmation of Label-Shift Corrections

    Jun 12, 2026Seungjin ChoiCovariate ShiftConformal Test Martingales

  3. Sharp Integrality Gaps in Calibration Distance

    Oct 5, 2026Zinan Wang, Xinhao YangCalibrated UncertaintyRecalibration