ReLU Neural Networks of Polynomial Size for Exact Maximum Flow Computation
Not on the map. Why?
Abstract (arXiv)
This paper studies the expressive power of artificial neural networks with rectified linear units. In order to study them as a model of real-valued computation, we introduce the concept of Max-Affine Arithmetic Programs and show equivalence between them and neural networks concerning natural complexity measures. We then use this result to show that two fundamental combinatorial optimization problems can be solved with polynomial-size neural networks. First, we show that for any undirected graph with n nodes, there is a neural network (with fixed weights and biases) of size O(n3) that takes the edge weights as input and computes the value of a minimum spanning tree of the graph. Second, we show that for any directed graph with n nodes and m arcs, there is a neural network of size O(m2n2) that takes the arc capacities as input and computes a maximum flow. Our results imply that these two problems can be solved with strongly polynomial time algorithms that solely use affine transformations and maxima computations, but no comparison-based branchings.
Key facts. Cited by 12 papers in the map (data through 14 Oct 2026). Builds on 24 of its 83 references that are in the map. First posted on arXiv 12 Feb 2021.
Builds on
24 of 83 references.
References · 24
- 2015Pointer Networkscited by 895
- 2018Attention, Learn to Solve Routing Problems!cited by 656
- 2018Machine Learning for Combinatorial Optimization: a Methodological Tour d'Horizoncited by 623
- 2016Neural Combinatorial Optimization with Reinforcement Learningcited by 608
- 2016Error bounds for approximations with deep ReLU networkscited by 574
- 2017Learning Combinatorial Optimization Algorithms over Graphscited by 561
- 2014On the Number of Linear Regions of Deep Neural Networkscited by 381
- 2016On the Expressive Power of Deep Neural Networkscited by 306
- 2016Understanding Deep Neural Networks with Rectified Linear Unitscited by 257
- 2016Benefits of depth in neural networkscited by 250
- 2015The Power of Depth for Feedforward Neural Networkscited by 183
- 2019Complexity of Linear Regions in Deep Networkscited by 167
12 more
- 2017Bounding and Counting Linear Regions of Deep Neural Networkscited by 161
- 2017Approximating Continuous Functions by ReLU Nets of Minimal Widthcited by 122
- 2017Universal Function Approximation by Deep Neural Nets with Bounded Width and ReLU Activationscited by 104
- 2015Representation Benefits of Deep Feedforward Networkscited by 86
- 2016Why Deep Neural Networks for Function Approximation?cited by 82
- 2016Depth-Width Tradeoffs in Approximating Natural Functions with Neural Networkscited by 59
- 2020Tight Hardness Results for Training Depth-2 ReLU Networkscited by 35
- 2018Learning Permutations with Sinkhorn Policy Gradientcited by 26
- 2017Lower bounds over Boolean inputs for deep neural networks with ReLU gatescited by 21
- 2018Neural Networks Should Be Wide Enough to Learn Disconnected Decision Regionscited by 17
- 2020Provably Good Solutions to the Knapsack Problem via Neural Networks of Bounded Sizecited by 16
- 2019Smoothing the gap between NP and ERcited by 10
Built on by
Citing papers per year: 2022: 1; 2023: 2; 2024: 3; 2025: 4; 2026: 2.
12 papers.
- 2023Training Neural Networks is NP-Hard in Fixed Dimensioncited by 23
- 2024Looped ReLU MLPs May Be All You Need as Practical Programmable Computerscited by 19
- 2025On the Expressiveness of Rational ReLU Neural Networks With Bounded Depthcited by 18
- 2022Training Fully Connected Neural Networks is existsℝ-Completecited by 15
- 2025Depth-Bounds for Neural Networks via the Braid Arrangementcited by 11
- 2024Decomposition Polyhedra of Piecewise Linear Functionscited by 9
- 2024Neural Networks and (Virtual) Extended Formulationscited by 9
- 2023Optimizing Solution-Samplers for Combinatorial Problems: The Landscape of Policy-Gradient Methodscited by 5
- 2025Arithmetic Circuits and Neural Networks for Regular Matroidscited by 2
- 2025On the expressivity of sparse maxout networkscited by 0
- 2026Tropical Circuits with Scalar Multiplication Gatescited by 0
- 2026Two Linear Programs are Better than Onecited by 0
- 2026Two Linear Programs are Better than One11 Oct 2026
- 2026Tropical Circuits with Scalar Multiplication Gates13 Jul 2026
- 2025Arithmetic Circuits and Neural Networks for Regular Matroids4 Nov 2025
- 2025On the expressivity of sparse maxout networks15 Oct 2025
- 2025Depth-Bounds for Neural Networks via the Braid Arrangement13 Feb 2025
- 2025On the Expressiveness of Rational ReLU Neural Networks With Bounded Depth10 Feb 2025
- 2024Neural Networks and (Virtual) Extended Formulations5 Nov 2024
- 2024Looped ReLU MLPs May Be All You Need as Practical Programmable Computers12 Oct 2024
- 2024Decomposition Polyhedra of Piecewise Linear Functions7 Oct 2024
- 2023Optimizing Solution-Samplers for Combinatorial Problems: The Landscape of Policy-Gradient Methods8 Oct 2023
- 2023Training Neural Networks is NP-Hard in Fixed Dimension29 Mar 2023
- 2022Training Fully Connected Neural Networks is existsℝ-Complete4 Apr 2022