Optimal Sets and Solution Paths of ReLU Networks

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Mishkin, Aaron, Pilanci, Mert
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911761020485632
author Mishkin, Aaron
Pilanci, Mert
author_facet Mishkin, Aaron
Pilanci, Mert
contents We develop an analytical framework to characterize the set of optimal ReLU neural networks by reformulating the non-convex training problem as a convex program. We show that the global optima of the convex parameterization are given by a polyhedral set and then extend this characterization to the optimal set of the non-convex training objective. Since all stationary points of the ReLU training problem can be represented as optima of sub-sampled convex programs, our work provides a general expression for all critical points of the non-convex objective. We then leverage our results to provide an optimal pruning algorithm for computing minimal networks, establish conditions for the regularization path of ReLU networks to be continuous, and develop sensitivity results for minimal ReLU networks.
format Preprint
id arxiv_https___arxiv_org_abs_2306_00119
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Optimal Sets and Solution Paths of ReLU Networks
Mishkin, Aaron
Pilanci, Mert
Machine Learning
We develop an analytical framework to characterize the set of optimal ReLU neural networks by reformulating the non-convex training problem as a convex program. We show that the global optima of the convex parameterization are given by a polyhedral set and then extend this characterization to the optimal set of the non-convex training objective. Since all stationary points of the ReLU training problem can be represented as optima of sub-sampled convex programs, our work provides a general expression for all critical points of the non-convex objective. We then leverage our results to provide an optimal pruning algorithm for computing minimal networks, establish conditions for the regularization path of ReLU networks to be continuous, and develop sensitivity results for minimal ReLU networks.
title Optimal Sets and Solution Paths of ReLU Networks
topic Machine Learning
url https://arxiv.org/abs/2306.00119