Abstract:Due to the non-convex nature of training Deep Neural Network (DNN) models, their effectiveness relies on the use of non-convex optimization heuristics. Traditional methods for training DNNs often require costly empirical methods to produce successful models and do not have a clear theoretical foundation. In this study, we examine the use of convex optimization theory and sparse recovery models to refine the training process of neural networks and provide a better interpretation of their optimal weights. We focus on training two-layer neural networks with piecewise linear activations and demonstrate that they can be formulated as a finite-dimensional convex program. These programs include a regularization term that promotes sparsity, which constitutes a variant of group Lasso. We first utilize semi-infinite programming theory to prove strong duality for finite width neural networks and then we express these architectures equivalently as high dimensional convex sparse recovery models. Remarkably, the worst-case complexity to solve the convex program is polynomial in the number of samples and number of neurons when the rank of the data matrix is bounded, which is the case in convolutional networks. To extend our method to training data of arbitrary rank, we develop a novel polynomial-time approximation scheme based on zonotope subsampling that comes with a guaranteed approximation ratio. We also show that all the stationary of the nonconvex training objective can be characterized as the global optimum of a subsampled convex program. Our convex models can be trained using standard convex solvers without resorting to heuristics or extensive hyper-parameter tuning unlike non-convex methods. Through extensive numerical experiments, we show that convex models can outperform traditional non-convex methods and are not sensitive to optimizer hyperparameters.

Adaptive group Lasso neural network models for functions of few variables and time-dependent data

Adaptive Lasso in high-dimensional settings

A sparse-group lasso

Sparse-Input Neural Network using Group Concave Regularization

Sparse-Input Neural Networks for High-dimensional Nonparametric Regression and Classification

Sparse neural network regression with variable selection

Adaptive deep learning for nonlinear time series models

Reconstruction of Complex Time-varying Weighted Networks Based on LASSO

A Non-linear Function-on-Function Model for Regression with Time Series Data

High dimensional data regression using Lasso model and neural networks with random weights.

The Linearized Alternating Direction Method of Multipliers for Sparse Group LAD Model.

Longitudinal LASSO: Jointly Learning Features and Temporal Contingency for Outcome Prediction

Non-asymptotic Oracle Inequalities for the Lasso and Group Lasso in high dimensional logistic model

Consistent group selection in high-dimensional linear regression

Generalized Newton–Raphson Algorithm for High Dimensional LASSO Regression

A least distance estimator for a multivariate regression model using deep neural networks

Bayesian Adaptive Lasso for the Partial Functional Linear Spatial Autoregressive Model

The Contextual Lasso: Sparse Linear Models via Deep Neural Networks

The Convex Landscape of Neural Networks: Characterizing Global Optima and Stationary Points via Lasso Models

Network-adaptive Robust Penalized Estimation of Time-Varying Coefficient Models with Longitudinal Data