Decision-Diagram Convex Hull for Neural Gates
Implementation & benchmark of arXiv:2607.28511 — Beyond Hand-Derived Inequalities: Decision Diagrams for Cut Generation in Binary Polynomial Optimization
Source paper: Beyond Hand-Derived Inequalities: Decision Diagrams for Cut Generation in Binary Polynomial Optimization arXiv:2607.28511 ⓘ · analyzed Aug 31, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Replace independent linearizations of binary neural-network gates and their higher-order interactions with a compact decision-diagram flow formulation. This preserves the exact convex hull of feasible gate configurations whenever the DD is exact, making MILP-based pruning, quantization, or architecture search substantially less vulnerable to fractional gate solutions.
Formulas
Mathematical statement
The paper represents a binary polynomial problem with vertex variables z_v in {0,1} and hyperedge variables z_e satisfying z_e = product_{v in e} z_v. A decision diagram B=(N,A) has a source r, sink t, layers N_i, and arcs a labeled by a vertex variable var(a)=v_i and assignment val(a) in {0,1}; E_1(a) is the set of hyperedges whose product evaluates to one on that arc. For every arc a, introduce nonnegative flow y_a. Flow conservation is imposed at every internal node, unit flow leaves r and enters t, and each variable is recovered by z_j = sum_{a in Gamma_j^1} y_a, where Gamma_j^1 is the set of arcs assigning j the value one. Because every source-sink path corresponds to one valid binary assignment and the network-flow polytope is integral, this extended formulation projects to the exact multilinear polytope MP_G rather than a weak product-by-product relaxation.
Implementation notes
Integrate this as the discrete-constraint layer of a MILP for structured pruning or neural architecture search, not as a replacement for the forward pass of a normally trained dense network. Let x_v be binary keep/use variables for channels, experts, bit choices, or edges, and let q_e represent a higher-order interaction such as x_i x_j x_k that appears in a FLOP, memory, latency, or accuracy surrogate. First choose an ordering of the gate variables. Construct a layered DD: each layer assigns one x_v; an arc stores its 0/1 value and the subset E_1(a) of interaction monomials already forced to one. Merge nodes with identical future-relevant state to control width. Create one continuous flow variable y_a per arc and add the following constraints to the MILP: flow conservation, unit source/sink flow, x_j = sum_{a in Gamma_j^1} y_a, and q_e = sum_{a in Gamma_e^1} y_a. Pseudocode is: build_DD(order, interactions); for arc a create y[a]>=0; add_flow_conservation(); add source_flow=1; for each variable j add x[j]=sum(y[a] for a in Gamma1[j]); for each monomial e add q[e]=sum(y[a] for a in Gamma1[e]); optimize the task-loss surrogate plus resource penalties. The DD construction and flow constraints come from the mathematics; DD width, ordering quality, and the task-loss coefficients are empirical. The first experiment should use a small ResNet on CIFAR-10 with channel keep variables and pair/triple resource interactions. Compare independent McCormick linearizations, big-M constraints, and the DD formulation at equal Gurobi time and parameter budget. Success means a larger integrality gap closure, higher validation accuracy at the same FLOPs, or a better feasible compression solution before timeout.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.