Convex Relaxations of ReLU Neural Networks Approximate Global Optima in Polynomial Time
Sungyoon Kim, Mert Pilanci
Abstract
In this paper, we study the optimality gap between two-layer ReLU networks regularized with weight decay and their convex relaxations. We show that when the training data is random, the relative optimality gap between the original problem and its relaxation can be bounded by a factor of O(log n^0.5), where n is the number of training samples. A simple application leads to a tractable polynomial-time algorithm that is guaranteed to solve the original non-convex problem up to a logarithmic factor. Moreover, under mild assumptions, we show that local gradient methods converge to a point with low training loss with high probability. Our result is an exponential improvement compared to existing results and sheds new light on understanding why local gradient methods work well.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e769c0f8-b9e4-43fa-820d-89829cc32bfcCited by top-tier papers6
- CRONOS: Enhancing Deep Learning with Scalable GPU Accelerated Convex Neural NetworksMiria Feng, Zachary Frangella, Mert PilanciNeurIPS 2024 · 6 citations
- Convex Approximation of Two-Layer ReLU Networks for Hidden State Differential PrivacyRob Romijnders, Antti KoskelaNeurIPS 2025 · 2 citations
- Learning Provably Improves the Convergence of Gradient DescentQingyu Song, Wei Lin, Hong XuNeurIPS 2025 · 1 citation
- A Recovery Guarantee for Sparse Neural NetworksSara Fridovich-Keil, Mert PilanciICLR 2026 · 1 citation
- Convex Optimization for Alignment and Preference Learning on a Single GPUMiria Feng, Mert PilanciICML 2026
Builds on11
- Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networksZiwei Ji, Matus TelgarskyICLR 2020 · 193 citations
- Neural Networks are Convex Regularizers: Exact Polynomial-time Convex Optimization Formulations for Two-layer NetworksMert Pilanci, Tolga ErgenICML 2020 · 142 citations
- Vector-output ReLU Neural Network Problems are Copositive Programs: Convex Analysis of Two Layer Networks and Polynomial-time AlgorithmsArda Sahiner, Tolga Ergen, John M. Pauly, Mert PilanciICLR 2021 · 45 citations
- Unraveling Attention via Convex Duality: Analysis and Interpretations of Vision TransformersArda Sahiner, Tolga Ergen, Batu Ozturkler, John M. Pauly et al.ICML 2022 · 36 citations
- Global Optimality Beyond Two Layers: Training Deep ReLU Networks via Convex ProgramsTolga Ergen, Mert PilanciICML 2021 · 35 citations
Related papers
- Fast Convex Optimization for Two-Layer ReLU Networks: Equivalent Model Classes and Cone DecompositionsAaron Mishkin, Arda Sahiner, Mert PilanciICML 2022 · 35 citations
- Parallel Deep Neural Networks Have Zero Duality GapYifei Wang, Tolga Ergen, Mert PilanciICLR 2023 · 1 citation
- Demystifying Batch Normalization in ReLU Networks: Equivalent Convex Optimization Models and Implicit RegularizationTolga Ergen, Arda Sahiner, Batu Ozturkler, John M. Pauly et al.ICLR 2022 · 34 citations
- Early Neuron Alignment in Two-layer ReLU Networks with Small InitializationHancheng Min, Enrique Mallada, René VidalICLR 2024 · 31 citations
- Convergence of the Gradient Flow for Shallow ReLU Networks on Weakly Interacting DataLéo Dana, Loucas Pillaud-Vivien, Francis BachNeurIPS 2025 · 1 citation
