Unveiling Hidden Convexity in Deep Learning: a Sparse Signal Processing Perspective
Organizations: Department of Electrical Engineering, Stanford University
Abstract
Deep neural networks (DNNs), particularly those using Rectified Linear Unit (ReLU) activation functions, have achieved remarkable success across diverse machine learning tasks, including image recognition, audio processing, and language modeling. Despite this success, the non-convex nature of DNN loss functions complicates optimization and limits theoretical understanding. In this paper, we highlight how recently developed convex equivalences of ReLU NNs and their connections to sparse signal processing models can address the challenges of training and understanding NNs. Recent research has uncovered several hidden convexities in the loss landscapes of certain NN architectures, notably two-layer ReLU networks and other deeper or varied architectures. This paper seeks to provide an accessible and educational overview that bridges recent advances in the mathematics of deep learning with traditional signal processing, encouraging broader signal processing applications.
Figures & tables
| Convexity w.r.t: | Approach | Motivation/Technique | Relation to SP |
|---|---|---|---|
| outer-layer weights [ 15 ] | column generation | boosting | matching pursuit |
| parameters [ 7 ] | convex duality, Lasso | hyperplane arrangements zonotopes | sparse recovery, 1-bit CS [ 9 ] |
| probability measures [ 23 , 24 ] | mean field integration | neural tangent kernel (NTK) [ 25 ] | kernel methods [ 26 ] |
| network output [ 4 ] | shift complexity to regularization | matrix factorization, nuclear norm [ 18 ] | CS: robust PCA [ 20 ] matrix completion [ 19 ] |
| quantized weights [ 27 ] | linear program | intersection graph analysis | graphical models [ 28 ] |
ReLU Neural Networks
Rectified linear unit (ReLU) neural networks use piecewise-linear activations to study approximation, learning dynamics, expressivity, and verification.
See all 111 papers