Hadamard-Based Weight Smoothing in Sparse Learning
- Hadamard-based weight smoothing is a framework that transforms non-smooth, sparsity-regularized problems into equivalent smooth ones using overparametrization with element-wise products.
- It introduces surrogate smooth penalties through auxiliary variables, allowing standard gradient-based optimizers to efficiently navigate sparse learning tasks.
- Empirical results demonstrate significant sparsity and model compression in high-dimensional regression and neural network training with minimal performance trade-offs.
Hadamard-based weight smoothing is a general framework for converting non-smooth, sparsity-regularized optimization problems into equivalent smooth problems via overparametrization with Hadamard (element-wise) products or powers. By introducing smooth surrogate variables and reparametrizations, this approach enables the use of standard gradient-based optimization algorithms for sparse learning tasks, while preserving the underlying minima structure of non-smooth objectives (Kolb et al., 2023).
1. Base Formulation and Motivation
Sparse regularization, as found in penalties such as the norm or group norms, is fundamental in high-dimensional regression and model compression. These penalties give rise to optimization problems of the form
where is a smooth loss (typically empirical risk), and is a non-smooth, sparsity-promoting regularizer (e.g., or for grouped weights). The non-differentiability and often non-convexity of (especially for with ) impede the application of standard SGD, leading to oscillatory dynamics near zero, poor sparsity, and slow convergence.
2. Hadamard Overparametrization: Construction and Variants
To address non-smooth regularization, auxiliary variables are introduced with a smooth surjection 0. The simplest case is the depth-2 Hadamard product parametrization (HPP):
1
More generally, the approach encompasses various structured parametrizations:
| Parametrization | Formula | Induced Regularization |
|---|---|---|
| HPP2 | 3 | 4 |
| GHPP (group) | 5 | group-6 |
| GHPP7 (mixed group) | 8 | 9 |
| HPowP0 | 1 | 2 |
Here, 3 is always smooth and surjective, implying every 4 has infinitely many preimages 5 such that 6.
3. Surrogate Smooth Penalties and Variational Equivalence
A key component is the smooth surrogate penalty 7 in the auxiliary space, usually a weighted 8 norm. For HPP, 9. By an elementary AM-GM argument,
0
meaning the surrogate penalty achieves the sparsity penalty exactly on the fiber 1. More generally,
2
with equality for optimal 3. This construction generates closed-form surrogates for:
- 4 via HPP5
- group 6 via GHPP
- mixed 7 via deeper group products
- general 8 for real 9 via HPowP
4. Equivalence of Minima and Theoretical Guarantees
The surrogate objective is defined as:
0
Under two mild conditions—local openness of 1 at optimal 2 and upper-hemicontinuity of the minimizer map 3—the set of local and global minima of 4 and 5 coincide:
- Any local minimum 6 of 7 corresponds to local minima 8 of 9 with 0.
- Conversely, local minima of 1 project to local minima of 2 via 3.
- Global minima of 4 and 5 match, since 6 majorizes 7 and achieves equality at constrained minima.
All parametrizations and surrogates are 8 except possibly at coordinate-wise zero; the regularizer handles these singularities. No spurious local minima are introduced.
5. Gradient-Based Optimization Algorithms
Hadamard-based weight smoothing enables the direct use of SGD or Adam for sparse regularized objectives. For 9 via HPP, gradient steps proceed as follows:
- Let 0.
- Compute 1 at 2.
- Update:
- 3
- 4
- 5
- 6
Deeper or group variants use appropriate Jacobian factors and distribute the 7 terms. Initialization can use small random values or AM-GM matched factors. This approach requires only substituting the 8 parameters with auxiliary variables and 9 regularization, without the need for proximal operators or custom solvers.
6. Empirical Evaluation and Use Cases
Several empirical studies demonstrate the practicality of Hadamard-based weight smoothing:
- High-Dimensional Regression: On synthetic problems 0, HPP-SGD closely matches Lasso regularization paths, produces exact sparsity, and converges reliably. Direct SGD on nonsmooth 1 is inferior, failing to produce zeros due to oscillation.
- Sparse Neural Network Training: Fully-connected LeNet-300-100 models trained with HPP2 on MNIST, using 3 regularization on 4, yield weights 5 with pronounced 6 sparsity. After one-shot pruning, up to 99% parameter reduction is achievable with minimal loss increase. Deeper factorizations (HPP7 for 8) further enhance sparsity.
- Filter-Sparse CNNs: Applying group Hadamard powers to convolution filter groups achieves up to 90% filter removal after training from scratch, with negligible accuracy degradation.
7. Comparison with Existing Methodologies
Hadamard-based weight smoothing unifies various sparsity-inducing formulations:
- Universality: Accommodates 9, 0 (1), 2, and more within a single SGD-compatible framework.
- Theoretical Guarantees: Ensures equivalence of all minima between original and surrogate problems, with no spurious solutions.
- Practical Performance: Matches specialized solvers (e.g., glmnet, SGL) in high-dimensional regression; delivers substantial sparsity in neural networks using standard SGD.
- Implementation Simplicity: Replaces 3 with surrogate variables; relies only on standard 4 regularization and smooth optimization.
- Computational Overhead: Increases parameter count moderately; shallow models incur minimal extra compute. For deep/large models, parameter sharing and twin initializations can mitigate costs.
Hadamard-based weight smoothing offers a plug-and-play technique for achieving exact sparse regularization via smooth objectives, applicable across a diverse range of models and compatible with modern deep learning toolkits and optimizers (Kolb et al., 2023).