Subset Selection with Shrinkage: Sparse Linear Modeling When the SNR Is Low I: The Discrete First-Order Algorithm Is a Descent Method Whose Successive Steps Shrink at Rate O(1/M)Research Paper
Motivation
Best subset selection fits a linear model using at most of the available features: it minimizes the residual sum of squares subject to the sparsity constraint , where counts the nonzero coordinates of . Mazumder, Radchenko and Dedieu (arXiv:1708.03288v4; Operations Research, 2023) show that best subsets overfits when the signal-to-noise ratio is low, and propose adding a shrinkage penalty with to the same constrained problem. Their statistical results (oracle inequalities, a lower bound for best subsets, a BIC-type choice of ) are the subject of the companion missions II–VI of this series.
The estimator is a mixed-integer program. To compute it in practice, and to warm-start the mixed-integer solver that certifies optimality, the paper uses a discrete first-order (DFO) method: a proximal-gradient iteration in which the proximal step is a projection-like operation onto the set of -sparse vectors with a norm penalty. The method descends from iterative hard thresholding (Blumensath & Davies, 2009), from proximal gradient methods for composite problems (Nesterov, 2013), and from the discrete first-order algorithms of Bertsimas, King and Mazumder (Annals of Statistics, 2016). This mission formalizes the paper's convergence guarantee for it, Proposition 2.
Setting
Vectors live in with the Euclidean inner product and norm . Write for the number of nonzero coordinates and . For the penalty is the norm itself, not its -th power.
Let be a differentiable convex function whose gradient is -Lipschitz:
The motivating case is with . The composite problem (7) is
For the quadratic model of around is
and the thresholding operator is the set of minimizers
which can contain several points when the have ties. A DFO sequence with constant starts at a -sparse and, for , picks any
Formalization targets
Goal: Proposition 2 (p. 10)
Assume in addition and . For every DFO sequence:
(a) if , then is non-increasing in and converges to some ;
(b) if , then for every
The statement holds for any of the stated class and any choice in the set-valued operator; nothing about least squares is assumed.
Milestones (proof in Supplement A.1, p. 35)
- (9), p. 8: for all and every , the bound the paper takes from Nesterov (2004).
- (20)–(21), p. 35: completing the square, . Hence the minimizers of over are exactly .
- (22), p. 35: for and in that set, .
- (23) and the next display, p. 35: summed over ,
- Proposition 1, p. 9 (companion, not used by the goal): every keeps the largest entries of in magnitude, for some ordering of ties. For it soft-thresholds them coordinatewise, and for it shrinks them as a group by , with the norm of those entries.
Significance
Proposition 2 is what makes the DFO method a sound component of the paper's computational pipeline. The objective values never increase, so every iterate is a feasible point at least as good as the warm start. Part (b) says that the step length is small after iterations. A point whose step vanishes is a fixed point of the update, which the paper calls a first-order stationary point (Definition 1). The stopping rule of the algorithm () is therefore reached in at most steps.
The result is proved in the paper, in half a page that defers its central inequality (22) to Proposition 6 of Bertsimas, King & Mazumder (2016). It has not been formalized. A machine-checked version produces three reusable pieces: a descent lemma for functions with Lipschitz gradient on , the variational characterization of the sparse proximal step, and a sufficient-decrease-to-rate argument for any descent method. These are the same pieces that iterative hard thresholding, proximal gradient, and block-coordinate methods for problems need.
Difficulty
The difficulty is moderate, and it lies in the infrastructure rather than in any single step. The descent lemma (9) needs the fundamental theorem of calculus along a segment for a function on a Euclidean space, combined with the Lipschitz bound on the gradient. The feasible set is not convex, so no convex-analysis argument (subgradient optimality, firm nonexpansiveness of a proximal map) applies to the thresholding step. Inequality (22) must instead be obtained by comparing the minimizer with the feasible competitor itself, which is why the hypothesis appears. Part (a) also needs a lower bound on for its limit to exist and be nonnegative, which the general class of does not provide on its own.
Formalization scope
is EuclideanSpace ℝ (Fin p), the gradient is Mathlib's gradient f, and coordinates are -based. Membership in is a predicate: feasibility plus minimality against every -sparse competitor. No selection from the set is fixed, and existence of minimizers is not assumed. A DFO sequence is any β : ℕ → ℝᵖ satisfying the update for every , indexed from as printed. The stopping rule and the support-restricted polishing Step 2 are not part of it. The penalty is for and the Euclidean norm for , and every theorem assumes and . The minimum over is a finite infimum over a nonempty index set.
Three hypotheses are added to Proposition 2 as printed, each disclosed in the item. The proof's claim "" is false for a general convex with Lipschitz gradient ( sends to ), so and are assumed. The update divides by , and allows , so is assumed. "Decreasing" is read as non-increasing. Convexity of is kept as in the paper's setting (7)–(8), though only the gradient bound is used. Proposition 1 is stated with an existential ordering: for a fixed sorting permutation the printed claim fails under ties.
The statement cannot be trivialized by restricting or the run. The goal quantifies over every of the class and every sequence obeying the update, the limit in (b) is the actual limit of the objective values, and the rate holds for every . Not formalized here: the mixed-integer formulations, the continuation and local-search heuristics of §2.4, Definition 1, and the paper's statistical results (missions II–VI). Contributions of reusable infrastructure, such as a general descent lemma or a lemma turning summed decrease into a rate, are welcome as separate theorems.
Selected references
- R. Mazumder, P. Radchenko, A. Dedieu, Subset Selection with Shrinkage: Sparse Linear Modeling When the SNR Is Low, arXiv:1708.03288v4, 2022; Operations Research 71(1), 2023. https://arxiv.org/abs/1708.03288v4
- D. Bertsimas, A. King, R. Mazumder, Best Subset Selection via a Modern Optimization Lens, Annals of Statistics 44(2), 2016. https://doi.org/10.1214/15-AOS1388
- Y. Nesterov, Gradient Methods for Minimizing Composite Functions, Mathematical Programming 140, 2013. https://doi.org/10.1007/s10107-012-0629-5
- Y. Nesterov, Introductory Lectures on Convex Optimization: A Basic Course, Kluwer, 2004. https://doi.org/10.1007/978-1-4419-8853-9
- T. Blumensath, M. E. Davies, Iterative Hard Thresholding for Compressed Sensing, Applied and Computational Harmonic Analysis 27(3), 2009. https://doi.org/10.1016/j.acha.2009.04.002