Exact Recovery Thresholds for Weighted Data Selection in Vector-Valued Linear Regression
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 , where , 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 . We further determine two more values of the weighted selection profile : at the near-threshold budget, , and at the spanning budget, for every , while for . For the smallest open intermediate cell we prove and , reduce the conjectured exact values and 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 on which no weighted selection of points recovers the optimal loss. All results are new only for ; the scalar case is due to Hanneke et al.