cs.LGJun 11, 2026

Is Spurious Correlation Removal Always Learnable?

Authors: Yibo ZhouBo LiHai-Miao HuHanzi WangXiaokang ZhangRuifan Zhang

Abstract

Invariant learning can fail even when the invariant structure is statistically identifiable. We show a conditional computational barrier: under a black-box samplable supervised sparse recovery primitive motivated by average-case sparse-recovery reductions, there exist \emph{samplable} multi-environment instances with a one-dimensional predictive invariant subspace (k=1k=1) that are learnable with polynomial samples by exhaustive search, while any polynomial-time constant-accuracy recovery algorithm would contradict the primitive. We further quantify environment diversity by a separation parameter γγ, which controls identifiability and the curvature of invariance objectives. Under sufficient diversity and local Gaussian regularity, the minimax risk is E[\dist(V^,Vinv)2]=Θ(k(dk)/(nE))\mathbb{E}[\dist(\hat{V},V_{\mathrm{inv}})^2]=Θ(k(d-k)/(n|\mathcal{E}|)), and under label-induced shifts a phase transition occurs at nk(dk)/(Eγ2)n^*\propto k(d-k)/(|\mathcal{E}|γ^2) with refined estimation error scaling proportional to 1/γ21/γ^2. Synthetic and real datasets illustrate the predicted gaps and transitions and motivate simple diversity diagnostics.

Explore similar work

Apr 28, 2026stat.ML

Robust Representation Learning through Explicit Environment Modeling

We consider learning from labeled data collected across multiple environments, where the data distribution may vary across these environments. This problem is commonly approached from a causal perspective, seeking invariant representations that retain causal factors while discarding spurious ones. However, this framework assumes that the environment has no direct effect on the target. In contrast, we consider settings in which this assumption fails, but still aim to learn representations that support robust prediction on average across previously unseen environments. To this end, we study representations learned by explicitly modeling variation across environments and then marginalizing that variation out. We analyze the resulting representations and characterize when they are preferable to those learned by causal invariant-representation methods. We propose a concrete method based on generalized random-intercept models, a class of predictors in which such marginalization is possible, and study their generalization properties. Empirically, we show that these models outperform invariant-learning methods across a range of challenging settings.
Yuli Slavutsky, David M. Blei
Feb 18, 2026math.ST

Separating Oblivious and Adaptive Models of Variable Selection

Sparse recovery is among the most well-studied problems in learning theory and high-dimensional statistics. In this work, we investigate the statistical and computational landscapes of sparse recovery with \ell_\infty error guarantees. This variant of the problem is motivated by \emph{variable selection} tasks, where the goal is to estimate the support of a kk-sparse signal in Rd\mathbb{R}^d. Our main contribution is a provable separation between the \emph{oblivious} (for each'') and \emph{adaptive} (for all'') models of \ell_\infty sparse recovery. We show that under an oblivious model, the optimal \ell_\infty error is attainable in near-linear time with klogd\approx k\log d samples, whereas in an adaptive model, k2\gtrsim k^2 samples are necessary for any algorithm to achieve this bound. This establishes a surprising contrast with the standard 2\ell_2 setting, where klogd\approx k \log d samples suffice even for adaptive sparse recovery. We conclude with a preliminary examination of a \emph{partially-adaptive} model, where we show nontrivial variable selection guarantees are possible with klogd\approx k\log d measurements.
Ziyun Chen, Jerry Li, Kevin Tian +1
Jul 8, 2026stat.ML

Statistical inverse learning and \ell^1-regularization

We study the recovery of sparse functions from finite, noisy, and indirect observations in the framework of statistical inverse learning. The unknown is modeled as an element of 1\ell^1, and observations are generated through a possibly nonlinear forward operator A:1HA:\ell^1\to H, where HH is a vector-valued reproducing kernel Hilbert space. We propose an 1\ell^1-regularized empirical risk minimizer and develop a theoretical analysis of its statistical properties. Under mild assumptions, we establish almost-sure consistency and derive non-asymptotic high-probability convergence rates in both the prediction and 1\ell^1 reconstruction norms. The rates depend on the source smoothness parameter rr, characterized by a variational source condition, and the effective dimension exponent bb, describing the polynomial spectral decay of the covariance operator. We further prove matching minimax lower bounds, showing that the obtained convergence rates are optimal. To relate the theory to practical sparsity models, we consider finitely smoothing operators of the form A=GSA=G\circ S, where SS is a synthesis operator, and show that approximation-space assumptions imply the required variational source conditions. In particular, we prove that membership in the approximation space ktk_t is equivalent to polynomial decay of the best nn-term approximation error. Finally, we verify the assumptions for two representative inverse problems: reaction coefficient identification in elliptic PDEs and sparse computed tomography. For filtered Radon transforms, we derive explicit effective-dimension asymptotics, yielding concrete convergence rates for standard image models and sparsifying systems.
Abhishake Rastogi, Tatiana A. Bubba, Tapio Helin +1