cs.LGAug 31, 2026

Exact Recovery Thresholds for Weighted Data Selection in Vector-Valued Linear Regression

Authors: Guangjian Zhang

Abstract

We resolve the threshold part of Question 4 of the COLT 2025 open problem "Data Selection for Regression Tasks" of Hanneke, Moran, Shlimovich and Yehudayoff. In vector-valued linear regression with square loss (x,y)(W)=Wxy22\ell_{(x,y)}(W)=|Wx-y|_2^2, where xRdx\in\mathbb{R}^d, yRmy\in\mathbb{R}^m and the learner is the empirical risk minimizer of minimal Frobenius norm, we prove that the minimal budget of weighted examples that recovers the full-data loss on every finite dataset is exactly n(d,m)=(m+1)dn^*(d,m)=(m+1)d. We further determine two more values of the weighted selection profile Fw(d,m,n)F_w(d,m,n): at the near-threshold budget, Fw(d,m,(m+1)d1)=1+1dm2F_w(d,m,(m+1)d-1)=1+\frac{1}{dm^2}, and at the spanning budget, Fw(d,m,d)=d+1F_w(d,m,d)=d+1 for every mm, while Fw(d,m,n)=F_w(d,m,n)=\infty for n<dn<d. For the smallest open intermediate cell (d,m)=(2,2)(d,m)=(2,2) we prove Fw(2,2,3)[13/8,15/8]F_w(2,2,3)\in[13/8,15/8] and Fw(2,2,4)[5/4,3/2]F_w(2,2,4)\in[5/4,3/2], reduce the conjectured exact values 13/813/8 and 5/45/4 to a finite moment problem on the circle with at most seven atoms, and establish strong structural evidence for the conjecture. The upper-bound techniques (a fixed-basis conic compression lemma, a determinant-facet rigidity theorem for maximal certificates, and sharp sparsification lemmas for zero-mean weighted point systems) are of independent interest. As a byproduct we correct an erroneous claim circulating in a recent unrefereed preprint, exhibiting an explicit dataset with m=2m=2 on which no weighted selection of 2d2d points recovers the optimal loss. All results are new only for m2m\ge 2; the scalar case m=1m=1 is due to Hanneke et al.

Explore similar work

May 16, 2026stat.ML

Multi-task Linear Regression without Eigenvalue Lower Bounds: Adaptivity, Robustness, and Safety

We study the multi-task linear regression problem in the presence of contaminated tasks. We address the setting where the unknown parameters of a majority of tasks are close in the 2\ell_2-norm, while a fraction of tasks are arbitrary outliers. Existing theoretical frameworks for this problem rely heavily on the assumption that the empirical second moment of each task has a minimum eigenvalue bounded away from zero (order Ω(1)Ω(1)). Crucially, this assumption fails in many high-dimensional scenarios, rendering prior guarantees vacuous. To overcome this limitation, we propose an estimator based on matrix-weighted norm regularization. We also introduce a relative balancedness condition, quantified by a balancedness constant, that compares each task's second moment with the average inlier geometry and relaxes the need for taskwise second-moment lower bounds. In favorable regimes with moderate balancedness, our prediction MSE bounds match the rate of Duan and Wang (2023) under substantially weaker spectral assumptions; the resulting task-overall MSE is minimax optimal up to logarithmic factors. Furthermore, we demonstrate that our estimator enjoys a safety guarantee: when the relevant balancedness constant is large or infinite, or when tasks are unrelated, the method performs no worse than independent task learning.
Seok-Jin Kim
Sep 3, 2026cs.LG

Restricted Eigenvalues Beyond Gaussian Width: Threshold Occupancy under Heavy Tails

Restricted eigenvalue (RE) bounds govern stable recovery by norm-regularized estimators. For isotropic sub-Gaussian measurements, the benchmark sample size is 1+w(A)21+w(A)^2, where w(A)w(A) is the Gaussian width of the normalized descent cone. The COLT 2015 open-problem note (Banerjee et al., 2015) asked whether the same law follows for heavy-tailed designs from a uniform small-ball condition alone. We give an explicit and systematic negative answer to the general question as formulated there: the proposed law fails in its full dimension-free, arbitrary-set form, and the missing obstruction is simultaneous threshold occupancy. A constant-width polyhedral descent cone with fixed small-ball constants has zero empirical RE on every sample path up to half the ambient dimension. More generally, every finite range space admits exact threshold encoding in an arbitrarily narrow spherical cap and a lift to a full polyhedral descent-cone section. For every fixed threshold VC dimension dd, as β0β\downarrow0, the sharp worst-case sample complexity is Θ(β1[dlog(1/β)+log(1/δ)])Θ(β^{-1}[d\log(1/β)+\log(1/δ)]). The separation persists under exact isotropy and all finite moments: on the same constant-width cone, Gaussian measurements succeed with O(1+log(1/δ))O(1+\log(1/δ)) samples, whereas an isotropic heavy-tailed design fails pathwise for np/logpn\lesssim\sqrt{p/\log p}. Gaussian smoothing yields an everywhere-positive CC^\infty density while retaining arbitrarily poor RE. Under isotropy, a distribution-free fallback governed by affine dimension times squared enclosing radius is sharp on this family.
Shi Fu, Huibo Xu, Qixin Zhang +1
Aug 19, 2026math.ST

Algorithms for adaptive and heteroskedastic linear regression at the computational threshold

We study finite-sample linear regression in the presence of varied and unknown label noise, focusing on the heteroskedastic and adaptive linear regression models. Heteroskedastic linear regression models settings where the labels are of varying quality. We receive nn pairs (Xi,Yi)(X_i,Y_i) with labels Yi=Xiβ+εiY_i=X_i^\topβ+\varepsilon_i, where εiN(0,σi2)\varepsilon_i\sim N(0,σ_i^2) and the variances are unknown to the estimator. One natural measurement of the difficulty of this problem is the number of samples mm for which σi21σ_i^2\le1 (larger mm is easier). We obtain a polynomial-time estimator with rate O~((nd3/m4)1/6)\tilde{O}((nd^3/m^4)^{1/6}) when md3/4n1/4m\gg d^{3/4}n^{1/4}, as well as nearly-matching lower bounds. For d=O(1)d=O(1), our estimator achieves error o(1)o(1) when mn1/4m\gg n^{1/4}, whereas L1L_1 regression and other traditional approaches require mn1/2m\gg n^{1/2}. In adaptive linear regression, the errors are drawn i.i.d. from an unknown distribution pp, and our goal is to design a generic estimator that performs nearly as well as the best custom estimator that knows pp. We introduce a (computationally inefficient) adaptive estimator that, so long as pp is a mixture of kk symmetric log-concave densities, achieves error comparable with the optimal estimator that knows pp and has Θ~(n/k)\tildeΘ(n/k) samples. For k=1k=1, we show that LqL_q regression (with data-dependent qq) gives a polynomial-time estimator. Finally, to study the computational limits of both problems, we introduce the planted linear regression problem, where XiN(0,Id)X_i\sim N(0,I_d), mm unknown samples are noiseless, and the rest have error εiN(0,1)\varepsilon_i\sim N(0,1). We conjecture that recovering ββ up to error d/n\ll\sqrt{d/n} (or exactly) may have an information-computation gap between m=d+1m=d+1 and md3/4n1/4m\sim d^{3/4}n^{1/4}, as is suggested by our near-matching polynomial-time estimator and statistical query (SQ) lower bound.
Spencer Compton, Tselil Schramm