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.

Landscape Learning for Neural Network Inversion

Neural Network Inversion in Adversarial Setting Via Background Knowledge Alignment

Adversarial Neural Network Inversion via Auxiliary Knowledge Alignment

Mutual Information Preserving Back-propagation: Learn to Invert for Faithful Attribution

Efficient Loss Landscape Reshaping for Convolutional Neural Networks

Invertible generative models for inverse problems: mitigating representation error and dataset bias

Novel GAN Inversion Model with Latent Space Constraints for Face Reconstruction

Meta-Auxiliary Network for 3D GAN Inversion

Learning Gradient Descent: Better Generalization and Longer Horizons

Visualizing the Loss Landscape of Neural Nets

Network Inversion and Its Applications

In-Domain GAN Inversion for Faithful Reconstruction and Editability

GI-NAS: Boosting Gradient Inversion Attacks through Adaptive Neural Architecture Search

Learning Geometry-Dependent and Physics-Based Inverse Image Reconstruction

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

Deep Learning and Inverse Problems

On Architecture Selection for Linear Inverse Problems with Untrained Neural Networks

Better Loss Landscape Visualization for Deep Neural Networks with Trajectory Information.

A deep learning perspective of the forward and inverse problems in exploration geophysics

LossLens: Diagnostics for Machine Learning through Loss Landscape Visual Analytics